o
    ¢h                     @   s<   d dl Z d dlZd dlmZ d dlmZ G dd deZdS )    N)College)BaseCommandc                   @   s    e Zd ZdZdd Zdd ZdS )Commandz7Insert or update a college entry from a JSON data file.c                 C   s(   |j dtddd |j dtddd d S )Nz--slugTzCollege slug)typerequiredhelpz--data_filezPath to JSON data file)add_argumentstr)selfparser r   @/var/www/apihucms/colleges/management/commands/insert_college.pyadd_arguments	   s   zCommand.add_argumentsc           	      O   s   |d }|d }t |d}t|}W d    n1 sw   Y  tjj||d\}}|r>| j| j	d| d d S | j| j	d| d d S )Nslug	data_filer)r   defaultsz	College 'z' created successfully.z' updated successfully.)
openjsonloadr   objectsupdate_or_createstdoutwritestyleSUCCESS)	r
   argsoptionsr   r   fdatacollegecreatedr   r   r   handle   s   
  zCommand.handleN)__name__
__module____qualname__r   r   r"   r   r   r   r   r      s    r   )sysr   colleges.modelsr   django.core.management.baser   r   r   r   r   r   <module>   s
    