__init__.py 3.41 KB
Newer Older
1
# Copyright 2013 GRNET S.A. All rights reserved.
Stavros Sachtouris's avatar
Stavros Sachtouris committed
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
#
# Redistribution and use in source and binary forms, with or
# without modification, are permitted provided that the following
# conditions are met:
#
#   1. Redistributions of source code must retain the above
#      copyright notice, this list of conditions and the following
#      disclaimer.
#
#   2. Redistributions in binary form must reproduce the above
#      copyright notice, this list of conditions and the following
#      disclaimer in the documentation and/or other materials
#      provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
#
# The views and conclusions contained in the software and
# documentation are those of the authors and should not be
# interpreted as representing official policies, either expressed
# or implied, of GRNET S.A.

34

35
36
def _matches(val1, val2, exactMath=True):
    """Case Insensitive match"""
37

38
39
    return (val1.lower() == val2.lower()) if (
        exactMath) else val1.lower().startswith(val2.lower())
40

41
42

def filter_out(d, prefix, exactMatch=False):
43
44
45
46
47
48
49
50
51
52
    """Remove entries that are prefixed with prefix (case insensitive)

    :param d: (dict) input

    :param prefix: (str) prefix to match input keys against

    :param exactMatch: (bool) key should fully match if True, just prefixed
        with prefix if False

    :returns: (dict) the updated d
53
    """
54

55
    ret = dict()
56
    for key, val in d.items():
57
        if not _matches(key, prefix, exactMath=exactMatch):
58
59
            ret[key] = val
    return ret
60

61
62

def filter_in(d, prefix, exactMatch=False):
63
    """Keep only entries of d prefixed with prefix
64

65
    :param d: (dict) input
66

67
    :param prefix: (str) prefix to match input keys against
68

69
70
    :param exactMatch: (bool) key should fully match if True, just prefixed
        with prefix if False
71

72
73
    :returns: (dict) the updated d
    """
74
    ret = dict()
75
    for key, val in d.items():
76
77
        if _matches(key, prefix, exactMath=exactMatch):
            ret[key] = val
78
    return ret
Stavros Sachtouris's avatar
Stavros Sachtouris committed
79

80

81
def path4url(*args):
82
83
84
85
    """
    :param args: (list of str)

    :returns: (str) a path in the form /args[0]/args[1]/...
86
    """
87

Stavros Sachtouris's avatar
Stavros Sachtouris committed
88
    r = '/'.join([''] + [arg.decode('utf-8') if (
89
        isinstance(arg, str)) else '%s' % arg for arg in args])
Stavros Sachtouris's avatar
Stavros Sachtouris committed
90
91
    while '//' in r:
        r = r.replace('//', '/')
92
    return ('/%s' % r.strip('/')) if r else ''
93
94
95
96
97
98
99


def readall(openfile, size, retries=7):
    """Read a file until size is reached"""
    remains = size if size > 0 else 0
    buf = ''
    for i in range(retries):
100
101
102
103
104
105
        tmp_buf = openfile.read(remains)
        if tmp_buf:
            buf += tmp_buf
            remains -= len(tmp_buf)
            if remains > 0:
                continue
106
107
        return buf
    raise IOError('Failed to read %s bytes from file' % size)