-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.py
67 lines (55 loc) · 2.41 KB
/
main.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
import os.path
import traceback
from time import sleep
from colorama import init, Fore, Style
import win32console
from pyfiglet import Figlet
from core import read_excel_table, process_user
def print_head() -> None:
figlet = Figlet(font="speed")
banner = figlet.renderText("Fast practice")
print(f"{Style.BRIGHT}{Fore.MAGENTA}{banner}")
print("All rights reserved by Pochatkov A.R.\n")
print(f"{Fore.RED}telegram: @andrpocc\n" + "vk: andrpocc\n")
def input_path_to_excel() -> str:
print(
f"{Fore.MAGENTA}Перенесите {Fore.RED}Excel{Fore.MAGENTA} файл в окно программы и нажмите клавишу ввода:\n{Fore.RED}{Style.BRIGHT}"
)
path_to_excel = input().replace('"', "")
print(f"\n{Fore.MAGENTA}PDF файлы будут сохранены по указанному выше пути!\n\n{Fore.RED}Таблица из файла:\n")
sleep(1)
return path_to_excel
def main():
init()
win32console.SetConsoleTitle("Fast Practice")
print_head()
excel_file_path = input_path_to_excel()
save_path = os.path.split(excel_file_path)[0]
table = read_excel_table(excel_file_path)
print(table)
print(f"{Fore.MAGENTA}\nВведите номер группы:{Fore.RED}\n")
group = input("Введите номер группы: ")
print()
broken = []
for _, row in table.iterrows():
name = str(row[1])
mark = str(row[2])
context = {"name": name, "mark": mark, "group": group}
print(f"{Fore.MAGENTA}Обрабатываю студента: {Fore.RED}{name}")
try:
process_user(context, save_path)
print(f"{Fore.MAGENTA}Успешно!\n")
except Exception:
print()
print(traceback.format_exc())
print(f"{Fore.MAGENTA}Возникла ошибка!\n")
broken.append(name)
input("Нажмите клавишу ввода, чтобы продолжить..")
if broken:
print(f"Следующие студенты не были обработаны из-за ошибки:{Fore.RED}")
for name in broken:
print(name)
print(f"Все записи обработаны и сохранены по пути: {Fore.RED}{save_path}\n")
input("Нажмите клавишу ввода для выхода из программы..")
if __name__ == "__main__":
main()