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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
# -*- coding: utf-8 -*-
# __init__.py
# Copyright (C) 2013 LEAP
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
"""
Some small and handy functions.
"""
import datetime
import itertools
import os
from leap.bitmask.config import flags
from leap.common.config import get_path_prefix as common_get_path_prefix
# functional goodies for a healthier life:
# We'll give your money back if it does not alleviate the eye strain, at least.
# levelname length == 8, since 'CRITICAL' is the longest
LOG_FORMAT = ('%(asctime)s - %(levelname)-8s - '
'L#%(lineno)-4s : %(name)s:%(funcName)s() - %(message)s')
def first(things):
"""
Return the head of a collection.
:param things: a sequence to extract the head from.
:type things: sequence
:return: object, or None
"""
try:
return things[0]
except (IndexError, TypeError):
return None
def flatten(things):
"""
Return a generator iterating through a flattened sequence.
:param things: a nested sequence, eg, a list of lists.
:type things: sequence
:rtype: generator
"""
return itertools.chain(*things)
# leap repetitive chores
def get_path_prefix():
return common_get_path_prefix(flags.STANDALONE)
def get_modification_ts(path):
"""
Gets modification time of a file.
:param path: the path to get ts from
:type path: str
:returns: modification time
:rtype: datetime object
"""
ts = os.path.getmtime(path)
return datetime.datetime.fromtimestamp(ts)
def update_modification_ts(path):
"""
Sets modification time of a file to current time.
:param path: the path to set ts to.
:type path: str
:returns: modification time
:rtype: datetime object
"""
os.utime(path, None)
return get_modification_ts(path)
def is_file(path):
"""
Returns True if the path exists and is a file.
"""
return os.path.isfile(path)
def is_empty_file(path):
"""
Returns True if the file at path is empty.
"""
return os.stat(path).st_size is 0
def make_address(user, provider):
"""
Return a full identifier for an user, as a email-like
identifier.
:param user: the username
:type user: basestring
:param provider: the provider domain
:type provider: basestring
"""
return "%s@%s" % (user, provider)
|
