U
    lHJe‘  ã                   @   sF   d dl mZ d dlmZ d dlmZ d dlmZ G dd„ dejƒZ	dS )é    )ÚList)Úmessages)Úrepo)ÚMessagingOperationsDictc                       sv   e Zd ZejjZdZejZ	ej
ZdZdZdZdZeedœdd„ƒZeee dœ‡ fdd	„ƒZeedœd
d„ƒZ‡  ZS )ÚCISEntitlementÚciszubuntu-pro-cis.gpgTZ	UbuntuCIS)Úreturnc                 C   s<   | j dkrdtjgiS dtjgi}d| jkr8tjg|d< |S )NÚusgZpost_enableZpre_can_enable)Ú_called_namer   ZCIS_USG_POST_ENABLEZCIS_POST_ENABLEZvalid_namesZCIS_IS_NOW_USG)ÚselfZret© r   ú;/usr/lib/python3/dist-packages/uaclient/entitlements/cis.pyÚ	messaging   s    
 ÿ
zCISEntitlement.messagingc                    s   | j dkrg S tƒ jS )Nr	   )r
   ÚsuperÚpackages©r   ©Ú	__class__r   r   r      s    
zCISEntitlement.packagesc                 C   s   | j dkrtjS tjS )Nr   )r
   r   Z	CIS_TITLEZCIS_USG_TITLEr   r   r   r   Útitle$   s    
zCISEntitlement.title)Ú__name__Ú
__module__Ú__qualname__r   ZurlsZUSG_DOCSZhelp_doc_urlÚnameZCIS_DESCRIPTIONZdescriptionZCIS_HELP_TEXTZ	help_textZrepo_key_fileZapt_noninteractiveZsupports_access_onlyÚoriginÚpropertyr   r   r   Ústrr   r   Ú__classcell__r   r   r   r   r      s   
r   N)
Útypingr   Zuaclientr   Zuaclient.entitlementsr   Zuaclient.typesr   ZRepoEntitlementr   r   r   r   r   Ú<module>   s   