08102025
This commit is contained in:
parent
9a0a8fd903
commit
7922dea546
526
main.py
526
main.py
|
|
@ -1,20 +1,21 @@
|
|||
import sqlite3
|
||||
import datetime
|
||||
# import pandas as pd
|
||||
# Контрольная работа илья
|
||||
import pandas as pd
|
||||
import os
|
||||
import sys
|
||||
import openpyxl
|
||||
|
||||
# Контрольная работа илья
|
||||
|
||||
conn = sqlite3.connect('partners.db')
|
||||
cursor = conn.cursor()
|
||||
def create_tables(db_connection):
|
||||
#Создает все необходимые таблицы в базе данных
|
||||
cursor = db_connection.cursor()
|
||||
|
||||
# Создаем таблицы
|
||||
cursor.executescript('''
|
||||
CREATE TABLE IF NOT EXISTS PartnerType (
|
||||
cursor.executescript('''
|
||||
CREATE TABLE IF NOT EXISTS PartnerType (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
type_name VARCHAR(100) NOT NULL UNIQUE);
|
||||
type_name VARCHAR(100) NOT NULL UNIQUE
|
||||
);
|
||||
|
||||
CREATE TABLE IF NOT EXISTS Partner (
|
||||
CREATE TABLE IF NOT EXISTS Partner (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
partner_type_id INTEGER REFERENCES PartnerType(id),
|
||||
name VARCHAR(255) NOT NULL,
|
||||
|
|
@ -25,21 +26,24 @@ CREATE TABLE IF NOT EXISTS Partner (
|
|||
inn VARCHAR(20) UNIQUE,
|
||||
rating DECIMAL(3,2) DEFAULT 0.0,
|
||||
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
||||
);
|
||||
CREATE TABLE IF NOT EXISTS ProductType (
|
||||
);
|
||||
|
||||
CREATE TABLE IF NOT EXISTS ProductType (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
type_name VARCHAR(100) NOT NULL UNIQUE,
|
||||
coefficient DECIMAL(5,2) NOT NULL DEFAULT 1.0
|
||||
);
|
||||
CREATE TABLE IF NOT EXISTS Product (
|
||||
);
|
||||
|
||||
CREATE TABLE IF NOT EXISTS Product (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
product_type_id INTEGER REFERENCES ProductType(id),
|
||||
product_name VARCHAR(255) NOT NULL,
|
||||
article VARCHAR(100) UNIQUE NOT NULL,
|
||||
min_partner_price DECIMAL(15,2) NOT NULL,
|
||||
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
||||
);
|
||||
CREATE TABLE IF NOT EXISTS Sale (
|
||||
);
|
||||
|
||||
CREATE TABLE IF NOT EXISTS Sale (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
product_id INTEGER REFERENCES Product(id),
|
||||
partner_id INTEGER REFERENCES Partner(id),
|
||||
|
|
@ -47,109 +51,405 @@ CREATE TABLE IF NOT EXISTS Sale (
|
|||
sale_date DATE NOT NULL,
|
||||
sale_price DECIMAL(15,2) NOT NULL,
|
||||
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
||||
);
|
||||
CREATE INDEX IF NOT EXISTS idx_partner_type ON Partner(partner_type_id);
|
||||
CREATE INDEX IF NOT EXISTS idx_partner_inn ON Partner(inn);
|
||||
CREATE INDEX IF NOT EXISTS idx_product_type ON Product(product_type_id);
|
||||
CREATE INDEX IF NOT EXISTS idx_product_article ON Product(article);
|
||||
CREATE INDEX IF NOT EXISTS idx_sale_product ON Sale(product_id);
|
||||
CREATE INDEX IF NOT EXISTS idx_sale_partner ON Sale(partner_id);
|
||||
CREATE INDEX IF NOT EXISTS idx_sale_date ON Sale(sale_date);
|
||||
''')
|
||||
partner_types = [
|
||||
('ЗАО',),
|
||||
('ООО',),
|
||||
('ПАО',),
|
||||
('ОАО',)
|
||||
]
|
||||
cursor.executemany('INSERT OR IGNORE INTO PartnerType (type_name) VALUES (?)', partner_types)
|
||||
);
|
||||
|
||||
partners = [
|
||||
(1, 'База Строитель', 'Иванова Александра Ивановна', 'aleksandraivanova@ml.ru',
|
||||
'493 123 45 67', '652050, Кемеровская область, город Юрга, ул. Лесная, 15', '2222455179', 7),
|
||||
(2, 'Паркет 29', 'Петров Василий Петрович', 'vppetrov@vl.ru',
|
||||
'987 123 56 78', '164500, Архангельская область, город Северодвинск, ул. Строителей, 18', '3333888520', 7),
|
||||
(3, 'Стройсервис', 'Соловьев Андрей Николаевич', 'ansolovev@st.ru',
|
||||
'812 223 32 00', '188910, Ленинградская область, город Приморск, ул. Парковая, 21', '4440391035', 7),
|
||||
(4, 'Ремонт и отделка', 'Воробьева Екатерина Валерьевна', 'ekaterina.vorobeva@ml.ru',
|
||||
'444 222 33 11', '143960, Московская область, город Реутов, ул. Свободы, 51', '1111520857', 5),
|
||||
(1, 'МонтажПро', 'Степанов Степан Сергеевич', 'stepanov@stepan.ru',
|
||||
'912 888 33 33', '309500, Белгородская область, город Старый Оскол, ул. Рабочая, 122', '5552431140', 10)
|
||||
]
|
||||
CREATE INDEX IF NOT EXISTS idx_partner_type ON Partner(partner_type_id);
|
||||
CREATE INDEX IF NOT EXISTS idx_partner_inn ON Partner(inn);
|
||||
CREATE INDEX IF NOT EXISTS idx_product_type ON Product(product_type_id);
|
||||
CREATE INDEX IF NOT EXISTS idx_product_article ON Product(article);
|
||||
CREATE INDEX IF NOT EXISTS idx_sale_product ON Sale(product_id);
|
||||
CREATE INDEX IF NOT EXISTS idx_sale_partner ON Sale(partner_id);
|
||||
CREATE INDEX IF NOT EXISTS idx_sale_date ON Sale(sale_date);
|
||||
''')
|
||||
|
||||
cursor.executemany('''
|
||||
INSERT OR IGNORE INTO Partner (partner_type_id, name, director, email, phone, legal_address, inn, rating)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?)
|
||||
''', partners)
|
||||
|
||||
# Заполняем типы продукции
|
||||
product_types = [
|
||||
('Ламинат', 2.35),
|
||||
('Массивная доска', 5.15),
|
||||
('Паркетная доска', 4.34),
|
||||
('Пробковое покрытие', 1.5)
|
||||
]
|
||||
|
||||
cursor.executemany('INSERT OR IGNORE INTO ProductType (type_name, coefficient) VALUES (?, ?)', product_types)
|
||||
|
||||
products = [
|
||||
(3, 'Паркетная доска Ясень темный однополосная 14 мм', '8758385', 4456.90),
|
||||
(3, 'Инженерная доска Дуб Французская елка однополосная 12 мм', '8858958', 7330.99),
|
||||
(1, 'Ламинат Дуб дымчато-белый 33 класс 12 мм', '7750282', 1799.33),
|
||||
(1, 'Ламинат Дуб серый 32 класс 8 мм с фаской', '7028748', 3890.41),
|
||||
(4, 'Пробковое напольное клеевое покрытие 32 класс 4 мм', '5012543', 5450.59)
|
||||
]
|
||||
|
||||
cursor.executemany('''
|
||||
INSERT OR IGNORE INTO Product (product_type_id, product_name, article, min_partner_price)
|
||||
VALUES (?, ?, ?, ?)
|
||||
''', products)
|
||||
|
||||
# Заполняем продажи
|
||||
sales_data = [
|
||||
# Формат: (product_name, partner_name, quantity, sale_date)
|
||||
('Паркетная доска Ясень темный однополосная 14 мм', 'База Строитель', 15500, '2023-03-23'),
|
||||
('Ламинат Дуб дымчато-белый 33 класс 12 мм', 'База Строитель', 12350, '2023-12-18'),
|
||||
('Ламинат Дуб серый 32 класс 8 мм с фаской', 'База Строитель', 37400, '2024-06-07'),
|
||||
('Инженерная доска Дуб Французская елка однополосная 12 мм', 'Паркет 29', 35000, '2022-12-02'),
|
||||
('Пробковое напольное клеевое покрытие 32 класс 4 мм', 'Паркет 29', 1250, '2023-05-17'),
|
||||
('Ламинат Дуб дымчато-белый 33 класс 12 мм', 'Паркет 29', 1000, '2024-06-07'),
|
||||
('Паркетная доска Ясень темный однополосная 14 мм', 'Паркет 29', 7550, '2024-07-01'),
|
||||
('Паркетная доска Ясень темный однополосная 14 мм', 'Стройсервис', 7250, '2023-01-22'),
|
||||
('Инженерная доска Дуб Французская елка однополосная 12 мм', 'Стройсервис', 2500, '2024-07-05'),
|
||||
('Ламинат Дуб серый 32 класс 8 мм с фаской', 'Ремонт и отделка', 59050, '2023-03-20'),
|
||||
('Ламинат Дуб дымчато-белый 33 класс 12 мм', 'Ремонт и отделка', 37200, '2024-03-12'),
|
||||
('Пробковое напольное клеевое покрытие 32 класс 4 мм', 'Ремонт и отделка', 4500, '2024-05-14'),
|
||||
('Ламинат Дуб дымчато-белый 33 класс 12 мм', 'МонтажПро', 50000, '2023-09-19'),
|
||||
('Ламинат Дуб серый 32 класс 8 мм с фаской', 'МонтажПро', 670000, '2023-11-10'),
|
||||
('Паркетная доска Ясень темный однополосная 14 мм', 'МонтажПро', 35000, '2024-04-15'),
|
||||
('Инженерная доска Дуб Французская елка однополосная 12 мм', 'МонтажПро', 25000, '2024-06-12')
|
||||
]
|
||||
db_connection.commit()
|
||||
print("✅ Таблицы успешно созданы")
|
||||
|
||||
|
||||
def get_ids(product_name, partner_name):
|
||||
cursor.execute('SELECT id FROM Product WHERE product_name = ?', (product_name,))
|
||||
product_id = cursor.fetchone()[0]
|
||||
def calculate_discount(total_quantity):
|
||||
if total_quantity >= 300000:
|
||||
return 15, '15% (более 300000 ед.)'
|
||||
elif total_quantity >= 50000:
|
||||
return 10, '10% (50000-300000 ед.)'
|
||||
elif total_quantity >= 10000:
|
||||
return 5, '5% (10000-50000 ед.)'
|
||||
else:
|
||||
return 0, '0% (до 10000 ед.)'
|
||||
|
||||
cursor.execute('SELECT id FROM Partner WHERE name = ?', (partner_name,))
|
||||
partner_id = cursor.fetchone()[0]
|
||||
|
||||
cursor.execute('SELECT min_partner_price FROM Product WHERE product_name = ?', (product_name,))
|
||||
sale_price = cursor.fetchone()[0]
|
||||
|
||||
return product_id, partner_id, sale_price
|
||||
|
||||
for sale in sales_data:
|
||||
product_name, partner_name, quantity, sale_date = sale
|
||||
def get_partners(db_path='partners.db', sort_by='rating', ascending=False):
|
||||
try:
|
||||
product_id, partner_id, sale_price = get_ids(product_name, partner_name)
|
||||
cursor.execute('''
|
||||
INSERT INTO Sale (product_id, partner_id, quantity, sale_date, sale_price)
|
||||
VALUES (?, ?, ?, ?, ?)
|
||||
''', (product_id, partner_id, quantity, sale_date, sale_price))
|
||||
conn = sqlite3.connect(db_path)
|
||||
query = """
|
||||
SELECT
|
||||
pt.type_name AS "Тип партнера",
|
||||
p.name AS "Наименование партнера",
|
||||
p.director AS "Директор",
|
||||
p.email AS "Электронная почта",
|
||||
p.phone AS "Телефон",
|
||||
p.legal_address AS "Юридический адрес",
|
||||
p.inn AS "ИНН",
|
||||
p.rating AS "Рейтинг",
|
||||
p.created_at AS "Дата создания",
|
||||
COALESCE(SUM(s.quantity), 0) as total_quantity
|
||||
FROM Partner p
|
||||
JOIN PartnerType pt ON p.partner_type_id = pt.id
|
||||
LEFT JOIN Sale s ON p.id = s.partner_id
|
||||
GROUP BY p.id
|
||||
"""
|
||||
|
||||
#Сортировку
|
||||
if sort_by == 'discount':
|
||||
query += " ORDER BY total_quantity"
|
||||
elif sort_by == 'rating':
|
||||
query += " ORDER BY p.rating"
|
||||
elif sort_by == 'name':
|
||||
query += " ORDER BY p.name"
|
||||
elif sort_by == 'type':
|
||||
query += " ORDER BY pt.type_name, p.name"
|
||||
|
||||
elif sort_by == 'inn':
|
||||
query += " ORDER BY p.inn"
|
||||
|
||||
else:
|
||||
query += " ORDER BY p.rating"
|
||||
|
||||
query += " DESC" if not ascending else " ASC"
|
||||
|
||||
df = pd.read_sql_query(query, conn)
|
||||
conn.close()
|
||||
|
||||
#Колонки со скидкой
|
||||
if not df.empty:
|
||||
df['Скидка'] = df['total_quantity'].apply(lambda x: calculate_discount(x)[0])
|
||||
df['Текст_скидки'] = df['total_quantity'].apply(lambda x: calculate_discount(x)[1])
|
||||
|
||||
return df
|
||||
|
||||
except sqlite3.Error as e:
|
||||
print(f"❌ Ошибка базы данных: {e}")
|
||||
return None
|
||||
except Exception as e:
|
||||
print(f"Ошибка при добавлении продажи: {sale}, ошибка: {e}")
|
||||
print(f"❌ Общая ошибка: {e}")
|
||||
return None
|
||||
|
||||
|
||||
conn.commit()
|
||||
conn.close()
|
||||
def get_sales_history(db_path='partners.db', limit=50, sort_by='date_desc'):
|
||||
#Получаем историю реализации продукции с разными вариантами сортировки
|
||||
try:
|
||||
conn = sqlite3.connect(db_path)
|
||||
query = """
|
||||
SELECT
|
||||
s.sale_date AS "Дата продажи",
|
||||
p.name AS "Партнер",
|
||||
pr.product_name AS "Продукция",
|
||||
pr.article AS "Артикул",
|
||||
s.quantity AS "Количество",
|
||||
s.sale_price AS "Цена за единицу",
|
||||
(s.quantity * s.sale_price) AS "Общая сумма",
|
||||
pt.type_name AS "Тип продукции"
|
||||
FROM Sale s
|
||||
JOIN Partner p ON s.partner_id = p.id
|
||||
JOIN Product pr ON s.product_id = pr.id
|
||||
JOIN ProductType pt ON pr.product_type_id = pt.id
|
||||
"""
|
||||
|
||||
#Сортировка по дате
|
||||
if sort_by == 'date_desc':
|
||||
query += " ORDER BY s.sale_date DESC, s.created_at DESC"
|
||||
elif sort_by == 'date_asc':
|
||||
query += " ORDER BY s.sale_date ASC, s.created_at ASC"
|
||||
elif sort_by == 'quantity_desc':
|
||||
query += " ORDER BY s.quantity DESC"
|
||||
elif sort_by == 'amount_desc':
|
||||
query += " ORDER BY (s.quantity * s.sale_price) DESC"
|
||||
else:
|
||||
query += " ORDER BY s.sale_date DESC, s.created_at DESC"
|
||||
|
||||
query += " LIMIT ?"
|
||||
|
||||
df = pd.read_sql_query(query, conn, params=(limit,))
|
||||
conn.close()
|
||||
return df
|
||||
except Exception as e:
|
||||
print(f"❌ Ошибка при получении истории продаж: {e}")
|
||||
return None
|
||||
|
||||
def print_partner_cards(partners_df):
|
||||
#Выводим карточку партнеров
|
||||
|
||||
if partners_df is None or partners_df.empty:
|
||||
print("📭 Нет данных о партнерах")
|
||||
return
|
||||
|
||||
for index, row in partners_df.iterrows():
|
||||
print("╔" + "═" * 69 + "╗")
|
||||
print(f"║ {'КАРТОЧКА ПАРТНЕРА':^67} ║")
|
||||
print("╠" + "═" * 69 + "╣")
|
||||
|
||||
type_text = f"Тип: {row['Тип партнера']}"
|
||||
name_text = f"Наименование: {row['Наименование партнера']}"
|
||||
director_text = f"Директор: {row['Директор']}"
|
||||
rating_text = f"Рейтинг: {row['Рейтинг']}"
|
||||
quantity_text = f"Продано единиц: {row['total_quantity']:,}"
|
||||
discount_text = f"Скидка: {row['Текст_скидки']}"
|
||||
email_text = f"Контакты: {row['Электронная почта']}"
|
||||
phone_text = f"Телефон: {row['Телефон']}"
|
||||
inn_text = f"ИНН: {row['ИНН']}"
|
||||
|
||||
address = row['Юридический адрес']
|
||||
if len(address) > 60:
|
||||
address = address[:57] + "..."
|
||||
address_text = f"Адрес: {address}"
|
||||
|
||||
print(f"║ {type_text:<67} ║")
|
||||
print(f"║ {name_text:<67} ║")
|
||||
print(f"║ {director_text:<67} ║")
|
||||
print(f"║ {rating_text:<67} ║")
|
||||
print(f"║ {quantity_text:<67} ║")
|
||||
print(f"║ {discount_text:<67} ║")
|
||||
print(f"║ {'':<67} ║")
|
||||
print(f"║ {email_text:<67} ║")
|
||||
print(f"║ {phone_text:<67} ║")
|
||||
print(f"║ {'':<67} ║")
|
||||
print(f"║ {inn_text:<67} ║")
|
||||
print(f"║ {address_text:<67} ║")
|
||||
print("╚" + "═" * 69 + "╝")
|
||||
print()
|
||||
|
||||
|
||||
def print_sales_cards(sales_df):
|
||||
#Выводим карточку исории реализации
|
||||
if sales_df is None or sales_df.empty:
|
||||
print("📭 Нет данных о продажах")
|
||||
return
|
||||
|
||||
print("╔" + "═" * 147 + "╗")
|
||||
print(f"║ {'ИСТОРИЯ РЕАЛИЗАЦИИ ПРОДУКЦИИ':^145} ║")
|
||||
print("╠" + "═" * 147 + "╣")
|
||||
|
||||
#Заголовок таблицы
|
||||
headers = ["Дата", "Партнер", "Продукция", "Артикул", "Кол-во", "Цена", "Сумма"]
|
||||
print("║ {:<12} {:<18} {:<59} {:<3} {:>9} {:>8} {:>13} ║".format(
|
||||
headers[0], headers[1], headers[2], headers[3], headers[4], headers[5], headers[6]
|
||||
))
|
||||
print("╠" + "═" * 147 + "╣")
|
||||
|
||||
#Данные таблицы
|
||||
for index, row in sales_df.iterrows():
|
||||
date = str(row['Дата продажи'])[:10]
|
||||
partner = row['Партнер']
|
||||
product = row['Продукция']
|
||||
article = str(row['Артикул'])
|
||||
quantity = f"{row['Количество']:,}"
|
||||
price = f"{row['Цена за единицу']:,.2f}"
|
||||
total = f"{row['Общая сумма']:,.2f}"
|
||||
|
||||
print(f"║ {date} ║ {partner.ljust(16)} ║ {product.ljust(57)} ║ {article.ljust(8)} ║ {quantity.ljust(8)} ║ {price.ljust(10)} ║ {total.ljust(18)} ║")
|
||||
|
||||
print("╚" + "═" * 147 + "╝")
|
||||
|
||||
#Итоги
|
||||
total_quantity = sales_df['Количество'].sum()
|
||||
total_amount = sales_df['Общая сумма'].sum()
|
||||
print(f"📊 ИТОГО: {len(sales_df)} продаж, {total_quantity:,} единиц, общая сумма: {total_amount:,.2f} руб.")
|
||||
def simple_choice(options):
|
||||
#Функция выбора
|
||||
print("\n" + "=" * 50)
|
||||
print("ВЫБЕРИТЕ ОПЦИЮ:")
|
||||
print("=" * 50)
|
||||
for i, option in enumerate(options, 1):
|
||||
print(f"{i}. {option}")
|
||||
|
||||
while True:
|
||||
try:
|
||||
choice = input(f"\nВведите номер (1-{len(options)}): ").strip()
|
||||
if choice.isdigit():
|
||||
choice_num = int(choice)
|
||||
if 1 <= choice_num <= len(options):
|
||||
return options[choice_num - 1]
|
||||
print(f"❌ Пожалуйста, введите число от 1 до {len(options)}")
|
||||
except (ValueError, EOFError):
|
||||
print("❌ Неверный ввод. Попробуйте еще раз.")
|
||||
except KeyboardInterrupt:
|
||||
print("\n👋 Программа прервана пользователем.")
|
||||
sys.exit(0)
|
||||
|
||||
|
||||
def check_database():
|
||||
#Проверка файлов базы двнных
|
||||
conn = sqlite3.connect('partners.db')
|
||||
try:
|
||||
cursor = conn.cursor()
|
||||
print("\n🔍 Проверка загруженных данных:")
|
||||
tables = ['PartnerType', 'Partner', 'ProductType', 'Product', 'Sale']
|
||||
for table in tables:
|
||||
cursor.execute(f"SELECT COUNT(*) FROM {table}")
|
||||
count = cursor.fetchone()[0]
|
||||
print(f" {table}: {count} записей")
|
||||
finally:
|
||||
conn.close()
|
||||
|
||||
|
||||
def load_data_from_files():
|
||||
#Загружает файлы в базу данных
|
||||
conn = sqlite3.connect('partners.db')
|
||||
cursor = conn.cursor()
|
||||
|
||||
try:
|
||||
#Запрос файлов
|
||||
print("\n📁 Введите имена файлов:")
|
||||
partners_file = input("Файл с партнерами: ")
|
||||
product_type_file = input("Файл с типами продукции: ")
|
||||
products_file = input("Файл с продуктами: ")
|
||||
sales_file = input("Файл с продажами: ")
|
||||
|
||||
#Загрузка типов партнеров
|
||||
print("\n1. 📊 Загрузка типов партнеров...")
|
||||
partners_df = pd.read_excel(partners_file)
|
||||
partner_types = partners_df['Тип партнера'].unique()
|
||||
for partner_type in partner_types:
|
||||
cursor.execute("INSERT OR IGNORE INTO PartnerType (type_name) VALUES (?)", (partner_type,))
|
||||
|
||||
#Загрузка партнеров
|
||||
print("2. 👥 Загрузка партнеров...")
|
||||
for _, row in partners_df.iterrows():
|
||||
cursor.execute("SELECT id FROM PartnerType WHERE type_name = ?", (row['Тип партнера'],))
|
||||
partner_type_id = cursor.fetchone()[0]
|
||||
|
||||
cursor.execute("""INSERT OR IGNORE INTO Partner
|
||||
(partner_type_id, name, director, email, phone, legal_address, inn, rating)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?)""",
|
||||
(partner_type_id, row['Наименование партнера'], row['Директор'],
|
||||
row['Электронная почта партнера'], row['Телефон партнера'],
|
||||
row['Юридический адрес партнера'], row['ИНН'], row['Рейтинг'])
|
||||
)
|
||||
|
||||
#Загрузка типов продукции
|
||||
print("3. 📦 Загрузка типов продукции...")
|
||||
product_type_df = pd.read_excel(product_type_file)
|
||||
for _, row in product_type_df.iterrows():
|
||||
cursor.execute("INSERT OR IGNORE INTO ProductType (type_name, coefficient) VALUES (?, ?)",
|
||||
(row['Тип продукции'], row['Коэффициент типа продукции'])
|
||||
)
|
||||
|
||||
#Загрузка продуктов
|
||||
print("4. 🏷️ Загрузка продуктов...")
|
||||
products_df = pd.read_excel(products_file)
|
||||
for _, row in products_df.iterrows():
|
||||
cursor.execute("SELECT id FROM ProductType WHERE type_name = ?", (row['Тип продукции'],))
|
||||
product_type_id = cursor.fetchone()[0]
|
||||
|
||||
cursor.execute("""INSERT OR IGNORE INTO Product
|
||||
(product_type_id, product_name, article, min_partner_price)
|
||||
VALUES (?, ?, ?, ?)""",
|
||||
(product_type_id, row['Наименование продукции'],
|
||||
row['Артикул'], row['Минимальная стоимость для партнера'])
|
||||
)
|
||||
|
||||
#Загрузка продаж
|
||||
print("5. 💰 Загрузка продаж...")
|
||||
sales_df = pd.read_excel(sales_file)
|
||||
loaded_sales = 0
|
||||
|
||||
for _, row in sales_df.iterrows():
|
||||
#Находим ID продукта
|
||||
cursor.execute("SELECT id FROM Product WHERE product_name = ?", (row['Продукция'],))
|
||||
product_result = cursor.fetchone()
|
||||
if not product_result:
|
||||
print(f" ⚠️ Продукт не найден: {row['Продукция']}")
|
||||
continue
|
||||
product_id = product_result[0]
|
||||
|
||||
#Находим ID партнера
|
||||
cursor.execute("SELECT id FROM Partner WHERE name = ?", (row['Наименование партнера'],))
|
||||
partner_result = cursor.fetchone()
|
||||
if not partner_result:
|
||||
print(f" ⚠️ Партнер не найден: {row['Наименование партнера']}")
|
||||
continue
|
||||
partner_id = partner_result[0]
|
||||
|
||||
#Получаем цену продажи
|
||||
cursor.execute("SELECT min_partner_price FROM Product WHERE id = ?", (product_id,))
|
||||
min_price_result = cursor.fetchone()
|
||||
sale_price = min_price_result[0] if min_price_result else 0
|
||||
|
||||
#Преобразуем дату
|
||||
sale_date = pd.to_datetime(row['Дата продажи']).strftime('%Y-%m-%d')
|
||||
|
||||
#Вставляем продажу
|
||||
cursor.execute("""INSERT INTO Sale
|
||||
(product_id, partner_id, quantity, sale_date, sale_price)
|
||||
VALUES (?, ?, ?, ?, ?)""",
|
||||
(product_id, partner_id, row['Количество продукции'], sale_date, sale_price)
|
||||
)
|
||||
loaded_sales += 1
|
||||
|
||||
conn.commit()
|
||||
print(f"✅ Данные успешно загружены! Загружено {loaded_sales} продаж")
|
||||
check_database()
|
||||
|
||||
except Exception as e:
|
||||
print(f"❌ Ошибка при загрузке данных: {e}")
|
||||
conn.rollback()
|
||||
finally:
|
||||
conn.close()
|
||||
|
||||
|
||||
def display_menu():
|
||||
#Главное меню программы
|
||||
menu_options = [
|
||||
'КАРТОЧКИ ПАРТНЕРОВ (сортировка по рейтингу)',
|
||||
'КАРТОЧКИ ПАРТНЕРОВ (сортировка по скидкам)',
|
||||
'КАРТОЧКИ ПАРТНЕРОВ (сортировка по ИНН)',
|
||||
'ИСТОРИЯ РЕАЛИЗАЦИИ',
|
||||
'ЗАПОЛНЕНИЕ БД из файлов',
|
||||
'ВЫХОД'
|
||||
]
|
||||
|
||||
while True:
|
||||
choice = simple_choice(menu_options)
|
||||
|
||||
if choice == 'КАРТОЧКИ ПАРТНЕРОВ (сортировка по рейтингу)':
|
||||
print("\n🎯 КАРТОЧКИ ПАРТНЕРОВ (сортировка по рейтингу)")
|
||||
print("═" * 70)
|
||||
partners = get_partners(sort_by='rating', ascending=False)
|
||||
print_partner_cards(partners)
|
||||
|
||||
elif choice == 'КАРТОЧКИ ПАРТНЕРОВ (сортировка по скидкам)':
|
||||
print("\n📊 КАРТОЧКИ ПАРТНЕРОВ (сортировка по скидкам)")
|
||||
print("═" * 70)
|
||||
partners = get_partners(sort_by='discount', ascending=False)
|
||||
print_partner_cards(partners)
|
||||
|
||||
elif choice == 'КАРТОЧКИ ПАРТНЕРОВ (сортировка по ИНН)':
|
||||
print("\n🔢 КАРТОЧКИ ПАРТНЕРОВ (сортировка по ИНН)")
|
||||
print("═" * 70)
|
||||
partners = get_partners(sort_by='inn', ascending=True)
|
||||
print_partner_cards(partners)
|
||||
|
||||
elif choice == 'ИСТОРИЯ РЕАЛИЗАЦИИ':
|
||||
print("\n📈 ИСТОРИЯ РЕАЛИЗАЦИИ ПРОДУКЦИИ")
|
||||
print("═" * 70)
|
||||
sales = get_sales_history(limit=50)
|
||||
print_sales_cards(sales)
|
||||
|
||||
elif choice == 'ЗАПОЛНЕНИЕ БД из файлов':
|
||||
if os.path.exists('partners.db'):
|
||||
os.remove('partners.db')
|
||||
print("🗑️ База данных partners.db удалена")
|
||||
else:
|
||||
print("ℹ️ База данных partners.db не найдена, создаем новую")
|
||||
|
||||
conn = sqlite3.connect('partners.db')
|
||||
create_tables(conn)
|
||||
conn.close()
|
||||
load_data_from_files()
|
||||
|
||||
elif choice == 'ВЫХОД':
|
||||
print("\n👋 До свидания!")
|
||||
break
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
display_menu()
|
||||
|
|
|
|||
Loading…
Reference in a new issue