Edit File: license.cpython-37.opt-1.pyc
B 5/9e� � @ s� d dl mZ d dl mZ d dlZd dlZd dlZd dlZd dlmZ d dlm Z d dl m Z d dl m Z d dlZd dlmZ d d lmZmZ d d lmZ de e e e e e dd �d d�Zeed�dd�ZG dd� d�ZdS )� )�print_function)�absolute_importN)�JSONDecodeError)�Path)�run)�Optional)�CloudlinuxLicenseLib)�print_dictionary�replace_params� )� parse_cloudlinux_packages_opts)�error�details�context�returnc C sj dt � � i}| dk r"| |d< d}n d|d< d}|dk r>||d< |dk rN||d< t�|tj� t�|� dS ) a� This function makes UI-specific JSON, prints its and exits from program. It sets "timestamp". And if @param error is specified, then field "result" will be filled with this parameter and exit code will be 1. Otherwise "result" will be filled with "success" and exit code will be 0. @param details: if it's specified, then field "details" will be will be filled with this parameter. @param context: if it's specified, then field "context" will be filled with this parameter. � timestampN�resultr �successr r r )�time�json�dump�sys�stdout�exit)r r r �outputZ exit_code� r �py/cllicense/license.py�_print_as_json_and_exit s r )�validation_errorsr c C s4 d}x*| � � D ] \}}|d|� d|� d�7 }qW |S )z� This function converts dict of validation errors in more human-readable presentation as str. @param validation_errors: errors @return: � z- z: � )�items)r r Zfieldr r r r � _beautify6 s r" c @ sf e Zd ZdZed�Zdd� Zdd� Zdd� Zd d � Z d d � Z ddd�Z e d�dd�Z dd�dd�ZdS )�CloudlinuxLicensez+ Class to check Cloudlinux license z%/etc/lve-utils/cloudlinux-license.inic C sX d| _ i | _t� | _| j�� rNt�� }|�| j� t |d d �rFdnd| _ nd| _ d S )NFZDEFAULTZis_stagez cln-stagingZcln) �_is_json�_optsr � _license_lib� CONFIG_PATH�exists� configparserZ ConfigParser�read�bool� _subdomain)�self�configr r r �__init__I s zCloudlinuxLicense.__init__c C sR d|k| _ | �|�| _| jd r2| �� | �� n | jd rF| �� n| �� dS )z$ Run command action z--json�setZ get_trialr )r$ � _parse_argsr% �_set_run�_get_run� _get_trial)r- �argvr r r r U s zCloudlinuxLicense.runc C s2 | j �� rd}nd}dt�� |d�}t|d� dS )z# Run 'get' command ZvalidZinvalidr )r r �licenseTN)r& Zget_license_statusr r )r- Zlicense_statusZ result_datar r r r3 e s zCloudlinuxLicense._get_runc C s2 | j d r| j�� | j d }|r.| j�|� dS )z# Run 'set' command z--ipz--keyN)r% r& Z!register_and_update_license_by_ipZ"register_and_update_license_by_key)r- �keyr r r r2 r s zCloudlinuxLicense._set_runc C s& t || j�\}}|s"| �t|�� |S )z% Parse CLI arguments )r r$ �_error_and_exitr )r- r5 Zstatus�datar r r r1 ~ s z CloudlinuxLicense._parse_argsr c C s� | j r$|�dt�� i� t|d� nXy tt|d �|�di � � W n6 tk rz } ztdt|�|f � W dd}~X Y nX t� |� dS )z� Print error and exit :param message dict: Dictionary with keys "result" as string and optional "context" as dict r Tr r zError: %s [%s]N) r$ �updater r �print�str�get�KeyErrorr r )r- �messageZ error_code�er r r r8 � s &z!CloudlinuxLicense._error_and_exit)r c C s, yt �| jd �}W n& tk r: td| jd d� Y nX y tjd| j� d�|d�}W n4 tjk r� } ztdt |�d� W d d }~X Y nX y |� � }W n0 tk r� } ztd|j d� W d d }~X Y nX d |kr�|d S |� d �d k�rtd d t |d �id� |� d �d k�r t|d � td� d S )Nz--configz Input config is not valid JSON.)r zhttp://z".cloudlinux.com/wapi/CL-Solo/trial)r zyWe tried to get trial, but cannot connect to server application. Please, try again later or contact our support for help.z|We tried to get trial, but something was gone wrong on server side. Please, try again later or contact our support for help.r7 �typeZ validationzVInput data validation failed: %(error_text)sPlease fix form errors and submit again.Z error_text�errors)r r? zmWe tried to get trial, but something was gone wrong. Please, try again later or contact our support for help.) r �loadsr% r r �requestsZpostr, ZRequestExceptionr< �textr= r" )r- r. Zresponser@ Z response_jsonr r r �_get_key� s2 zCloudlinuxLicense._get_keyNc C s� | � � }tddd|dd| j� d�gdd�}|jd krLtd d �|j�� d �� td dgdd�}|jd kr�td d �|j�� d �� nt� d S )NZ clnreg_ksz--forcez--activationkeyz --serverUrlzhttps://xmlrpc.z.cloudlinux.com/XMLRPCT)Zcapture_outputr zRWe created trial but unable to register server. Please, try to run this command: `� z3` a little bit later. Contact our support for help.z/usr/bin/cldetectz--update-license)rF r r, � returncoder �join�args)r- r7 r r r r r4 � s z CloudlinuxLicense._get_trial)r )�__name__� __module__� __qualname__�__doc__r r' r/ r r3 r2 r1 r8 r< rF r4 r r r r r# C s +r# )NNN)Z __future__r r r) r r r r Zpathlibr � subprocessr �typingr rD Z cllicenser Z cli_utilsr r Zcl_license_arg_parser r< �dictr r" r# r r r r �<module> s" "