common.py
4.2 KB
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
129
130
131
132
133
134
135
136
137
138
139
140
141
142
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time : 2020-02-10 23:12
# @Author : Lemon
# @File : common.py
# @Software: PyCharm
import datetime
import demjson
import base64
from enum import Enum
from attr import attrs, attrib
@attrs(cmp=False)
class WorkPlace():
employer_name = attrib("")
employer_id = attrib("")
position_text = attrib("")
position_id = attrib("")
location_name = attrib("")
location_id = attrib("")
description = attrib("")
current = attrib(False, type=bool)
start_time = attrib(None, type=datetime.date)
end_time = attrib(None, type=datetime.date)
def to_dict(self):
data = {
'employer_name': self.employer_name,
'employer_id': self.employer_id,
'position_text': self.position_text,
'position_id': self.position_id,
'location_name': self.location_name,
'location_id': self.location_id,
'description': self.description,
'date_start[year]': str(self.start_time.year),
'date_start[month]': str(self.start_time.month),
'date_start[day]': str(self.start_time.day),
'date_end[year]': str(self.end_time.year) if self.end_time else '',
'date_end[month]': str(self.end_time.month) if self.end_time else '',
'date_end[day]': str(self.end_time.day) if self.end_time else '',
}
if self.current:
data['date[current]'] = 'on'
return data
@attrs(cmp=False)
class College():
school_text = attrib("")
school_id = attrib("")
subject = attrib([], type=list, converter=lambda x: [] if x is None else x)
school_type = attrib("college")
description = attrib("")
graduated = attrib(False, type=bool) # 'graduated': 'on',
start_time = attrib(None, type=datetime.date)
end_time = attrib(None, type=datetime.date)
def to_dict(self):
data = {
'school_text': self.school_text,
'school_id': self.school_id,
'date_start[year]': str(self.start_time.year),
'date_start[month]': str(self.start_time.month),
'date_start[day]': str(self.start_time.day),
'date_end[year]': str(self.end_time.year) if self.end_time else '',
'date_end[month]': str(self.end_time.month) if self.end_time else '',
'date_end[day]': str(self.end_time.day) if self.end_time else '',
'description': self.description,
'school_type': self.school_type
}
if self.graduated:
data['graduated'] = 'on'
if self.subject:
for i, sub in enumerate(self.subject):
assert len(sub) == 2, '格式不对，要求格式（concentration_text，concentration_ids）'
data["concentration_text[%d]" % i] = sub[0]
data["concentration_ids[%d]" % i] = sub[1]
return data
class TaskStatus():
EXCEPTION = -2
FAILED = -1
EXECUTE = 0
SUCCESS = 1
class Exchange(Enum):
CLIENT = 0
# SERVER = 1
# NSQ = 2
# MQTT = 3
@classmethod
def get(cls, code) -> Enum:
if isinstance(code, str):
code = int(code)
keys = [x for x in dir(cls) if not x.startswith('__')]
for key in keys:
member = getattr(cls, key)
if member.value == code:
return member
return None
def todict(obj, include: list = None):
keys = dir(obj)
res = {}
if include:
for key in include:
try:
value = getattr(obj, key)
res[key] = value
except:
pass
return res
else:
for key in keys:
if not key.startswith('_'):
value = getattr(obj, key)
res[key] = value
return res
def tobase64(obj):
if isinstance(obj, (dict, list)):
obj = demjson.encode(obj)
if isinstance(obj, str):
obj = obj.encode('utf-8')
if isinstance(obj, bytes):
bin = base64.b64encode(obj)
return str(bin, 'utf-8')
raise BaseException("must str,list,dict,bytes")
def frombase64(base):
string = base64.b64decode(base).decode()
return demjson.decode(string)