odweta/solon
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: general bugfixes, display errors

Browse Source
This commit is contained in:
mnjx
2023-04-23 20:05:21 +02:00
parent 93fe4ff01c
commit d9c1b6fc92
6 changed files with 532 additions and 453 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lib/data/grades.csv
+1
View File
@@ -0,0 +1 @@
1
1
src/lib/data/schedule_tdy_tmr.csv
+1
View File
@@ -0,0 +1 @@
1
1
src/lib/data/schedule_week.csv
+1
View File
@@ -0,0 +1 @@
1
1
src/lib/schedule_tdy_tmr.py
+252 -215
View File
@@ -1,268 +1,305 @@
# fetch the schedule for today and tomorrow from the landing page of ŠOL
from lib import networking as nw from lib import networking as nw
from bs4 import BeautifulSoup from bs4 import BeautifulSoup
not_login = 0 not_login = 0
def fetch_locally() -> list: def fetch_locally() -> list:
global not_login global not_login
if not_login == 1: if not_login == 1:
not_login = 0 not_login = 0
#print("Could not sign in (you may have entered invalid login credentials)") #print("Could not sign in (you may have entered invalid login credentials)")
#print("Fetching the schedule for today and tomorrow locally\n") #print("Fetching the schedule for this week locally\n")
schedule_tdy_tmr = schedule_tdy_tmr_from_file() schedule_tdy_tmr = schedule_tdy_tmr_from_file()
return schedule_tdy_tmr return schedule_tdy_tmr
def get_schedule_tdy_tmr(url: str): def get_schedule_tdy_tmr(url: str):
try: try:
html = nw.get_html_from(url) html = nw.get_html_from(url)
soup = BeautifulSoup(html.text, "lxml") soup = BeautifulSoup(html.text, "lxml")
table = soup.find("table", class_="DctTable") table = soup.find("table", id="ctl00_main_boxKalendar_ctl00_ctl04")
rows_odd = table.find_all("tr", class_="RowOdd") rows_odd = table.find_all("tr", class_="RowOdd")
rows_even = table.find_all("tr", class_="RowEven") rows_even = table.find_all("tr", class_="RowEven")
rows = rows_odd + rows_even rows = rows_odd + rows_even
schedule_tdy_tmr = [] schedule_tdy_tmr = []
for i in range(len(rows)): for i in range(len(rows)):
if rows[i].find("td", class_="KuvSuplovanaHodina") is not None and rows[i].find("td", class_="KuvSuplujiciHodina") is None: if rows[i].find("td", class_="KuvSuplovanaHodina") is not None and rows[i].find("td", class_="KuvSuplujiciHodina") is None and rows[i].find("td", class_="KuvSkolniAkceHodina") is None and rows[i].find("td", class_="DctInnerTableType10DataTD") is None:
continue continue
subjects = [] subjects = []
wheres = [] wheres = []
cells = rows[i].find_all("td", attrs={"class": "DctCell"}) cells = rows[i].find_all("td", attrs={"class": "DctCell"})
for j in range(len(cells)):
if cells[j].text == "":
new_span = soup.new_tag("span", attrs={'class': 'KuvBunkaRozvrhNadpis'})
new_span.string = "Volno"
cells[j].append(new_span)
if rows[i].find("td", class_="KuvHeaderNadpis") is not None: header_ = table.find("tr").find_all("td", class_="KuvHeaderNadpis")
day = rows[i].find("td", class_="KuvHeaderNadpis").text header = []
for val in header_:
header.append(val.text)
date = rows[i].find("td", class_="KuvHeaderText").text for j in range(len(cells)):
if cells[j].text == "" and header[j] != " - ":
new_span = soup.new_tag("span", attrs={'class': 'KuvBunkaRozvrhNadpis'})
new_span.string = "Volno"
cells[j].append(new_span)
if rows[i].find("td", class_="KuvHeaderNadpis") is not None:
day = rows[i].find("td", class_="KuvHeaderNadpis").text
subjects_ = rows[i].find_all("span", class_="KuvBunkaRozvrhNadpis") date = rows[i].find("td", class_="KuvHeaderText").text
temp = []
for j in range(len(subjects_)):
temp.append(subjects_[j].text)
subjects = temp
F_con = False
try:
while subjects[-1] == "Volno":
del subjects[-1]
if subjects == []:
F_con = True
except IndexError:
pass
wheres_ = rows[i].find_all("span", class_="KuvBunkaRozvrhText") subjects_ = rows[i].find_all("span", class_="KuvBunkaRozvrhNadpis")
temp = [] temp = []
for j in range(len(wheres_)): for j in range(len(subjects_)):
if len(subjects[j]) < 10: if "KuvSuplovanaHodina" not in subjects_[j].parent["class"]:
temp.append(wheres_[j].text[4:]) temp.append(subjects_[j].text)
else: else:
temp.append(" - ") temp.append("Volno")
wheres = temp subjects = temp
F_continue = False
try:
while subjects[-1] == "Volno":
del subjects[-1]
if subjects == []:
F_continue = True
except IndexError:
pass
for j in range(len(cells)): if subjects == []:
if cells[j].get("colspan") is not None: F_continue = True
if int(cells[j].get("colspan")) == 2:
subjects.insert(j, subjects[j])
wheres.insert(j, wheres[j])
for j in range(len(subjects)):
if subjects[j] == "Volno":
wheres.insert(j, " - ")
if F_con == False: wheres_ = rows[i].find_all("span", class_="KuvBunkaRozvrhText")
schedule_tdy_tmr_row = { temp = []
"DAY": day, for j in range(len(wheres_)):
"DATE": date, if len(subjects[j]) < 10:
"SUBJECTS": subjects, if len(wheres_[j].text[4:]) > 10:
"WHERES": wheres temp.append(" - ")
} elif "KuvSuplovanaHodina" not in wheres_[j].parent["class"]:
temp.append(wheres_[j].text[4:])
else:
temp.append(" - ")
wheres = temp
for j in range(len(cells)):
if cells[j].get("colspan") is not None:
if int(cells[j].get("colspan")) == 2:
try:
subjects.insert(j, subjects[j])
wheres.insert(j, wheres[j])
except IndexError:
pass
for j in range(len(subjects)):
if subjects[j] == "Volno":
wheres.insert(j, " - ")
while True:
if len(subjects) > len(wheres):
del subjects[-1]
elif len(subjects) < len(wheres):
del wheres[-1]
else:
break
schedule_tdy_tmr.append(schedule_tdy_tmr_row) if F_continue == False:
schedule_tdy_tmr_row = {
return schedule_tdy_tmr "DAY": day,
except AttributeError: "DATE": date,
global not_login "SUBJECTS": subjects,
"WHERES": wheres
not_login = 1 }
schedule = fetch_locally()
return schedule schedule_tdy_tmr.append(schedule_tdy_tmr_row)
return schedule_tdy_tmr
except AttributeError:
global not_login
not_login = 1
schedule = fetch_locally()
return schedule
def schedule_tdy_tmr_from_file_to_list(): def schedule_tdy_tmr_from_file_to_list():
schedule = [] schedule = []
schedule_ = [] schedule_ = []
try: try:
with open("src/lib/data/schedule_tdy_tmr.csv") as f: with open("src/lib/data/schedule_tdy_tmr.csv") as f:
import csv import csv
reader = csv.reader(f) reader = csv.reader(f)
for row in reader: for row in reader:
temp = [] temp = []
for col in row: for col in row:
temp.append(col.strip()) temp.append(col.strip())
schedule_.append(temp) schedule_.append(temp)
for i in range(0, len(schedule_), 2): for i in range(0, len(schedule_), 2):
del schedule_[i][0] del schedule_[i][0]
temp = [] temp = []
for i in range(1, len(schedule_), 2): for i in range(1, len(schedule_), 2):
temp.append(schedule_[i][0]) temp.append(schedule_[i][0])
del schedule_[i][0] del schedule_[i][0]
schedule.append(temp) schedule.append(temp)
for i in range(10): for i in range(10):
temp = [] temp = []
for j in range(10): for j in range(10):
try: try:
temp.append(schedule_[j][0]) temp.append(schedule_[j][0])
del schedule_[j][0] del schedule_[j][0]
except IndexError: except IndexError:
pass pass
schedule.append(temp) schedule.append(temp)
except FileNotFoundError: except FileNotFoundError:
print("No schedule_tdy_tmr.csv file found\n") print("No schedule_tdy_tmr.csv file found\n")
return schedule return schedule
def get_whitespace(key: str, idx=0) -> int: def get_whitespace(key: str, idx=0) -> int:
schedule = schedule_tdy_tmr_from_file_to_list() schedule = schedule_tdy_tmr_from_file_to_list()
max_len = 0 max_len = 0
if key == "DATE": if key == "DATE":
try: try:
max_len = len(max(schedule[0], key=len)) max_len = len(max(schedule[0], key=len))
except ValueError: except ValueError:
pass pass
elif key == "WHERES": elif key == "WHERES":
try: try:
max_len = len(max(schedule[idx], key=len)) max_len = len(max(schedule[idx], key=len))
except ValueError: except ValueError:
pass pass
return max_len return max_len
def show_schedule_tdy_tmr(): def show_schedule_tdy_tmr():
# if there is internet access, load stuff from the web # if there is internet access, load stuff from the web
# if not, load the schedule.csv file # if not, load the schedule.csv file
if nw.is_connected() == True: if nw.is_connected() == True:
schedule = get_schedule_tdy_tmr("/SOL/App/Spolecne/KZZ010_RychlyPrehled.aspx") schedule = get_schedule_tdy_tmr("/SOL/App/Spolecne/KZZ010_RychlyPrehled.aspx")
schedule_tdy_tmr_to_file(schedule) schedule_tdy_tmr_to_file(schedule)
else: else:
schedule = schedule_tdy_tmr_from_file() schedule = schedule_tdy_tmr_from_file()
print("") print("")
print("--------------------------") print("--------------------------")
print(" Dnešní a zítřejší rozvrh ") print(" Dnesni a zitrejsi rozvrh ")
print("--------------------------\n") print("--------------------------\n")
if schedule == []: if schedule == []:
print("N/A\n") print("N/A\n")
import os import os
if os.path.exists("src/lib/data/schedule_tdy_tmr.csv"): if os.path.exists("src/lib/data/schedule_tdy_tmr.csv"):
os.remove("src/lib/data/schedule_tdy_tmr.csv") os.remove("src/lib/data/schedule_tdy_tmr.csv")
return return
for i in range(len(schedule)): for i in range(len(schedule)):
max_len = get_whitespace("DATE") max_len = get_whitespace("DATE")
whitespace = " " * (max_len - len(schedule[i]["DAY"])) whitespace = " " * (max_len - len(schedule[i]["DAY"]))
print(schedule[i]["DAY"] + whitespace, end=" ") print(schedule[i]["DAY"] + whitespace, end=" ")
for j in range(len(schedule[i]["SUBJECTS"])): for j in range(len(schedule[i]["SUBJECTS"])):
max_len = get_whitespace("WHERES", j+1) max_len = get_whitespace("WHERES", j+1)
whitespace = " " * (max_len - len(schedule[i]["SUBJECTS"][j])) whitespace = " " * (max_len - len(schedule[i]["SUBJECTS"][j]))
print("| " + schedule[i]["SUBJECTS"][j] + whitespace, end=" ") print("| " + schedule[i]["SUBJECTS"][j] + whitespace, end=" ")
print("") print("")
max_len = get_whitespace("DATE") max_len = get_whitespace("DATE")
whitespace = " " * (max_len - len(schedule[i]["DATE"])) whitespace = " " * (max_len - len(schedule[i]["DATE"]))
print(schedule[i]["DATE"] + whitespace, end=" ") print(schedule[i]["DATE"] + whitespace, end=" ")
for j in range(len(schedule[i]["WHERES"])): for j in range(len(schedule[i]["WHERES"])):
max_len = get_whitespace("WHERES", j+1) max_len = get_whitespace("WHERES", j+1)
whitespace = " " * (max_len - len(schedule[i]["WHERES"][j])) whitespace = " " * (max_len - len(schedule[i]["WHERES"][j]))
print("| " + schedule[i]["WHERES"][j] + whitespace, end=" ") print("| " + schedule[i]["WHERES"][j] + whitespace, end=" ")
print("")
print("")
def schedule_tdy_tmr_to_file(schedule=None): def schedule_tdy_tmr_to_file(schedule=None):
import csv import csv
if schedule is None: if schedule is None:
try: try:
schedule = get_schedule_tdy_tmr("/SOL/App/Spolecne/KZZ010_RychlyPrehled.aspx") schedule = get_schedule_tdy_tmr("/SOL/App/Spolecne/KZZ010_RychlyPrehled.aspx")
except Exception: except Exception:
print("An error occured, you probably can't access the internet right now") print("An error occured, you probably can't access the internet right now")
return return
with open("src/lib/data/schedule_tdy_tmr.csv", mode="w", newline="") as f: with open("src/lib/data/schedule_tdy_tmr.csv", mode="w", newline="") as f:
data = [] data = []
for i in range(len(schedule)): for i in range(len(schedule)):
temp = [] temp = []
temp.append(schedule[i]["DAY"]) try:
for j in range(len(schedule[i]["SUBJECTS"])): temp.append(schedule[i]["DAY"])
temp.append(schedule[i]["SUBJECTS"][j]) except KeyError:
data.append(temp) temp.append("N/A")
for j in range(len(schedule[i]["SUBJECTS"])):
temp = [] try:
temp.append(schedule[i]["DATE"]) temp.append(schedule[i]["SUBJECTS"][j])
for j in range(len(schedule[i]["WHERES"])): except KeyError:
temp.append(schedule[i]["WHERES"][j]) temp.append("N/A")
data.append(temp) data.append(temp)
temp = []
try:
temp.append(schedule[i]["DATE"])
except KeyError:
temp.append("N/A")
for j in range(len(schedule[i]["WHERES"])):
try:
temp.append(schedule[i]["WHERES"][j])
except KeyError:
temp.append("N/A")
data.append(temp)
writer = csv.writer(f) writer = csv.writer(f)
for row in data: for row in data:
writer.writerow(row) writer.writerow(row)
def schedule_tdy_tmr_from_file() -> list: def schedule_tdy_tmr_from_file() -> list:
import csv import csv
schedule = [] schedule = []
try: try:
with open("src/lib/data/schedule_tdy_tmr.csv", mode="r") as f: with open("src/lib/data/schedule_tdy_tmr.csv", mode="r") as f:
reader = csv.reader(f) reader = csv.reader(f)
rows = [] rows = []
for row in reader: for row in reader:
rows.append(row) rows.append(row)
for row in range(0, len(rows), 2): for row in range(0, len(rows), 2):
temp = {} temp = {}
temp["DAY"] = rows[row][0] temp["DAY"] = rows[row][0]
try: try:
temp["DATE"] = rows[row+1][0] temp["DATE"] = rows[row+1][0]
except IndexError: except IndexError:
pass pass
subjects = [] subjects = []
for i in range(1, len(rows[row])): for i in range(1, len(rows[row])):
subjects.append(rows[row][i]) subjects.append(rows[row][i])
temp["SUBJECTS"] = subjects temp["SUBJECTS"] = subjects
wheres = [] wheres = []
try: try:
for i in range(1, len(rows[row+1])): for i in range(1, len(rows[row+1])):
wheres.append(rows[row+1][i]) wheres.append(rows[row+1][i])
except IndexError: except IndexError:
pass pass
temp["WHERES"] = wheres temp["WHERES"] = wheres
schedule.append(temp) schedule.append(temp)
except FileNotFoundError: except FileNotFoundError:
print("No schedule_tdy_tmr.csv file found\n") print("No schedule_tdy_tmr.csv file found\n")
return schedule return schedule
src/lib/schedule_week.py
+254 -226
View File
@@ -4,284 +4,312 @@ from bs4 import BeautifulSoup
not_login = 0 not_login = 0
def fetch_locally() -> list: def fetch_locally() -> list:
global not_login global not_login
if not_login == 1: if not_login == 1:
not_login = 0 not_login = 0
#print("Could not sign in (you may have entered invalid login credentials)") #print("Could not sign in (you may have entered invalid login credentials)")
#print("Fetching the schedule for this week locally\n") #print("Fetching the schedule for this week locally\n")
schedule_week = schedule_week_from_file() schedule_week = schedule_week_from_file()
return schedule_week return schedule_week
def sort_schedule_week(schedule_week: list[dict]) -> list[dict]: def sort_schedule_week(schedule_week: list[dict]) -> list[dict]:
sorted_schedule_week: list[dict] = [{}, {}, {}, {}, {}] sorted_schedule_week: list[dict] = [{}, {}, {}, {}, {}]
for i in range(len(schedule_week)): for i in range(len(schedule_week)):
if schedule_week[i]["DAY"] == "Po": if schedule_week[i]["DAY"] == "Po":
sorted_schedule_week[0] = schedule_week[i] sorted_schedule_week[0] = schedule_week[i]
if schedule_week[i]["DAY"] == "Út": if schedule_week[i]["DAY"] == "Út":
sorted_schedule_week[1] = schedule_week[i] sorted_schedule_week[1] = schedule_week[i]
if schedule_week[i]["DAY"] == "St": if schedule_week[i]["DAY"] == "St":
sorted_schedule_week[2] = schedule_week[i] sorted_schedule_week[2] = schedule_week[i]
if schedule_week[i]["DAY"] == "Čt": if schedule_week[i]["DAY"] == "Čt":
sorted_schedule_week[3] = schedule_week[i] sorted_schedule_week[3] = schedule_week[i]
if schedule_week[i]["DAY"] == "Pá": if schedule_week[i]["DAY"] == "Pá":
sorted_schedule_week[4] = schedule_week[i] sorted_schedule_week[4] = schedule_week[i]
return sorted_schedule_week return sorted_schedule_week
def get_schedule_week(url: str): def get_schedule_week(url: str):
try: try:
html = nw.get_html_from(url) html = nw.get_html_from(url)
soup = BeautifulSoup(html.text, "lxml") soup = BeautifulSoup(html.text, "lxml")
table = soup.find("table", id="CCADynamicCalendarTable") table = soup.find("table", id="CCADynamicCalendarTable")
rows_odd = table.find_all("tr", class_="RowOdd") rows_odd = table.find_all("tr", class_="RowOdd")
rows_even = table.find_all("tr", class_="RowEven") rows_even = table.find_all("tr", class_="RowEven")
rows = rows_odd + rows_even rows = rows_odd + rows_even
schedule_week = [] schedule_week = []
for i in range(len(rows)): for i in range(len(rows)):
if rows[i].find("td", class_="KuvSuplovanaHodina") is not None and rows[i].find("td", class_="KuvSuplujiciHodina") is None: if rows[i].find("td", class_="KuvSuplovanaHodina") is not None and rows[i].find("td", class_="KuvSuplujiciHodina") is None and rows[i].find("td", class_="KuvSkolniAkceHodina") is None:
continue continue
subjects = [] subjects = []
wheres = [] wheres = []
cells = rows[i].find_all("td", attrs={"class": "DctCell"}) cells = rows[i].find_all("td", attrs={"class": "DctCell"})
for j in range(len(cells)):
if cells[j].text == "":
new_span = soup.new_tag("span", attrs={'class': 'KuvBunkaRozvrhNadpis'})
new_span.string = "Volno"
cells[j].append(new_span)
if rows[i].find("td", class_="KuvHeaderNadpis") is not None: header_ = table.find("tr").find_all("td", class_="KuvHeaderNadpis")
day = rows[i].find("td", class_="KuvHeaderNadpis").text header = []
for val in header_:
header.append(val.text)
date = rows[i].find("td", class_="KuvHeaderText").text for j in range(len(cells)):
if cells[j].text == "" and header[j] != " - ":
new_span = soup.new_tag("span", attrs={'class': 'KuvBunkaRozvrhNadpis'})
new_span.string = "Volno"
cells[j].append(new_span)
subjects_ = rows[i].find_all("span", class_="KuvBunkaRozvrhNadpis") if rows[i].find("td", class_="KuvHeaderNadpis") is not None:
temp = [] day = rows[i].find("td", class_="KuvHeaderNadpis").text
for j in range(len(subjects_)):
temp.append(subjects_[j].text)
subjects = temp
F_con = False
try:
while subjects[-1] == "Volno":
del subjects[-1]
if subjects == []:
F_con = True
except IndexError:
pass
wheres_ = rows[i].find_all("span", class_="KuvBunkaRozvrhText") date = rows[i].find("td", class_="KuvHeaderText").text
temp = []
for j in range(len(wheres_)):
if len(subjects[j]) < 10:
temp.append(wheres_[j].text[4:])
else:
temp.append(" - ")
wheres = temp
for j in range(len(cells)): subjects_ = rows[i].find_all("span", class_="KuvBunkaRozvrhNadpis")
if cells[j].get("colspan") is not None: temp = []
if int(cells[j].get("colspan")) == 2: for j in range(len(subjects_)):
subjects.insert(j, subjects[j]) if "KuvSuplovanaHodina" not in subjects_[j].parent["class"]:
wheres.insert(j, wheres[j]) temp.append(subjects_[j].text)
for j in range(len(subjects)): subjects = temp
if subjects[j] == "Volno":
wheres.insert(j, " - ") F_continue = False
try:
while subjects[-1] == "Volno":
del subjects[-1]
if subjects == []:
F_continue = True
except IndexError:
pass
if F_con == False: if subjects == []:
schedule_week_row = { F_continue = True
"DAY": day,
"DATE": date,
"SUBJECTS": subjects,
"WHERES": wheres
}
schedule_week.append(schedule_week_row) wheres_ = rows[i].find_all("span", class_="KuvBunkaRozvrhText")
temp = []
for j in range(len(wheres_)):
try:
if len(subjects[j]) < 10:
if len(wheres_[j].text[4:]) > 10:
temp.append(" - ")
elif "KuvSuplovanaHodina" not in wheres_[j].parent["class"]:
temp.append(wheres_[j].text[4:])
else:
temp.append(" - ")
except IndexError:
pass
wheres = temp
for j in range(len(cells)):
if cells[j].get("colspan") is not None:
if int(cells[j].get("colspan")) == 2:
subjects.insert(j, subjects[j-1])
wheres.insert(j, wheres[j-1])
for j in range(len(subjects)):
if subjects[j] == "Volno":
wheres.insert(j, " - ")
sorted_schedule_week = sort_schedule_week(schedule_week) if F_continue == False:
schedule_week_row = {
"DAY": day,
"DATE": date,
"SUBJECTS": subjects,
"WHERES": wheres
}
schedule_week.append(schedule_week_row)
return sorted_schedule_week sorted_schedule_week = sort_schedule_week(schedule_week)
except AttributeError:
global not_login return sorted_schedule_week
except AttributeError:
not_login = 1 global not_login
schedule = fetch_locally()
return schedule not_login = 1
schedule = fetch_locally()
return schedule
def schedule_week_from_file_to_list(): def schedule_week_from_file_to_list():
schedule = [] schedule = []
schedule_ = [] schedule_ = []
try: try:
with open("src/lib/data/schedule_week.csv") as f: with open("src/lib/data/schedule_week.csv") as f:
import csv import csv
reader = csv.reader(f) reader = csv.reader(f)
for row in reader: for row in reader:
temp = [] temp = []
for col in row: for col in row:
temp.append(col.strip()) temp.append(col.strip())
schedule_.append(temp) schedule_.append(temp)
for i in range(0, len(schedule_), 2): for i in range(0, len(schedule_), 2):
del schedule_[i][0] del schedule_[i][0]
temp = [] temp = []
for i in range(1, len(schedule_), 2): for i in range(1, len(schedule_), 2):
temp.append(schedule_[i][0]) temp.append(schedule_[i][0])
del schedule_[i][0] del schedule_[i][0]
schedule.append(temp) schedule.append(temp)
for i in range(10): for i in range(10):
temp = [] temp = []
for j in range(10): for j in range(10):
try: try:
temp.append(schedule_[j][0]) temp.append(schedule_[j][0])
del schedule_[j][0] del schedule_[j][0]
except IndexError: except IndexError:
pass pass
schedule.append(temp) schedule.append(temp)
except FileNotFoundError: except FileNotFoundError:
print("No schedule_week.csv file found\n") print("No schedule_week.csv file found\n")
return schedule return schedule
def get_whitespace(key: str, idx=0) -> int: def get_whitespace(key: str, idx=0) -> int:
schedule = schedule_week_from_file_to_list() schedule = schedule_week_from_file_to_list()
max_len = 0 max_len = 0
if key == "DATE": if key == "DATE":
try: try:
max_len = len(max(schedule[0], key=len)) max_len = len(max(schedule[0], key=len))
except ValueError: except ValueError:
pass pass
elif key == "WHERES": elif key == "WHERES":
try: try:
max_len = len(max(schedule[idx], key=len)) max_len = len(max(schedule[idx], key=len))
except ValueError: except ValueError:
pass pass
return max_len return max_len
def show_schedule_week(): def show_schedule_week():
# if there is internet access, load stuff from the web # if there is internet access, load stuff from the web
# if not, load the schedule.csv file # if not, load the schedule.csv file
if nw.is_connected() == True: if nw.is_connected() == True:
schedule = get_schedule_week("/SOL/App/Kalendar/KZK001_KalendarTyden.aspx") schedule = get_schedule_week("/SOL/App/Kalendar/KZK001_KalendarTyden.aspx")
schedule_week_to_file(schedule) schedule_week_to_file(schedule)
else: else:
schedule = schedule_week_from_file() schedule = schedule_week_from_file()
print("") print("")
print("----------------") print("----------------")
print(" Týdenní rozvrh ") print(" Týdenní rozvrh ")
print("----------------\n") print("----------------\n")
if schedule == []: if schedule == []:
print("N/A\n") print("N/A\n")
import os import os
if os.path.exists("src/lib/data/schedule_week.csv"): if os.path.exists("src/lib/data/schedule_week.csv"):
os.remove("src/lib/data/schedule_week.csv") os.remove("src/lib/data/schedule_week.csv")
return return
for i in range(len(schedule)): for i in range(len(schedule)):
max_len = get_whitespace("DATE") max_len = get_whitespace("DATE")
whitespace = " " * (max_len - len(schedule[i]["DAY"])) whitespace = " " * (max_len - len(schedule[i]["DAY"]))
print(schedule[i]["DAY"] + whitespace, end=" ") print(schedule[i]["DAY"] + whitespace, end=" ")
for j in range(len(schedule[i]["SUBJECTS"])): for j in range(len(schedule[i]["SUBJECTS"])):
max_len = get_whitespace("WHERES", j+1) max_len = get_whitespace("WHERES", j+1)
whitespace = " " * (max_len - len(schedule[i]["SUBJECTS"][j])) whitespace = " " * (max_len - len(schedule[i]["SUBJECTS"][j]))
print("| " + schedule[i]["SUBJECTS"][j] + whitespace, end=" ") print("| " + schedule[i]["SUBJECTS"][j] + whitespace, end=" ")
print("") print("")
max_len = get_whitespace("DATE") max_len = get_whitespace("DATE")
whitespace = " " * (max_len - len(schedule[i]["DATE"])) whitespace = " " * (max_len - len(schedule[i]["DATE"]))
print(schedule[i]["DATE"] + whitespace, end=" ") print(schedule[i]["DATE"] + whitespace, end=" ")
for j in range(len(schedule[i]["WHERES"])): for j in range(len(schedule[i]["WHERES"])):
max_len = get_whitespace("WHERES", j+1) max_len = get_whitespace("WHERES", j+1)
whitespace = " " * (max_len - len(schedule[i]["WHERES"][j])) whitespace = " " * (max_len - len(schedule[i]["WHERES"][j]))
print("| " + schedule[i]["WHERES"][j] + whitespace, end=" ") print("| " + schedule[i]["WHERES"][j] + whitespace, end=" ")
print("") print("")
def schedule_week_to_file(schedule=None): def schedule_week_to_file(schedule=None):
import csv import csv
if schedule is None: if schedule is None:
try: try:
schedule = get_schedule_week("/SOL/App/Kalendar/KZK001_KalendarTyden.aspx") schedule = get_schedule_week("/SOL/App/Kalendar/KZK001_KalendarTyden.aspx")
except Exception: except Exception:
print("An error occured, you probably can't access the internet right now") print("An error occured, you probably can't access the internet right now")
return return
with open("src/lib/data/schedule_week.csv", mode="w", newline="") as f: with open("src/lib/data/schedule_week.csv", mode="w", newline="") as f:
data = [] data = []
for i in range(len(schedule)): for i in range(len(schedule)):
temp = [] temp = []
temp.append(schedule[i]["DAY"]) try:
for j in range(len(schedule[i]["SUBJECTS"])): temp.append(schedule[i]["DAY"])
temp.append(schedule[i]["SUBJECTS"][j]) except KeyError:
data.append(temp) temp.append("N/A")
for j in range(len(schedule[i]["SUBJECTS"])):
temp = [] try:
temp.append(schedule[i]["DATE"]) temp.append(schedule[i]["SUBJECTS"][j])
for j in range(len(schedule[i]["WHERES"])): except KeyError:
temp.append(schedule[i]["WHERES"][j]) temp.append("N/A")
data.append(temp) data.append(temp)
temp = []
try:
temp.append(schedule[i]["DATE"])
except KeyError:
temp.append("N/A")
for j in range(len(schedule[i]["WHERES"])):
try:
temp.append(schedule[i]["WHERES"][j])
except KeyError:
temp.append("N/A")
data.append(temp)
writer = csv.writer(f) writer = csv.writer(f)
for row in data: for row in data:
writer.writerow(row) writer.writerow(row)
def schedule_week_from_file() -> list: def schedule_week_from_file() -> list:
import csv import csv
schedule = [] schedule = []
try: try:
with open("src/lib/data/schedule_week.csv", mode="r") as f: with open("src/lib/data/schedule_week.csv", mode="r") as f:
reader = csv.reader(f) reader = csv.reader(f)
rows = [] rows = []
for row in reader: for row in reader:
rows.append(row) rows.append(row)
for row in range(0, len(rows), 2): for row in range(0, len(rows), 2):
temp = {} temp = {}
temp["DAY"] = rows[row][0] temp["DAY"] = rows[row][0]
try: try:
temp["DATE"] = rows[row+1][0] temp["DATE"] = rows[row+1][0]
except IndexError: except IndexError:
pass pass
subjects = [] subjects = []
for i in range(1, len(rows[row])): for i in range(1, len(rows[row])):
subjects.append(rows[row][i]) subjects.append(rows[row][i])
temp["SUBJECTS"] = subjects temp["SUBJECTS"] = subjects
wheres = [] wheres = []
try: try:
for i in range(1, len(rows[row+1])): for i in range(1, len(rows[row+1])):
wheres.append(rows[row+1][i]) wheres.append(rows[row+1][i])
except IndexError: except IndexError:
pass pass
temp["WHERES"] = wheres temp["WHERES"] = wheres
schedule.append(temp) schedule.append(temp)
except FileNotFoundError: except FileNotFoundError:
print("No schedule_week.csv file found\n") print("No schedule_week.csv file found\n")
return schedule return schedule
src/main.py
+23 -12
View File
@@ -4,16 +4,27 @@ from lib import schedule_week as sw
from lib import schedule_tdy_tmr as st from lib import schedule_tdy_tmr as st
if __name__ == "__main__": if __name__ == "__main__":
user = str(input("Zadejte přihlašovací jméno: ")) user = str(input("Zadejte přihlašovací jméno: "))
pwd = str(input("Zadejte heslo: ")) pwd = str(input("Zadejte heslo: "))
if nw.is_connected() == True:
nw.login(user, pwd)
if nw.is_connected() == True: #sw.get_schedule_week("/SOL/App/Kalendar/KZK001_KalendarTyden.aspx")
nw.login(user, pwd) #sw.schedule_week_to_file()
#sw.get_schedule_week("/SOL/App/Kalendar/KZK001_KalendarTyden.aspx") #sw.schedule_week_from_file()
sw.show_schedule_week() sw.show_schedule_week()
#gr.get_grades("/SOL/App/Hodnoceni/KZH001_HodnVypisStud.aspx")
gr.show_grades()
#sw.schedule_week_to_file() #gr.get_grades("/SOL/App/Hodnoceni/KZH001_HodnVypisStud.aspx")
#gr.grades_to_file() #gr.grades_to_file()
#sw.schedule_week_from_file() #gr.grades_from_file()
st.show_schedule_tdy_tmr() gr.show_grades()
#st.get_schedule("/SOL/App/Spolecne/KZZ010_RychlyPrehled.aspx")
#st.schedule_tdy_tmr_to_file()
#st.schedule_tdy_tmr_from_file()
st.show_schedule_tdy_tmr()