summaryrefslogtreecommitdiffstats
path: root/lib/ansible/plugins/test/uri.py
blob: 7ef33812cf9f587efc4e02549e613815d5b7aa51 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
# (c) Ansible Project

# Make coding more python3-ish
from __future__ import (absolute_import, division, print_function)
__metaclass__ = type

from urllib.parse import urlparse


def is_uri(value, schemes=None):
    ''' Will verify that the string passed is a valid 'URI', if given a list of valid schemes it will match those '''
    try:
        x = urlparse(value)
        isit = all([x.scheme is not None, x.path is not None, not schemes or x.scheme in schemes])
    except Exception as e:
        isit = False
    return isit


def is_url(value, schemes=None):
    ''' Will verify that the string passed is a valid 'URL' '''

    isit = is_uri(value, schemes)
    if isit:
        try:
            x = urlparse(value)
            isit = bool(x.netloc or x.scheme == 'file')
        except Exception as e:
            isit = False
    return isit


def is_urn(value):
    return is_uri(value, ['urn'])


class TestModule(object):
    ''' Ansible URI jinja2 test '''

    def tests(self):
        return {
            # file testing
            'uri': is_uri,
            'url': is_url,
            'urn': is_urn,
        }