o
    Ήhx                     @   s8   d dl Z d dlmZ d dlmZmZ G dd deZdS )    N)BaseCommand)CityStatec                   @   s   e Zd ZdZdd ZdS )Commandz-Import cities from CSV and assign state by idc           	      O   s   t ddddG}t|}|D ]7}|d }t|d }ztjj|d}tjj||d W q tj	yF   | j
| jd	| d
|  Y qw W d    n1 sQw   Y  | j
| jd d S )Nz%/var/www/apihucms/City-2025-07-30.csv zutf-8)newlineencoding	city_namestate)pk)namer
   z	State id z not found for city zCities imported successfully.)opencsv
DictReaderintr   objectsgetr   get_or_createDoesNotExiststdoutwritestyleWARNINGSUCCESS)	selfargsoptionscsvfilereaderrowr	   state_idr
    r!   =/var/www/apihucms/career/management/commands/import_cities.pyhandle   s   
$
zCommand.handleN)__name__
__module____qualname__helpr#   r!   r!   r!   r"   r      s    r   )r   django.core.management.baser   career.modelsr   r   r   r!   r!   r!   r"   <module>   s    