Edit File: validations.cpython-37.pyc
B V�d � @ s� d Z ddlZddlmZ ddlmZmZmZmZm Z m Z ddl m Z edee dd � d d �ed �eed d � �dd � dd �ed�eedd � �dd � dd �ed�eedd � �dd � dd �ed�eedd � �dd � dd �ed �eed d � �d d � d d �ed �eed!d � �d"d � d#d �ed$�eed%d � �d&d � d'd �ed(�ed)d � �ed*�eed+d � �d,d � d-d �i �Zedee d.d � d/d �ed0�e e d�ed1�eed2�ei�Zeed3�d4d5�Zeed6�d7d8�ZdS )9zF This module contains validation schema for config file loaded values � N)� Namespace)�Schema�And�Optional�Use� SchemaError�Or� )�SSAError�commandc C s | dkS )N)z set-configz get-configzget-ssa-statusz get-reportz enable-ssaz disable-ssazget-stat� )�cr r �J/opt/alt/python37/lib/python3.7/site-packages/ssa/ssa_utils/validations.py�<lambda> � r zInvalid command)�errorZdomains_numberc C s t | �S )N)�int)�nr r r r r c C s | dkS )Nr r )r r r r r r z.number of domains should be a positive integerZ urls_numberc C s t | �S )N)r )r r r r r r c C s | dkS )Nr r )r r r r r r z+number of URLs should be a positive integerZrequests_durationc C s t | �S )N)�float)r r r r r r c C s | dkS )Ng r )r r r r r r z#request duration should be positiveZrequest_numberc C s t | �S )N)r )r r r r r " r c C s | dkS )Nr r )r r r r r # r z/number of requests should be a positive integer�timec C s t | �S )N)r )r r r r r % r c C s d| kodkS S )Nr � r )r r r r r & r z8time should be a positive integer between 1 and 12, or 0Z correlationc C s | � � S )N)� capitalize)�sr r r r ( r c C s | dkS )N)�On�Offr )r r r r r ( r z*bad correlation value, should be On or OffZcorrelation_coefficientc C s t | �S )N)r )r r r r r * r c C s d| k odkS S )Ng g �?r )r r r r r + r z?correlation coefficient could not be negative or greater than 1Z ignore_listc C s t | �S )N)�str)r r r r r - r Z summary_notification_enabledc C s | � � S )N)r )r r r r r / r c C s | dkS )N)r r r )r r r r r / r z+bad notification value, should be On or Offc C s | dkS )N)�enable�disable�statusr )r r r r r 5 r zInvalid autotracing command�user�allZ list_disabled)�data�returnc C sZ y t �| �}|d }|d= ||fS tk rT } ztdt|�� ���W dd}~X Y nX dS )zs Validate given data. Return: tuple with 2 elements: command string and list of the validated options r zInvalid configuration: N)�cmdline_schema�validater r r )r! �d�cmd�er r r r$ = s r$ )� input_argsr" c C sT yt f t�| ��S tk rN } z tt�ddt|�� �i���W dd}~X Y nX dS )z� Validate given input with schema Input arguments expected in s dict form :param input_args: dict with input data �resultzInput validation error: N)r �autotracing_cmdline_schemar$ r � SystemExit�json�dumpsr )r( r' r r r �validate_autotracingK s r. )�__doc__r, �argparser Zschemar r r r r r �internal.exceptionsr r r# �boolr* �dict�tupler$ r. r r r r �<module> sN