o
    h
                     @   s,   d dl mZ d dlmZ G dd deZdS )    )BaseCommand)ExpenseCategoryc                   @   s   e Zd ZdZdd ZdS )Commandz!Set up initial expense categoriesc                 O   s   ddddddddddd	dd
dddddddddddddddddg
}d}|D ]5}t jj|d d|d id\}}|rU|d7 }| j| jd|j  q.| j| jd|j  q.| j| jd| d d S )NzTravel & Transportationz6Business travel, taxi fares, fuel, vehicle maintenance)namedescriptionzOffice Suppliesz(Stationery, equipment, software licenseszMeals & Entertainmentz2Business meals, client entertainment, team luncheszTraining & Developmentz-Training courses, conferences, certificationszMarketing & Advertisingz6Promotional materials, advertising campaigns, brandingzUtilities & Communicationsz'Internet, phone bills, utility paymentszEquipment & Toolsz.Technical equipment, tools, hardware purchaseszProfessional Servicesz&Consulting, legal services, accountingzMaintenance & Repairsz'Equipment repairs, facility maintenanceMiscellaneouszOther business-related expensesr   r   r   )r   defaults   zCreated expense category: z!Expense category already exists: zSuccessfully set up z new expense categories!)	r   objectsget_or_createstdoutwritestyleSUCCESSr   WARNING)selfargsoptions
categoriescreated_countcategory_datacategorycreated r   X/var/www/html/optinet_system/expenditure/management/commands/setup_expense_categories.pyhandle   sd   +


zCommand.handleN)__name__
__module____qualname__helpr   r   r   r   r   r      s    r   N)django.core.management.baser   expenditure.modelsr   r   r   r   r   r   <module>   s   