-
Notifications
You must be signed in to change notification settings - Fork 0
/
local.py
52 lines (48 loc) · 1.68 KB
/
local.py
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
import pandas as pd
import os
from contest import Contest
def read_local_contests():
file_path = 'data/contests.csv'
local_future_contests = list()
if os.path.exists(file_path):
df = pd.read_csv(file_path)
for contest_info in df.to_dict('records'):
conest = Contest(
name_str=contest_info["name"],
time_str=contest_info["time"],
length_str=contest_info["length"],
tz_str=contest_info["timezone"]
)
local_future_contests.append(conest)
else:
pd.DataFrame(columns=[
'name',
'time',
'length',
"timezone"
]).to_csv(file_path, index=False)
return local_future_contests
def save_contests_2_local(contest_list:list[Contest]):
contest_info_list = list()
for contest in contest_list:
contest_info_list.append({
"name": contest.name,
"time": contest.get_time_str_en(),
"length": contest.length_str,
"timezone": contest.tz_str,
})
pd.DataFrame(contest_info_list, columns=[
'name',
'time',
'length',
"timezone"
]).to_csv('data/contests.csv', index=False)
def read_phone_numbers():
df = pd.read_csv('data/phoneNumbers.csv', dtype={'phone numbers': str}).fillna(0)
phone_numbers = pd.Series(df['message count'].values, index=df['phone numbers']).to_dict()
return phone_numbers
def save_phone_numbers(phone_numbers:dict):
pd.DataFrame(
list(phone_numbers.items()),
columns=['phone numbers', 'message count'],
).to_csv('data/phoneNumbers.csv', index=False)