-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathOJTImeTool.py
executable file
·180 lines (170 loc) · 7.5 KB
/
OJTImeTool.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
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
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
#!/Users/edgrass/Documents/Vscode/OJTime/venv/bin/python
import re
import requests
import datetime
from bs4 import BeautifulSoup
from datetime import datetime, timedelta
from selenium import webdriver
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.common.by import By
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from webdriver_manager.chrome import ChromeDriverManager
from colorama import Fore, Style, init
init()
lake_blue = "\033[38;2;0;191;255m"
matcha_green = "\033[38;2;153;204;51m"
orange = "\033[38;5;214m"
reset_color = Style.RESET_ALL
def get_codeforces_contests():
print(f"{lake_blue}CODEFORCES:{reset_color}")
url = 'https://codeforces.com/api/contest.list'
response = requests.get(url)
if response.status_code != 200:
print("Error: Unable to fetch Codeforces contest data.")
return
contests = response.json()['result']
upcoming_contests = [contest for contest in contests if contest['startTimeSeconds'] > datetime.now().timestamp()]
if not upcoming_contests:
print("No upcoming Codeforces contests.")
return
print(f"{lake_blue}Upcoming Codeforces Contests:{reset_color}")
print(f"")
upcoming_contests.reverse()
for contest in upcoming_contests:
start_time = datetime.fromtimestamp(contest['startTimeSeconds'])
duration = str(timedelta(seconds=contest['durationSeconds']))
print(f"{lake_blue}Contest: {contest['name']}{reset_color}")
print(f"{lake_blue}Start Time: {start_time.strftime('%Y-%m-%d %H:%M:%S')}{reset_color}")
print(f"{lake_blue}Duration: {duration}{reset_color}")
print(f"{lake_blue}-{reset_color}" * 40)
def get_atcoder_contests():
print(f"{matcha_green}ATCODER:{reset_color}")
options = Options()
options.add_argument("--headless")
options.add_argument("--disable-gpu")
options.add_argument("--no-sandbox")
service = Service(ChromeDriverManager().install())
driver = webdriver.Chrome(service=service, options=options)
driver.get("https://atcoder.jp/contests")
WebDriverWait(driver, 10).until(EC.visibility_of_element_located((By.CSS_SELECTOR, "table.table-default")))
contests = driver.find_elements(By.CSS_SELECTOR, "table.table-default tr")
current_time = datetime.now()
print(f"{matcha_green}Upcoming Atcoder Contests:{reset_color}")
print(f"")
for contest in contests:
columns = contest.find_elements(By.TAG_NAME, "td")
if len(columns) >= 4:
contest_time_str = columns[0].text.strip()
contest_time_str = re.sub(r'\([A-Za-z]*\)', '', contest_time_str).strip()
try:
contest_time = datetime.strptime(contest_time_str, '%Y-%m-%d %H:%M')
except ValueError:
print(f"{matcha_green}Atcoder比赛时间格式不匹配,跳过: {contest_time_str}{reset_color}")
continue
contest_name = columns[1].text.strip()
contest_duration = columns[2].text.strip()
contest_status = columns[3].text.strip()
if current_time < contest_time:
print(f"{matcha_green}Contest: {contest_name[4:]}{reset_color}")
print(f"{matcha_green}Start Time: {contest_time}{reset_color}")
print(f"{matcha_green}Duration: {contest_duration}:00{reset_color}")
print("-" * 40)
driver.quit()
def codechef_generate_contests(start_contest_number, current_date, limit=5):
contests = []
contest_time = datetime.strptime("2024-12-11 22:30:00", "%Y-%m-%d %H:%M:%S")
duration = "2:00:00"
count = 0
while count < limit:
if contest_time >= current_date:
contest_name = f"Starters {start_contest_number}"
contests.append({
"name": contest_name,
"start_time": contest_time.strftime("%Y-%m-%d %H:%M:%S"),
"duration": duration
})
count += 1
start_contest_number += 1
contest_time += timedelta(days=7)
return contests
def get_codechef_contests():
print(f"{orange}CODECHEF:")
now = datetime.now()
contests = codechef_generate_contests(164, now, limit=5)
print(f"{orange}Upcoming Codechef Contests:")
print(f"")
if contests:
for contest in contests:
name = contest["name"]
start_time = contest["start_time"]
duration = contest["duration"]
print(f"{orange}Contest: {name}")
print(f"{orange}Start Time: {start_time}")
print(f"{orange}Duration: {duration}")
print(f"{orange}-" * 40)
def convert_duration_to_hms(match_time_text):
duration_match = re.search(r"时长:(\d+)小时(?:\d+分钟)?", match_time_text)
if duration_match:
hours = int(duration_match.group(1))
minutes = 0
minute_match = re.search(r"(\d+)分钟", match_time_text)
if minute_match:
minutes = int(minute_match.group(1))
return f"{hours:02}:{minutes:02}:00"
else:
return "Unknown"
def get_nowcoder_contests():
print(f"{Fore.MAGENTA}NOWCODER:{Style.RESET_ALL}")
options = Options()
options.add_argument("--headless")
options.add_argument("--disable-gpu")
options.add_argument("--no-sandbox")
service = Service(ChromeDriverManager().install())
driver = webdriver.Chrome(service=service, options=options)
driver.get("https://ac.nowcoder.com/acm/contest/vip-index")
try:
WebDriverWait(driver, 10).until(
EC.visibility_of_element_located((By.CSS_SELECTOR, "div.platform-item-main"))
)
except Exception as e:
print(f"{Fore.MAGENTA}页面加载失败: {e}{Style.RESET_ALL}")
driver.quit()
return
contests = driver.find_elements(By.CSS_SELECTOR, "div.platform-item-main")
current_time = datetime.now()
print(f"{Fore.MAGENTA}Upcoming Nowcoder Contests:{Style.RESET_ALL}")
print(f"")
for contest in contests:
try:
contest_name = contest.find_element(By.CSS_SELECTOR, "h4 a").text.strip()
match_time_text = contest.find_element(By.CSS_SELECTOR, "li.match-time-icon").text.strip()
match = re.search(r'(\d{4}-\d{2}-\d{2} \d{2}:\d{2})', match_time_text)
if not match:
print(f"{Fore.MAGENTA}Nowcoder比赛时间格式不匹配，跳过: {match_time_text}{Style.RESET_ALL}")
continue
start_time_str = match.group(1)
start_time = datetime.strptime(start_time_str, "%Y-%m-%d %H:%M")
duration_match = re.search(r"时长:(\d+小时(?:\d+分钟)?)", match_time_text)
duration_display = convert_duration_to_hms(match_time_text)
duration = duration_match.group(1) if duration_match else "Unknown"
if current_time < start_time:
print(f"{Fore.MAGENTA}Contest: {contest_name}{Style.RESET_ALL}")
print(f"{Fore.MAGENTA}Start Time: {start_time}{Style.RESET_ALL}")
print(f"{Fore.MAGENTA}Duration: {duration_display}{Style.RESET_ALL}")
print("-" * 40)
except Exception as e:
print(f"{Fore.MAGENTA}处理比赛数据时出错: {e}{Style.RESET_ALL}")
continue
driver.quit()
def get_all_contests():
print(f"")
get_codeforces_contests()
print(f"")
get_atcoder_contests()
print(f"")
get_nowcoder_contests()
print(f"")
get_codechef_contests()
get_all_contests()