File: //opt/alt/python37/lib/python3.7/site-packages/ssa/ssa_utils/__pycache__/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