B
    !_b                 @   s4   d dl mZmZ d dlmZmZ G dd deZdS )    )BaseCommandCommandError)
serialListupdateModelFromNodec               @   s    e Zd ZdZdd Zdd ZdS )Commandz7Update local licenses DB using data from cellframe-nodec             C   s   |j dtd d S )Nz
--pageSize)type)add_argumentint)selfparser r   I/var/www/html/tbcc-django/licenses/management/commands/update_licenses.pyadd_arguments   s    zCommand.add_argumentsc       
      O   s   t ddd }|dkr.| j| jd d S |d r@|d }nd}xZtd||D ]J}t ||}|d }x2|D ]*}t|}	|	sn| j| jd	|  qnqnW qRW | j| jd
 d S )N
total_only)
totalStatetotalr   zLicenses list is emptypageSized   serialsz'Can't find license info with serial: %szSuccessfully updated licenses)	r   stdoutwritestyleINFOranger   WARNINGSUCCESS)
r
   argsoptionsr   r   offsetinfor   serialfoundr   r   r   handle   s     



zCommand.handleN)__name__
__module____qualname__helpr   r"   r   r   r   r   r      s   r   N)django.core.management.baser   r   Zlicenses.clir   r   r   r   r   r   r   <module>   s   