Personal frequently used public python function code
Time
1
2
3
4
5
6
7
| import time
def get_day_time():
return time.strftime("%Y-%m-%d", time.localtime())
def get_detail_time():
return time.strftime("%Y-%m-%d_%H:%M:%S", time.localtime())
|
Log
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
| import os
import time
def get_detail_time():
return time.strftime("%Y-%m-%d_%H:%M:%S", time.localtime())
log_level_dict = {
'DEBUG': 1,
'INFO': 2,
'WARNING': 3,
'ERROR': 4,
'CRITICAL': 5
}
def print_log(log_info, log_level='INFO'):
os.makedirs('./log', exist_ok=True)
log_dir = './log/nlp.log'
current_level = log_level_dict['INFO']
log = "[{0}] {1}: {2}\n".format(log_level, get_detail_time(), log_info)
if log_level_dict[log_level] >= current_level:
with open(log_dir, 'a+') as f:
f.write(log)
|
Write / Read File
1
2
3
4
5
6
7
| def write_file(file_name, data):
with open(file_name, 'w') as f:
f.write(data)
def read_file(file_name):
with open(file_name, 'r') as f:
return f.read()
|
Write / Read Json File
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
| import json
def write_file(file_name, data):
with open(file_name, 'w') as f:
f.write(data)
def read_file(file_name):
with open(file_name, 'r') as f:
return f.read()
def json_to_dict(json_data):
return json.loads(json_data)
def dict_to_json(dict_data):
return json.dumps(dict_data, indent=4, ensure_ascii=False)
def write_json_file(file_name, dic):
with open(file_name, 'w') as f:
f.write(dict_to_json(dic))
def read_json_file(file_name):
with open(file_name, 'r', encoding='utf8') as f:
return json.load(f)
|
Get / Post Request
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
| import requests
def send_req_by_get(url):
return requests.get(url).text
def send_req_by_post(url, body):
return requests.post(
url,
data=body,
headers={'Content-Type': 'application/json'}
).text
'''Need some code shown above'''
def get_raw_data(data):
if 'code' in data:
if data['code'] == 200:
return data['data']
raise Exception('Fail To Send Request! Error Message: {0} {1}'.format(data['code'], data['message']))
def get_rsp_data(url, req_type, body=""):
if req_type.upper() == "GET":
return get_raw_data(json_to_dict(send_req_by_get(url)))
elif req_type.upper() == "POST":
if body == "":
raise Exception('Empty body to post.')
else:
return get_raw_data(json_to_dict(send_req_by_post(url, body)))
else:
raise Exception('Error reqType: {}'.format(req_type))
|