blob: 0214d7d7e0d3b195adf15c6fefe5a715360f4360 (
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
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
|
import os
import sys
import string
import random
import time
try:
import pwd
except ImportError:
import getpass
pwd = None
using_python_3 = True if sys.version_info.major == 3 else False
def get_current_user():
if pwd:
return pwd.getpwuid(os.geteuid()).pw_name
else:
return getpass.getuser()
def user_input():
if using_python_3:
return input()
else:
# using python version 2
return raw_input()
def random_id_gen_unsafe(length=8):
"""
A generator for creating a random chars id of specific length
:parameters:
length : int
the desired length of the generated id
default: 8
:return:
a random id with each next() request.
"""
id_chars = string.ascii_lowercase + string.digits
while True:
return_id = ''
for i in range(length):
return_id += random.choice(id_chars)
yield return_id
class random_id_gen:
""" Thread safe version of random_id_gen_unsafe """
def __init__(self, length=8):
self.id_chars = string.ascii_lowercase + string.digits
self.length = length
def next(self):
return ''.join(random.choice(self.id_chars) for _ in range(self.length))
__next__ = next
# try to get number from input, return None in case of fail
def get_number(input):
try:
return long(input)
except:
try:
return int(input)
except:
return None
def list_intersect(l1, l2):
return list(filter(lambda x: x in l2, l1))
def list_difference (l1, l2):
return list(filter(lambda x: x not in l2, l1))
def is_sub_list (l1, l2):
return set(l1) <= set(l2)
# a simple passive timer
class PassiveTimer(object):
# timeout_sec = None means forever
def __init__ (self, timeout_sec):
if timeout_sec != None:
self.expr_sec = time.time() + timeout_sec
else:
self.expr_sec = None
def has_expired (self):
# if no timeout was set - return always false
if self.expr_sec == None:
return False
return (time.time() > self.expr_sec)
|
