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
118
119
120
121
122
123
124
125
126
127
128
|
# Copyright 2016 Egor Tensin <Egor.Tensin@gmail.com>
# This file is licensed under the terms of the MIT License.
# See LICENSE.txt for details.
from datetime import datetime
from enum import Enum
from numbers import Real, Integral
class Field(Enum):
UID = 'uid'
FIRST_NAME = 'first_name'
LAST_NAME = 'last_name'
SCREEN_NAME = 'screen_name'
ONLINE = 'online'
LAST_SEEN = 'last_seen'
def __str__(self):
return self.value
class User:
def __init__(self, impl):
self._impl = impl
def __str__(self):
return str(self._impl)
def __eq__(self, other):
return self.get_uid() == other.get_uid()
def __hash__(self):
return hash(self.get_uid())
def __iter__(self):
return iter(self._impl)
def __contains__(self, field):
if field is Field.LAST_SEEN:
return self._has_last_seen()
return self._normalize_field(field) in self._impl
def __getitem__(self, field):
if field is Field.LAST_SEEN:
return self._get_last_seen()
return self._impl[self._normalize_field(field)]
def __setitem__(self, field, value):
if field is Field.LAST_SEEN:
self._set_last_seen(value)
else:
self._impl[self._normalize_field(field)] = value
@staticmethod
def _normalize_field(field):
if isinstance(field, Field):
return field.value
return field
def get_uid(self):
return self[Field.UID]
def get_first_name(self):
return self[Field.FIRST_NAME]
def set_first_name(self, name):
self[Field.FIRST_NAME] = name
def has_last_name(self):
return Field.LAST_NAME in self and self.get_last_name()
def get_last_name(self):
return self[Field.LAST_NAME]
def set_last_name(self, name):
self[Field.LAST_NAME] = name
def has_screen_name(self):
return Field.SCREEN_NAME in self
def get_screen_name(self):
if self.has_screen_name():
return self[Field.SCREEN_NAME]
else:
return 'id' + str(self.get_uid())
def set_screen_name(self, name):
self[Field.SCREEN_NAME] = name
def has_online(self):
return Field.ONLINE in self
def is_online(self):
return bool(self[Field.ONLINE])
def set_online(self, value=True):
self[Field.ONLINE] = value
@staticmethod
def _last_seen_from_timestamp(t):
return datetime.fromtimestamp(t)
@staticmethod
def _last_seen_to_timestamp(t):
if isinstance(t, datetime):
return t.timestamp()
elif isinstance(t, Real) or isinstance(t, Integral):
return t
else:
raise TypeError('"last seen" time must be either a `datetime` or a POSIX timestamp')
def _has_last_seen(self):
return Field.LAST_SEEN.value in self._impl and 'time' in self._impl[Field.LAST_SEEN.value]
def has_last_seen(self):
return self._has_last_seen()
def _get_last_seen(self):
return self._last_seen_from_timestamp(self._impl[Field.LAST_SEEN.value]['time'])
def get_last_seen(self):
return self._get_last_seen()
def _set_last_seen(self, t):
if Field.LAST_SEEN.value not in self._impl:
self._impl[Field.LAST_SEEN.value] = {}
self._impl[Field.LAST_SEEN.value]['time'] = self._last_seen_to_timestamp(t)
def set_last_seen(self, t):
self._set_last_seen(t)
|
