o
    ›Ï´f‘  ã                   @   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defdd„ƒZedee f‡ fdd	„ƒZedefd
d„ƒZ‡  ZS )ÚCISEntitlementÚciszubuntu-pro-cis.gpgTÚ	UbuntuCISÚreturnc                 C   s<   | j dkrdtjgiS dtjgi}d| jv rtjg|d< |S )NÚusgÚpost_enableÚpre_can_enable)Ú_called_namer   ÚCIS_USG_POST_ENABLEÚCIS_POST_ENABLEÚvalid_namesÚCIS_IS_NOW_USG)ÚselfÚ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   Ú	CIS_TITLEÚCIS_USG_TITLEr   r   r   r   Útitle$   s   
zCISEntitlement.title)Ú__name__Ú
__module__Ú__qualname__r   ÚurlsÚUSG_DOCSÚhelp_doc_urlÚnameÚCIS_DESCRIPTIONÚdescriptionÚCIS_HELP_TEXTÚ	help_textÚrepo_key_fileÚapt_noninteractiveÚsupports_access_onlyÚoriginÚpropertyr   r   r   Ústrr   r   Ú__classcell__r   r   r   r   r      s    
r   N)
Útypingr   Úuaclientr   Úuaclient.entitlementsr   Úuaclient.typesr   ÚRepoEntitlementr   r   r   r   r   Ú<module>   s
    