693 lines
31 KiB
Python
693 lines
31 KiB
Python
import sys
|
||
import sqlite3
|
||
from datetime import datetime, date
|
||
from PyQt6.QtWidgets import (QApplication, QMainWindow, QWidget, QVBoxLayout,
|
||
QHBoxLayout, QTabWidget, QTableWidget, QTableWidgetItem,
|
||
QPushButton, QLabel, QLineEdit, QComboBox, QDateEdit,
|
||
QTextEdit, QMessageBox, QHeaderView, QGroupBox,
|
||
QFormLayout, QSpinBox, QCheckBox, QTimeEdit)
|
||
from PyQt6.QtCore import Qt, QDate
|
||
from PyQt6.QtGui import QFont, QPalette, QColor
|
||
from PyQt6.QtCharts import QChart, QChartView, QPieSeries, QBarSeries, QBarSet, QBarCategoryAxis, QValueAxis
|
||
|
||
class FitnessApp(QMainWindow):
|
||
def __init__(self):
|
||
super().__init__()
|
||
self.initDB()
|
||
self.initUI()
|
||
|
||
def initDB(self):
|
||
"""Инициализация базы данных"""
|
||
self.conn = sqlite3.connect('fitness.db')
|
||
self.cursor = self.conn.cursor()
|
||
|
||
# Создание таблиц
|
||
self.create_tables()
|
||
# Заполнение тестовыми данными
|
||
self.insert_sample_data()
|
||
|
||
def create_tables(self):
|
||
"""Создание таблиц базы данных"""
|
||
tables = [
|
||
"""
|
||
CREATE TABLE IF NOT EXISTS Users (
|
||
userID INTEGER PRIMARY KEY,
|
||
fio TEXT NOT NULL,
|
||
phone TEXT,
|
||
email TEXT,
|
||
login TEXT UNIQUE,
|
||
password TEXT,
|
||
userType TEXT,
|
||
specialization TEXT,
|
||
birthDate DATE
|
||
)
|
||
""",
|
||
"""
|
||
CREATE TABLE IF NOT EXISTS Memberships (
|
||
membershipID INTEGER PRIMARY KEY,
|
||
clientID INTEGER,
|
||
membershipType TEXT,
|
||
startDate DATE,
|
||
endDate DATE,
|
||
visitsTotal INTEGER,
|
||
visitsUsed INTEGER,
|
||
zones TEXT,
|
||
membershipStatus TEXT,
|
||
cost REAL,
|
||
adminID INTEGER,
|
||
FOREIGN KEY (clientID) REFERENCES Users(userID),
|
||
FOREIGN KEY (adminID) REFERENCES Users(userID)
|
||
)
|
||
""",
|
||
"""
|
||
CREATE TABLE IF NOT EXISTS Visits (
|
||
visitID INTEGER PRIMARY KEY,
|
||
clientID INTEGER,
|
||
visitDate DATE,
|
||
checkInTime TIME,
|
||
checkOutTime TIME,
|
||
zone TEXT,
|
||
membershipID INTEGER,
|
||
FOREIGN KEY (clientID) REFERENCES Users(userID),
|
||
FOREIGN KEY (membershipID) REFERENCES Memberships(membershipID)
|
||
)
|
||
""",
|
||
"""
|
||
CREATE TABLE IF NOT EXISTS GroupClasses (
|
||
classID INTEGER PRIMARY KEY,
|
||
className TEXT,
|
||
trainerID INTEGER,
|
||
classDate DATE,
|
||
startTime TIME,
|
||
endTime TIME,
|
||
hall TEXT,
|
||
maxParticipants INTEGER,
|
||
enrolledParticipants INTEGER,
|
||
classStatus TEXT,
|
||
FOREIGN KEY (trainerID) REFERENCES Users(userID)
|
||
)
|
||
""",
|
||
"""
|
||
CREATE TABLE IF NOT EXISTS PersonalTraining (
|
||
trainingID INTEGER PRIMARY KEY,
|
||
clientID INTEGER,
|
||
trainerID INTEGER,
|
||
trainingDate DATE,
|
||
startTime TIME,
|
||
endTime TIME,
|
||
exercises TEXT,
|
||
notes TEXT,
|
||
progressMetrics TEXT,
|
||
FOREIGN KEY (clientID) REFERENCES Users(userID),
|
||
FOREIGN KEY (trainerID) REFERENCES Users(userID)
|
||
)
|
||
""",
|
||
"""
|
||
CREATE TABLE IF NOT EXISTS ClassRegistrations (
|
||
registrationID INTEGER PRIMARY KEY,
|
||
classID INTEGER,
|
||
clientID INTEGER,
|
||
registrationDate DATE,
|
||
status TEXT,
|
||
FOREIGN KEY (classID) REFERENCES GroupClasses(classID),
|
||
FOREIGN KEY (clientID) REFERENCES Users(userID)
|
||
)
|
||
""",
|
||
"""
|
||
CREATE TABLE IF NOT EXISTS EquipmentRequests (
|
||
requestID INTEGER PRIMARY KEY,
|
||
trainerID INTEGER,
|
||
equipment TEXT,
|
||
quantity INTEGER,
|
||
requestDate DATE,
|
||
status TEXT,
|
||
FOREIGN KEY (trainerID) REFERENCES Users(userID)
|
||
)
|
||
""",
|
||
"""
|
||
CREATE TABLE IF NOT EXISTS ShiftSwaps (
|
||
swapID INTEGER PRIMARY KEY,
|
||
trainerID1 INTEGER,
|
||
trainerID2 INTEGER,
|
||
shiftDate DATE,
|
||
status TEXT,
|
||
FOREIGN KEY (trainerID1) REFERENCES Users(userID),
|
||
FOREIGN KEY (trainerID2) REFERENCES Users(userID)
|
||
)
|
||
"""
|
||
]
|
||
|
||
for table in tables:
|
||
self.cursor.execute(table)
|
||
self.conn.commit()
|
||
|
||
def insert_sample_data(self):
|
||
"""Вставка тестовых данных"""
|
||
# Проверяем, есть ли уже данные
|
||
self.cursor.execute("SELECT COUNT(*) FROM Users")
|
||
if self.cursor.fetchone()[0] > 0:
|
||
return
|
||
|
||
users = [
|
||
(1, 'Сидорова Марина Петровна', '89219014567', 'director@fitness.ru', 'director1', 'pass1', 'Директор', '', '1980-05-15'),
|
||
(2, 'Романова Анна Сергеевна', '89210125678', 'admin1@fitness.ru', 'admin1', 'pass2', 'Администратор', '', '1992-08-22'),
|
||
(4, 'Яковлева Елена Викторовна', '89211236789', 'admin2@fitness.ru', 'admin2', 'pass3', 'Администратор', '', '1988-11-10'),
|
||
(7, 'Петров Дмитрий Александрович', '89212347890', 'petrov@fitness.ru', 'trainer1', 'pass4', 'Тренер', 'Силовые тренировки', '1985-03-18'),
|
||
(9, 'Смирнова Ольга Игоревна', '89213458901', 'smirnova@fitness.ru', 'trainer2', 'pass5', 'Тренер', 'Йога, Пилатес', '1990-07-25'),
|
||
(11, 'Козлов Сергей Николаевич', '89214569012', 'kozlov@fitness.ru', 'trainer3', 'pass6', 'Тренер', 'Плавание', '1987-12-05'),
|
||
(16, 'Федорова Екатерина Дмитриевна', '89161112236', 'fedorova@mail.ru', 'client1', 'pass7', 'Клиент', '', '1995-04-12'),
|
||
(21, 'Михайлов Алексей Владимирович', '89162223347', 'mikhailov@gmail.com', 'client2', 'pass8', 'Клиент', '', '1988-09-30'),
|
||
(26, 'Новикова Ирина Сергеевна', '89163334458', 'novikova@yandex.ru', 'client3', 'pass9', 'Клиент', '', '1992-06-18'),
|
||
(30, 'Соколов Игорь Петрович', '89164445569', 'sokolov@mail.ru', 'client4', 'pass10', 'Клиент', '', '1983-02-28'),
|
||
(34, 'Павлова Мария Александровна', '89165556670', 'pavlova@gmail.com', 'client5', 'pass11', 'Клиент', '', '1997-11-07')
|
||
]
|
||
|
||
memberships = [
|
||
(1, 16, 'Месячный безлимит', '2024-06-01', '2024-06-30', 999, 42, 'Зал, Бассейн, Групповые', 'Активен', 5000.00, 2),
|
||
(2, 21, '12 посещений', '2024-06-05', '2024-09-05', 12, 8, 'Зал', 'Активен', 4000.00, 2),
|
||
(3, 26, 'Годовой VIP', '2024-01-10', '2025-01-10', 999, 156, 'Все зоны', 'Активен', 45000.00, 4),
|
||
(4, 30, 'Разовое посещение', '2024-06-15', '2024-06-15', 1, 1, 'Зал', 'Завершен', 500.00, 2),
|
||
(5, 34, 'Квартальный', '2024-06-01', '2024-08-31', 999, 15, 'Зал, Групповые', 'Активен', 12000.00, 4)
|
||
]
|
||
|
||
visits = [
|
||
(1, 16, '2024-06-15', '08:30', '10:15', 'Тренажерный зал', 1),
|
||
(2, 21, '2024-06-15', '09:00', '10:30', 'Тренажерный зал', 2),
|
||
(3, 26, '2024-06-15', '07:00', '08:30', 'Бассейн', 3),
|
||
(4, 16, '2024-06-15', '18:00', '19:45', 'Групповое занятие', 1),
|
||
(5, 34, '2024-06-15', '19:00', '20:30', 'Групповое занятие', 5)
|
||
]
|
||
|
||
group_classes = [
|
||
(1, 'Йога для начинающих', 9, '2024-06-16', '10:00', '11:00', 'Зал 2', 15, 12, 'Запланировано'),
|
||
(2, 'Силовая аэробика', 7, '2024-06-16', '18:00', '19:00', 'Зал 1', 20, 18, 'Запланировано'),
|
||
(3, 'Пилатес', 9, '2024-06-17', '11:00', '12:00', 'Зал 2', 12, 12, 'Группа заполнена'),
|
||
(4, 'Аквааэробика', 11, '2024-06-17', '15:00', '16:00', 'Бассейн', 10, 7, 'Запланировано'),
|
||
(5, 'Бокс', 7, '2024-06-18', '19:00', '20:30', 'Зал 3', 8, 5, 'Запланировано')
|
||
]
|
||
|
||
personal_training = [
|
||
(1, 26, 7, '2024-06-14', '16:00', '17:00', 'Жим лежа, Приседания, Тяга блока', 'Хорошая техника', 'Жим 80кг x 8'),
|
||
(2, 16, 9, '2024-06-13', '10:00', '11:00', 'Асаны йоги, Растяжка', 'Улучшилась гибкость', ''),
|
||
(3, 21, 7, '2024-06-12', '14:00', '15:00', 'Становая тяга, Жим гантелей', 'Нужно работать над техникой', 'Становая 60кг x 6')
|
||
]
|
||
|
||
# Вставка данных
|
||
self.cursor.executemany("INSERT INTO Users VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)", users)
|
||
self.cursor.executemany("INSERT INTO Memberships VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", memberships)
|
||
self.cursor.executemany("INSERT INTO Visits VALUES (?, ?, ?, ?, ?, ?, ?)", visits)
|
||
self.cursor.executemany("INSERT INTO GroupClasses VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", group_classes)
|
||
self.cursor.executemany("INSERT INTO PersonalTraining VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)", personal_training)
|
||
self.conn.commit()
|
||
|
||
def initUI(self):
|
||
"""Инициализация пользовательского интерфейса"""
|
||
self.setWindowTitle('Фитнес-клуб - Система управления')
|
||
self.setGeometry(100, 100, 1200, 800)
|
||
|
||
# Центральный виджет с вкладками для разных ролей
|
||
self.tabs = QTabWidget()
|
||
|
||
# Вкладка для администратора
|
||
self.admin_tab = QWidget()
|
||
self.init_admin_tab()
|
||
self.tabs.addTab(self.admin_tab, "Администратор")
|
||
|
||
# Вкладка для тренера
|
||
self.trainer_tab = QWidget()
|
||
self.init_trainer_tab()
|
||
self.tabs.addTab(self.trainer_tab, "Тренер")
|
||
|
||
# Вкладка для директора
|
||
self.director_tab = QWidget()
|
||
self.init_director_tab()
|
||
self.tabs.addTab(self.director_tab, "Директор")
|
||
|
||
self.setCentralWidget(self.tabs)
|
||
|
||
def init_admin_tab(self):
|
||
"""Инициализация вкладки администратора"""
|
||
layout = QVBoxLayout()
|
||
|
||
# Группа управления расписанием
|
||
schedule_group = QGroupBox("Управление расписанием групповых занятий")
|
||
schedule_layout = QVBoxLayout()
|
||
|
||
# Таблица групповых занятий
|
||
self.classes_table = QTableWidget()
|
||
self.classes_table.setColumnCount(10)
|
||
self.classes_table.setHorizontalHeaderLabels([
|
||
'ID', 'Название', 'Тренер', 'Дата', 'Время начала',
|
||
'Время окончания', 'Зал', 'Макс. участников', 'Записано', 'Статус'
|
||
])
|
||
schedule_layout.addWidget(self.classes_table)
|
||
|
||
# Кнопки управления
|
||
btn_layout = QHBoxLayout()
|
||
self.add_class_btn = QPushButton("Добавить занятие")
|
||
self.edit_class_btn = QPushButton("Редактировать")
|
||
self.delete_class_btn = QPushButton("Удалить")
|
||
|
||
btn_layout.addWidget(self.add_class_btn)
|
||
btn_layout.addWidget(self.edit_class_btn)
|
||
btn_layout.addWidget(self.delete_class_btn)
|
||
|
||
schedule_layout.addLayout(btn_layout)
|
||
schedule_group.setLayout(schedule_layout)
|
||
layout.addWidget(schedule_group)
|
||
|
||
# Группа управления абонементами
|
||
membership_group = QGroupBox("Управление абонементами")
|
||
membership_layout = QVBoxLayout()
|
||
|
||
self.memberships_table = QTableWidget()
|
||
self.memberships_table.setColumnCount(8)
|
||
self.memberships_table.setHorizontalHeaderLabels([
|
||
'ID', 'Клиент', 'Тип', 'Начало', 'Окончание',
|
||
'Использовано/Всего', 'Статус', 'Стоимость'
|
||
])
|
||
membership_layout.addWidget(self.memberships_table)
|
||
|
||
# Кнопки для управления абонементами
|
||
membership_btn_layout = QHBoxLayout()
|
||
self.change_price_btn = QPushButton("Изменить стоимость")
|
||
self.freeze_membership_btn = QPushButton("Заморозить абонемент")
|
||
|
||
membership_btn_layout.addWidget(self.change_price_btn)
|
||
membership_btn_layout.addWidget(self.freeze_membership_btn)
|
||
membership_layout.addLayout(membership_btn_layout)
|
||
|
||
membership_group.setLayout(membership_layout)
|
||
layout.addWidget(membership_group)
|
||
|
||
# Группа статистики
|
||
stats_group = QGroupBox("Статистика и отчетность")
|
||
stats_layout = QHBoxLayout()
|
||
|
||
# Левая часть - форма для фильтров
|
||
stats_form = QFormLayout()
|
||
self.stats_start_date = QDateEdit()
|
||
self.stats_end_date = QDateEdit()
|
||
self.stats_start_date.setDate(QDate.currentDate().addMonths(-1))
|
||
self.stats_end_date.setDate(QDate.currentDate())
|
||
|
||
stats_form.addRow("С:", self.stats_start_date)
|
||
stats_form.addRow("По:", self.stats_end_date)
|
||
|
||
self.generate_stats_btn = QPushButton("Сформировать отчет")
|
||
stats_form.addRow(self.generate_stats_btn)
|
||
|
||
stats_layout.addLayout(stats_form)
|
||
|
||
# Правая часть - диаграмма
|
||
self.stats_chart = QChartView()
|
||
stats_layout.addWidget(self.stats_chart)
|
||
|
||
stats_group.setLayout(stats_layout)
|
||
layout.addWidget(stats_group)
|
||
|
||
# Загрузка данных
|
||
self.load_classes_data()
|
||
self.load_memberships_data()
|
||
|
||
# Подключение сигналов
|
||
self.add_class_btn.clicked.connect(self.add_class)
|
||
self.generate_stats_btn.clicked.connect(self.generate_stats)
|
||
|
||
self.admin_tab.setLayout(layout)
|
||
|
||
def init_trainer_tab(self):
|
||
"""Инициализация вкладки тренера"""
|
||
layout = QVBoxLayout()
|
||
|
||
# Группа программ тренировок
|
||
programs_group = QGroupBox("Индивидуальные программы тренировок")
|
||
programs_layout = QVBoxLayout()
|
||
|
||
self.programs_table = QTableWidget()
|
||
self.programs_table.setColumnCount(6)
|
||
self.programs_table.setHorizontalHeaderLabels([
|
||
'ID', 'Клиент', 'Дата', 'Упражнения', 'Заметки', 'Прогресс'
|
||
])
|
||
programs_layout.addWidget(self.programs_table)
|
||
|
||
programs_btn_layout = QHBoxLayout()
|
||
self.add_program_btn = QPushButton("Добавить программу")
|
||
self.edit_program_btn = QPushButton("Редактировать")
|
||
|
||
programs_btn_layout.addWidget(self.add_program_btn)
|
||
programs_btn_layout.addWidget(self.edit_program_btn)
|
||
programs_layout.addLayout(programs_btn_layout)
|
||
|
||
programs_group.setLayout(programs_layout)
|
||
layout.addWidget(programs_group)
|
||
|
||
# Группа базы упражнений
|
||
exercises_group = QGroupBox("База упражнений")
|
||
exercises_layout = QVBoxLayout()
|
||
|
||
self.exercises_table = QTableWidget()
|
||
self.exercises_table.setColumnCount(3)
|
||
self.exercises_table.setHorizontalHeaderLabels(['ID', 'Название', 'Группа мышц'])
|
||
exercises_layout.addWidget(self.exercises_table)
|
||
|
||
exercises_btn_layout = QHBoxLayout()
|
||
self.add_exercise_btn = QPushButton("Добавить упражнение")
|
||
exercises_btn_layout.addWidget(self.add_exercise_btn)
|
||
exercises_layout.addLayout(exercises_btn_layout)
|
||
|
||
exercises_group.setLayout(exercises_layout)
|
||
layout.addWidget(exercises_group)
|
||
|
||
# Группа запросов оборудования
|
||
equipment_group = QGroupBox("Запросы оборудования")
|
||
equipment_layout = QVBoxLayout()
|
||
|
||
self.equipment_table = QTableWidget()
|
||
self.equipment_table.setColumnCount(5)
|
||
self.equipment_table.setHorizontalHeaderLabels([
|
||
'ID', 'Оборудование', 'Количество', 'Дата запроса', 'Статус'
|
||
])
|
||
equipment_layout.addWidget(self.equipment_table)
|
||
|
||
equipment_btn_layout = QHBoxLayout()
|
||
self.request_equipment_btn = QPushButton("Запросить оборудование")
|
||
equipment_btn_layout.addWidget(self.request_equipment_btn)
|
||
equipment_layout.addLayout(equipment_btn_layout)
|
||
|
||
equipment_group.setLayout(equipment_layout)
|
||
layout.addWidget(equipment_group)
|
||
|
||
# Загрузка данных
|
||
self.load_programs_data()
|
||
self.load_equipment_data()
|
||
|
||
self.trainer_tab.setLayout(layout)
|
||
|
||
def init_director_tab(self):
|
||
"""Инициализация вкладки директора"""
|
||
layout = QVBoxLayout()
|
||
|
||
# Общая статистика
|
||
overall_stats_group = QGroupBox("Общая статистика клуба")
|
||
overall_layout = QHBoxLayout()
|
||
|
||
# Левая часть - ключевые показатели
|
||
metrics_layout = QFormLayout()
|
||
self.total_members_label = QLabel("0")
|
||
self.active_members_label = QLabel("0")
|
||
self.monthly_revenue_label = QLabel("0 руб.")
|
||
self.attendance_rate_label = QLabel("0%")
|
||
|
||
metrics_layout.addRow("Всего клиентов:", self.total_members_label)
|
||
metrics_layout.addRow("Активных абонементов:", self.active_members_label)
|
||
metrics_layout.addRow("Доход за месяц:", self.monthly_revenue_label)
|
||
metrics_layout.addRow("Посещаемость:", self.attendance_rate_label)
|
||
|
||
overall_layout.addLayout(metrics_layout)
|
||
|
||
# Правая часть - диаграмма доходов
|
||
self.revenue_chart = QChartView()
|
||
overall_layout.addWidget(self.revenue_chart)
|
||
|
||
overall_stats_group.setLayout(overall_layout)
|
||
layout.addWidget(overall_stats_group)
|
||
|
||
# Эффективность тренеров
|
||
trainers_group = QGroupBox("Эффективность тренеров")
|
||
trainers_layout = QVBoxLayout()
|
||
|
||
self.trainers_table = QTableWidget()
|
||
self.trainers_table.setColumnCount(6)
|
||
self.trainers_table.setHorizontalHeaderLabels([
|
||
'Тренер', 'Групповые занятия', 'Персональные тренировки',
|
||
'Средняя оценка', 'Доход', 'Бонусы'
|
||
])
|
||
trainers_layout.addWidget(self.trainers_table)
|
||
|
||
trainers_group.setLayout(trainers_layout)
|
||
layout.addWidget(trainers_group)
|
||
|
||
# Финансовые показатели
|
||
finance_group = QGroupBox("Финансовые показатели")
|
||
finance_layout = QVBoxLayout()
|
||
|
||
self.finance_table = QTableWidget()
|
||
self.finance_table.setColumnCount(4)
|
||
self.finance_table.setHorizontalHeaderLabels([
|
||
'Период', 'Доход', 'Расходы', 'Прибыль'
|
||
])
|
||
finance_layout.addWidget(self.finance_table)
|
||
|
||
finance_group.setLayout(finance_layout)
|
||
layout.addWidget(finance_group)
|
||
|
||
# Загрузка данных
|
||
self.load_director_data()
|
||
|
||
self.director_tab.setLayout(layout)
|
||
|
||
def load_classes_data(self):
|
||
"""Загрузка данных о групповых занятиях"""
|
||
self.cursor.execute("""
|
||
SELECT gc.classID, gc.className, u.fio, gc.classDate, gc.startTime,
|
||
gc.endTime, gc.hall, gc.maxParticipants, gc.enrolledParticipants, gc.classStatus
|
||
FROM GroupClasses gc
|
||
LEFT JOIN Users u ON gc.trainerID = u.userID
|
||
ORDER BY gc.classDate, gc.startTime
|
||
""")
|
||
classes = self.cursor.fetchall()
|
||
|
||
self.classes_table.setRowCount(len(classes))
|
||
for row, class_data in enumerate(classes):
|
||
for col, data in enumerate(class_data):
|
||
self.classes_table.setItem(row, col, QTableWidgetItem(str(data)))
|
||
|
||
def load_memberships_data(self):
|
||
"""Загрузка данных об абонементах"""
|
||
self.cursor.execute("""
|
||
SELECT m.membershipID, u.fio, m.membershipType, m.startDate, m.endDate,
|
||
m.visitsUsed || '/' || m.visitsTotal, m.membershipStatus, m.cost
|
||
FROM Memberships m
|
||
JOIN Users u ON m.clientID = u.userID
|
||
ORDER BY m.endDate
|
||
""")
|
||
memberships = self.cursor.fetchall()
|
||
|
||
self.memberships_table.setRowCount(len(memberships))
|
||
for row, membership_data in enumerate(memberships):
|
||
for col, data in enumerate(membership_data):
|
||
self.memberships_table.setItem(row, col, QTableWidgetItem(str(data)))
|
||
|
||
def load_programs_data(self):
|
||
"""Загрузка данных о программах тренировок"""
|
||
self.cursor.execute("""
|
||
SELECT pt.trainingID, u.fio, pt.trainingDate, pt.exercises, pt.notes, pt.progressMetrics
|
||
FROM PersonalTraining pt
|
||
JOIN Users u ON pt.clientID = u.userID
|
||
ORDER BY pt.trainingDate DESC
|
||
""")
|
||
programs = self.cursor.fetchall()
|
||
|
||
self.programs_table.setRowCount(len(programs))
|
||
for row, program_data in enumerate(programs):
|
||
for col, data in enumerate(program_data):
|
||
self.programs_table.setItem(row, col, QTableWidgetItem(str(data)))
|
||
|
||
def load_equipment_data(self):
|
||
"""Загрузка данных о запросах оборудования"""
|
||
self.cursor.execute("""
|
||
SELECT requestID, equipment, quantity, requestDate, status
|
||
FROM EquipmentRequests
|
||
ORDER BY requestDate DESC
|
||
""")
|
||
equipment = self.cursor.fetchall()
|
||
|
||
self.equipment_table.setRowCount(len(equipment))
|
||
for row, equipment_data in enumerate(equipment):
|
||
for col, data in enumerate(equipment_data):
|
||
self.equipment_table.setItem(row, col, QTableWidgetItem(str(data)))
|
||
|
||
def load_director_data(self):
|
||
"""Загрузка данных для директора"""
|
||
# Общая статистика
|
||
self.cursor.execute("SELECT COUNT(*) FROM Users WHERE userType = 'Клиент'")
|
||
total_clients = self.cursor.fetchone()[0]
|
||
self.total_members_label.setText(str(total_clients))
|
||
|
||
self.cursor.execute("SELECT COUNT(*) FROM Memberships WHERE membershipStatus = 'Активен'")
|
||
active_memberships = self.cursor.fetchone()[0]
|
||
self.active_members_label.setText(str(active_memberships))
|
||
|
||
self.cursor.execute("""
|
||
SELECT SUM(cost) FROM Memberships
|
||
WHERE strftime('%Y-%m', startDate) = strftime('%Y-%m', 'now')
|
||
""")
|
||
monthly_revenue = self.cursor.fetchone()[0] or 0
|
||
self.monthly_revenue_label.setText(f"{monthly_revenue:.2f} руб.")
|
||
|
||
# Диаграмма доходов
|
||
self.cursor.execute("""
|
||
SELECT membershipType, SUM(cost)
|
||
FROM Memberships
|
||
WHERE strftime('%Y', startDate) = strftime('%Y', 'now')
|
||
GROUP BY membershipType
|
||
""")
|
||
revenue_data = self.cursor.fetchall()
|
||
|
||
series = QPieSeries()
|
||
for membership_type, revenue in revenue_data:
|
||
series.append(membership_type, revenue)
|
||
|
||
chart = QChart()
|
||
chart.addSeries(series)
|
||
chart.setTitle("Доходы по типам абонементов")
|
||
chart.legend().setVisible(True)
|
||
chart.legend().setAlignment(Qt.AlignmentFlag.AlignBottom)
|
||
|
||
self.revenue_chart.setChart(chart)
|
||
|
||
# Данные по тренерам
|
||
self.cursor.execute("""
|
||
SELECT u.fio,
|
||
COUNT(DISTINCT gc.classID) as group_classes,
|
||
COUNT(DISTINCT pt.trainingID) as personal_trainings,
|
||
'4.5' as avg_rating,
|
||
SUM(m.cost * 0.1) as revenue,
|
||
COUNT(DISTINCT pt.trainingID) * 100 as bonuses
|
||
FROM Users u
|
||
LEFT JOIN GroupClasses gc ON u.userID = gc.trainerID
|
||
LEFT JOIN PersonalTraining pt ON u.userID = pt.trainerID
|
||
LEFT JOIN Memberships m ON pt.clientID = m.clientID
|
||
WHERE u.userType = 'Тренер'
|
||
GROUP BY u.userID, u.fio
|
||
""")
|
||
trainers = self.cursor.fetchall()
|
||
|
||
self.trainers_table.setRowCount(len(trainers))
|
||
for row, trainer_data in enumerate(trainers):
|
||
for col, data in enumerate(trainer_data):
|
||
self.trainers_table.setItem(row, col, QTableWidgetItem(str(data)))
|
||
|
||
def add_class(self):
|
||
"""Добавление нового группового занятия"""
|
||
dialog = AddClassDialog(self)
|
||
if dialog.exec():
|
||
class_data = dialog.get_data()
|
||
try:
|
||
self.cursor.execute("""
|
||
INSERT INTO GroupClasses (className, trainerID, classDate, startTime, endTime, hall, maxParticipants, enrolledParticipants, classStatus)
|
||
VALUES (?, ?, ?, ?, ?, ?, ?, 0, 'Запланировано')
|
||
""", class_data)
|
||
self.conn.commit()
|
||
self.load_classes_data()
|
||
QMessageBox.information(self, "Успех", "Занятие успешно добавлено!")
|
||
except Exception as e:
|
||
QMessageBox.critical(self, "Ошибка", f"Не удалось добавить занятие: {str(e)}")
|
||
|
||
def generate_stats(self):
|
||
"""Генерация статистики посещаемости"""
|
||
start_date = self.stats_start_date.date().toString("yyyy-MM-dd")
|
||
end_date = self.stats_end_date.date().toString("yyyy-MM-dd")
|
||
|
||
self.cursor.execute("""
|
||
SELECT zone, COUNT(*) as visits
|
||
FROM Visits
|
||
WHERE visitDate BETWEEN ? AND ?
|
||
GROUP BY zone
|
||
""", (start_date, end_date))
|
||
zone_stats = self.cursor.fetchall()
|
||
|
||
series = QBarSeries()
|
||
bar_set = QBarSet("Посещения по зонам")
|
||
|
||
categories = []
|
||
visits = []
|
||
|
||
for zone, count in zone_stats:
|
||
categories.append(zone)
|
||
visits.append(count)
|
||
|
||
bar_set.append(visits)
|
||
series.append(bar_set)
|
||
|
||
chart = QChart()
|
||
chart.addSeries(series)
|
||
chart.setTitle(f"Посещаемость по зонам ({start_date} - {end_date})")
|
||
|
||
axis_x = QBarCategoryAxis()
|
||
axis_x.append(categories)
|
||
chart.addAxis(axis_x, Qt.AlignmentFlag.AlignBottom)
|
||
series.attachAxis(axis_x)
|
||
|
||
axis_y = QValueAxis()
|
||
chart.addAxis(axis_y, Qt.AlignmentFlag.AlignLeft)
|
||
series.attachAxis(axis_y)
|
||
|
||
self.stats_chart.setChart(chart)
|
||
|
||
class AddClassDialog(QMessageBox):
|
||
def __init__(self, parent=None):
|
||
super().__init__(parent)
|
||
self.setWindowTitle("Добавить групповое занятие")
|
||
self.setModal(True)
|
||
|
||
self.class_name = QLineEdit()
|
||
self.trainer = QComboBox()
|
||
self.class_date = QDateEdit()
|
||
self.start_time = QTimeEdit()
|
||
self.end_time = QTimeEdit()
|
||
self.hall = QComboBox()
|
||
self.max_participants = QSpinBox()
|
||
|
||
self.class_date.setDate(QDate.currentDate())
|
||
self.start_time.setTime(QTime.currentTime())
|
||
self.end_time.setTime(QTime.currentTime().addSecs(3600))
|
||
self.max_participants.setRange(1, 100)
|
||
|
||
# Заполнение комбобоксов
|
||
parent.cursor.execute("SELECT userID, fio FROM Users WHERE userType = 'Тренер'")
|
||
trainers = parent.cursor.fetchall()
|
||
for trainer_id, fio in trainers:
|
||
self.trainer.addItem(fio, trainer_id)
|
||
|
||
self.hall.addItems(['Зал 1', 'Зал 2', 'Зал 3', 'Бассейн'])
|
||
|
||
layout = QFormLayout()
|
||
layout.addRow("Название:", self.class_name)
|
||
layout.addRow("Тренер:", self.trainer)
|
||
layout.addRow("Дата:", self.class_date)
|
||
layout.addRow("Время начала:", self.start_time)
|
||
layout.addRow("Время окончания:", self.end_time)
|
||
layout.addRow("Зал:", self.hall)
|
||
layout.addRow("Макс. участников:", self.max_participants)
|
||
|
||
widget = QWidget()
|
||
widget.setLayout(layout)
|
||
self.layout().addWidget(widget, 0, 0, 1, self.layout().columnCount())
|
||
|
||
self.addButton(QPushButton("Добавить"), QMessageBox.ButtonRole.AcceptRole)
|
||
self.addButton(QPushButton("Отмена"), QMessageBox.ButtonRole.RejectRole)
|
||
|
||
def get_data(self):
|
||
"""Получение данных из формы"""
|
||
return (
|
||
self.class_name.text(),
|
||
self.trainer.currentData(),
|
||
self.class_date.date().toString("yyyy-MM-dd"),
|
||
self.start_time.time().toString("hh:mm"),
|
||
self.end_time.time().toString("hh:mm"),
|
||
self.hall.currentText(),
|
||
self.max_participants.value()
|
||
)
|
||
|
||
if __name__ == '__main__':
|
||
app = QApplication(sys.argv)
|
||
|
||
# Установка стиля
|
||
app.setStyle('Fusion')
|
||
|
||
window = FitnessApp()
|
||
window.show()
|
||
|
||
sys.exit(app.exec())
|