This commit is contained in:
user_3d 2025-11-05 14:01:55 +05:00
parent 184f6c0412
commit fe9b3a1f5e

564
main.py
View file

@ -1,25 +1,19 @@
import sqlite3 import sqlite3
import datetime
import pandas as pd import pandas as pd
import os import os
import sys
import openpyxl
#Черепанова #Черепанова
if os.path.exists('partners.db'):
os.remove('partners.db')
print(f"База данных {'partners.db'} удалена")
else:
print(f"База данных {'partners.db'} не найдена, создаем новую")
conn = sqlite3.connect('partners.db') def create_tables(db_connection):
cursor = conn.cursor() cursor = db_connection.cursor()
# Создаем таблицы
cursor.executescript(''' cursor.executescript('''
CREATE TABLE IF NOT EXISTS PartnerType ( CREATE TABLE IF NOT EXISTS PartnerType (
id INTEGER PRIMARY KEY AUTOINCREMENT, 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, id INTEGER PRIMARY KEY AUTOINCREMENT,
@ -33,11 +27,13 @@ CREATE TABLE IF NOT EXISTS Partner (
rating DECIMAL(3,2) DEFAULT 0.0, rating DECIMAL(3,2) DEFAULT 0.0,
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
); );
CREATE TABLE IF NOT EXISTS ProductType ( CREATE TABLE IF NOT EXISTS ProductType (
id INTEGER PRIMARY KEY AUTOINCREMENT, id INTEGER PRIMARY KEY AUTOINCREMENT,
type_name VARCHAR(100) NOT NULL UNIQUE, type_name VARCHAR(100) NOT NULL UNIQUE,
coefficient DECIMAL(5,2) NOT NULL DEFAULT 1.0 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, id INTEGER PRIMARY KEY AUTOINCREMENT,
product_type_id INTEGER REFERENCES ProductType(id), product_type_id INTEGER REFERENCES ProductType(id),
@ -46,6 +42,7 @@ CREATE TABLE IF NOT EXISTS Product (
min_partner_price DECIMAL(15,2) NOT NULL, min_partner_price DECIMAL(15,2) NOT NULL,
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
); );
CREATE TABLE IF NOT EXISTS Sale ( CREATE TABLE IF NOT EXISTS Sale (
id INTEGER PRIMARY KEY AUTOINCREMENT, id INTEGER PRIMARY KEY AUTOINCREMENT,
product_id INTEGER REFERENCES Product(id), product_id INTEGER REFERENCES Product(id),
@ -55,6 +52,7 @@ CREATE TABLE IF NOT EXISTS Sale (
sale_price DECIMAL(15,2) NOT NULL, sale_price DECIMAL(15,2) NOT NULL,
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP 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_type ON Partner(partner_type_id);
CREATE INDEX IF NOT EXISTS idx_partner_inn ON Partner(inn); 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_type ON Product(product_type_id);
@ -63,170 +61,23 @@ 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_partner ON Sale(partner_id);
CREATE INDEX IF NOT EXISTS idx_sale_date ON Sale(sale_date); 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 = [ db_connection.commit()
(1, 'База Строитель', 'Иванова Александра Ивановна', 'aleksandraivanova@ml.ru', print("✅ Таблицы успешно созданы")
'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)
]
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')
]
def get_ids(product_name, partner_name): def calculate_discount(total_quantity):
cursor.execute('SELECT id FROM Product WHERE product_name = ?', (product_name,)) if total_quantity >= 300000:
product_id = cursor.fetchone()[0] return 15, '15% (более 300000 ед.)'
elif total_quantity >= 50000:
cursor.execute('SELECT id FROM Partner WHERE name = ?', (partner_name,)) return 10, '10% (50000-300000 ед.)'
partner_id = cursor.fetchone()[0] elif total_quantity >= 10000:
return 5, '5% (10000-50000 ед.)'
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
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))
except Exception as e:
print(f"Ошибка при добавлении продажи: {sale}, ошибка: {e}")
# Сохраняем изменения
conn.commit()
def get_partners(db_path='partners.db', sort_by= 'rating', ascending =False):
try:
# Подключаемся к базе данных
conn = sqlite3. connect (db_path)
# SQL запрос для получения данных о партнерах
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 "Дата создания"
FROM Partner p
JOIN PartnerType pt ON p.partner_type_id = pt.id
"""
if 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"
else: else:
query += " ORDER BY p.rating" return 0, '0% (до 10000 ед.)'
query += " DESC" if not ascending else " ASC"
df = pd.read_sql_query(query, conn)
# Закрываем соединение
conn.close()
return df
except sqlite3.Error as e:
print(f"Ошибка базы данных: {e}")
return None
except Exception as e:
print(f"Общая ошибка: {e}")
return None
def print_partners(partners_df):
"""ФУНКЦИЯ для красивого вывода данных о партнерах"""
if partners_df is None or partners_df.empty:
print("Нет данных о партнерах")
return
# Выводим данные в удобном формате
print("=" * 120)
print("{:<7} {:<20} {:<35} {:<20} {:<10}".format(
"Тип", "Наименование", "Директор","Телефон","Рейтинг"
))
print("=" * 120)
for _, row in partners_df.iterrows():
print("{:<7} {:<20} {:<35} {:<20} {:<10}".format(
row['Тип партнера'] [:7],
row['Наименование партнера'] [:20],
row['Директор'] [:35],
row['Телефон'][:20],
row['Рейтинг']
))
print("=" * 120)
print(f"Всего партнеров: {len(partners_df)}")
def get_partners_with_stats(db_path='partners.db'): def get_partners(db_path='partners.db', sort_by='rating', ascending=False):
"""Функция для получения партнеров со статистикой продаж"""
try: try:
conn = sqlite3.connect(db_path) conn = sqlite3.connect(db_path)
query = """ query = """
@ -234,89 +85,370 @@ def get_partners_with_stats(db_path='partners.db'):
pt.type_name AS "Тип партнера", pt.type_name AS "Тип партнера",
p.name AS "Наименование партнера", p.name AS "Наименование партнера",
p.director AS "Директор", p.director AS "Директор",
p.email AS "Электронная почта",
p.phone AS "Телефон",
p.legal_address AS "Юридический адрес",
p.inn AS "ИНН",
p.rating AS "Рейтинг", p.rating AS "Рейтинг",
COUNT (s.id) AS "Количество продаж", p.created_at AS "Дата создания",
SUM(COALESCE(s.quantity, 0)) AS "Общее количество товаров", COALESCE(SUM(s.quantity), 0) as total_quantity
SUM(COALESCE(s.quantity * s.sale_price, 0)) AS "Общий объем продаж",
p.created_at AS "Дата регистрации"
FROM Partner p FROM Partner p
JOIN PartnerType pt ON p.partner_type_id = pt.id JOIN PartnerType pt ON p.partner_type_id = pt.id
LEFT JOIN Sale s ON p.id = s.partner_id LEFT JOIN Sale s ON p.id = s.partner_id
GROUP BY p.id GROUP BY p.id
ORDER BY "Общий объем продаж" DESC""" """
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) df = pd.read_sql_query(query, conn)
conn.close() 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 return df
except sqlite3.Error as e:
print(f"❌ Ошибка базы данных: {e}")
return None
except Exception as e: except Exception as e:
print(f"Общая ошибка: {e}") print(f"❌ Общая ошибка: {e}")
return None return None
# print("╔" + "═" * 69 + "╗")
# print(f"║ {'КАРТОЧКА ПАРТНЕРА':^67} ║") def get_sales_history(db_path='partners.db', limit=50, sort_by='date_desc'):
# print("╠" + "═" * 69 + "╣") try:
# print("╚" + "═" * 69 + "╝") 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): def print_partner_cards(partners_df):
"""ФУНКЦИЯ для вывода партнеров в виде карточек с выравниванием"""
if partners_df is None or partners_df.empty: if partners_df is None or partners_df.empty:
print("Нет данных о партнерах") print("📭 Нет данных о партнерах")
return return
for index, row in partners_df.iterrows(): for index, row in partners_df.iterrows():
print("╔" + "═" * 70 + "╗") print("╔" + "═" * 69 + "╗")
print(f"║{'КАРТОЧКА ПАРТНЕРА':^70}║") print(f"║ {'КАРТОЧКА ПАРТНЕРА':^67} ║")
print("╠" + "═" * 70 + "╣") print("╠" + "═" * 69 + "╣")
# Используем разные кавычки для избежания конфликта
type_text = f"Тип: {row['Тип партнера']}" type_text = f"Тип: {row['Тип партнера']}"
name_text = f"Наименование: {row['Наименование партнера']}" name_text = f"Наименование: {row['Наименование партнера']}"
director_text = f"Директор: {row['Директор']}" director_text = f"Директор: {row['Директор']}"
rating_text = f"Рейтинг: {row['Рейтинг']}" rating_text = f"Рейтинг: {row['Рейтинг']}"
quantity_text = f"Продано единиц: {row['total_quantity']:,}"
discount_text = f"Скидка: {row['Текст_скидки']}"
email_text = f"Контакты: {row['Электронная почта']}" email_text = f"Контакты: {row['Электронная почта']}"
phone_text = f"Телефон: {row['Телефон']}" phone_text = f"Телефон: {row['Телефон']}"
inn_text = f"ИНН: {row['ИНН']}" inn_text = f"ИНН: {row['ИНН']}"
address_text = row['Юридический адрес']
print(f"║{type_text:<70}║") address = row['Юридический адрес']
print(f"║{name_text:<70}║") if len(address) > 60:
print(f"║{director_text:<70}║") address = address[:57] + "..."
print(f"║{rating_text:<70}║") address_text = f"Адрес: {address}"
print(f"║{'':<70}║")
print(f"║{email_text:<70}║") print(f"║ {type_text:<67} ║")
print(f"║{phone_text:<70}║") print(f"║ {name_text:<67} ║")
print(f"║{'':<70}║") print(f"║ {director_text:<67} ║")
print(f"║{inn_text:<70}║") print(f"║ {rating_text:<67} ║")
print(f"║{address_text:<70}║") print(f"║ {quantity_text:<67} ║")
print("╚" + "═" * 70 + "╝") 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() print()
if __name__ == "__main__":
# Проверяем данные def print_sales_cards(sales_df):
print("Данные успешно добавлены в базу данных!")
print("\nКоличество записей в таблицах:") 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'] tables = ['PartnerType', 'Partner', 'ProductType', 'Product', 'Sale']
for table in tables: for table in tables:
cursor.execute(f'SELECT COUNT(*) FROM {table}') cursor.execute(f"SELECT COUNT(*) FROM {table}")
count = cursor.fetchone()[0] count = cursor.fetchone()[0]
print(f" {table}: {count} записей") print(f" {table}: {count} записей")
finally:
conn.close() conn.close()
print("1. КАРТОЧКИ ПАРТНЕРОВ (сортировка по рейтингу)")
print("=" * 70) 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("Файл с продажами: ")
# 1. Загрузка типов партнеров
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,))
# 2. Загрузка партнеров
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['Рейтинг'])
)
# 3. Загрузка типов продукции
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['Коэффициент типа продукции'])
)
# 4. Загрузка продуктов
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['Минимальная стоимость для партнера'])
)
# 5. Загрузка продаж
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) partners = get_partners(sort_by='rating', ascending=False)
print_partner_cards(partners) print_partner_cards(partners)
# print("\n3. Партнеры со статистикой продаж:") elif choice == 'КАРТОЧКИ ПАРТНЕРОВ (сортировка по скидкам)':
# partners_stats = get_partners_with_stats() print("\n📊 КАРТОЧКИ ПАРТНЕРОВ (сортировка по скидкам)")
# print("═" * 70)
# if partners_stats is not None: partners = get_partners(sort_by='discount', ascending=False)
# print(partners_stats.to_string(index=False)) print_partner_cards(partners)
# elif choice == 'КАРТОЧКИ ПАРТНЕРОВ (сортировка по ИНН)':
# print("\n4. Полные данные о партнерах:") print("\n🔢 КАРТОЧКИ ПАРТНЕРОВ (сортировка по ИНН)")
# # Выводим полную информацию в табличном виде print("═" * 70)
# full_data = get_partners() partners = get_partners(sort_by='inn', ascending=True)
# if full_data is not None: print_partner_cards(partners)
# print(full_data.to_string(index=False))
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()