Function validate_input

step-2-create-inventory.py:160–212

(input_field, value, hosts_yaml)

    return count

def validate_input(input_field, value, hosts_yaml):
    IP_ADDRESSES = ['ansible_host', 'relay_host_ip']
    NUMBER = ['backend_port', 'gophish_admin_port',]
    STRING = ['identification', 'send_from', 'relay_host', 'dns_beacon_subdomain']
    NAME = ['name']
    CAMPAIGN_NAME = ['CampaignName']
    DOMAIN = ['domain_name']
    URL = ['nginx_bounce_site']

    try:
        if input_field in IP_ADDRESSES:
            IP(value)

        elif input_field in NUMBER:
            if not value.isdigit():
                print('Provided value is not a number')
                return False

        elif input_field in STRING:
            if (' ' in value) or \
                ('"' in value) or \
                ("'" in value) or \
                ('/' in value) or \
                ('\\' in value):
                print('Provided value is not a valid string')
                return False

        elif input_field in NAME:
            names = get_all_names(hosts_yaml)
            if value in names:
                print('Provided name already exists')
                return False
        elif input_field in CAMPAIGN_NAME:
            if not value.isalnum():
                print('Provided value is not alphanumeric')
                return False
        elif input_field in DOMAIN:
            if not re.match(r'''
            (?=^.{,253}$)          # max. length 253 chars
            (?!^.+\.\d+$)          # TLD is not fully numerical
            (?=^[^-.].+[^-.]$)     # doesn't start/end with '-' or '.'
            (?!^.+(\.-|-\.).+$)    # levels don't start/end with '-'
            (?:[a-z\d-]            # uses only allowed chars
            {1,63}(\.|$))          # max. level length 63 chars
            {2,127}                # max. 127 levels
            ''',value, re.X | re.I):
                return False
        elif input_field in URL:
            if not validators.url(value):
                return False
    except:
        return False
    return True

###############################################################################
# Print functions