CodeScannerSystem/main.py at main · sandexzx/CodeScannerSystem · GitHub

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
import os
import sys
from scanner_handler import ScannerHandler
from config_manager import load_config, save_config, SCANNER_FILE_PATH
from rich.console import Console
from rich.panel import Panel
from rich.table import Table
from rich.prompt import Prompt, Confirm
from rich import print as rprint
from rich.progress import Progress, SpinnerColumn, TextColumn

console = Console()

def clear_screen():
    os.system('cls' if os.name == 'nt' else 'clear')

def print_menu():
    console.print(Panel.fit(
        "[bold blue]Система учета товаров с отслеживанием файла данных сканера[/bold blue]",
        border_style="blue"
    ))

    table = Table(show_header=False, box=None)
    table.add_row("1", "Продолжить смену")
    table.add_row("2", "Начать новую смену")
    table.add_row("3", "Показать текущие настройки")
    table.add_row("4", "Изменить путь к файлу сканера")
    table.add_row("5", "Изменить вместимость коробки")
    table.add_row("6", "Выход")

    console.print(table)

def show_settings():
    config = load_config()
    table = Table(title="Текущие настройки", show_header=True, header_style="bold magenta")
    table.add_column("Параметр", style="cyan")
    table.add_column("Значение", style="green")

    table.add_row("Путь к файлу сканера", config['scanner_file_path'])
    table.add_row("Вместимость коробки", str(config['box_capacity']))

    console.print(table)
    Prompt.ask("\nНажмите Enter для продолжения")

def change_scanner_file():
    config = load_config()
    new_path = Prompt.ask("\nВведите новый путь к файлу сканера")
    if new_path:
        config["scanner_file_path"] = new_path
        save_config(config)
        console.print(f"[green]Путь к файлу сканера изменен на:[/green] {new_path}")
    Prompt.ask("\nНажмите Enter для продолжения")

def change_box_capacity():
    config = load_config()
    while True:
        try:
            new_capacity = int(Prompt.ask("\nВведите новую вместимость коробки"))
            if new_capacity > 0:
                config["box_capacity"] = new_capacity
                save_config(config)
                console.print(f"[green]Вместимость коробки изменена на:[/green] {new_capacity}")
                break
            else:
                console.print("[red]Вместимость должна быть положительным числом![/red]")
        except ValueError:
            console.print("[red]Пожалуйста, введите корректное число![/red]")
    Prompt.ask("\nНажмите Enter для продолжения")

def main():
    while True:
        clear_screen()
        print_menu()

        choice = Prompt.ask("\nВыберите действие", choices=["1", "2", "3", "4", "5", "6"])

        if choice == "1":
            clear_screen()
            console.print("[bold blue]Продолжение существующей смены...[/bold blue]")
            handler = ScannerHandler(start_new_session=False)
            handler.start_monitoring()
        elif choice == "2":
            clear_screen()
            console.print("[bold blue]Запуск новой смены...[/bold blue]")
            handler = ScannerHandler(start_new_session=True)
            handler.start_monitoring()
        elif choice == "3":
            show_settings()
        elif choice == "4":
            change_scanner_file()
        elif choice == "5":
            change_box_capacity()
        elif choice == "6":
            if Confirm.ask("\nВы уверены, что хотите выйти?"):
                console.print("\n[yellow]Завершение работы программы...[/yellow]")
                sys.exit(0)
        else:
            console.print("\n[red]Неверный выбор! Попробуйте снова.[/red]")
            Prompt.ask("Нажмите Enter для продолжения")

if __name__ == "__main__":
    main()