This commit is contained in:
parent 8145700735
commit 15fc17f6b9
2 changed files with 369 additions and 241 deletions

594
main.py
View file

@ -1,59 +1,56 @@
import sqlite3 import sqlite3
# import datetime
import pandas as pd import pandas as pd
import os import os
import sys import sys
# alina2 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")
cursor = conn.cursor()
cursor.executescript(''' def create_tables(db_connection):
CREATE TABLE IF NOT EXISTS PartnerType( """Создает все необходимые таблицы в базе данных"""
id INTEGER PRIMARY KEY AUTOINCREMENT, cursor = db_connection.cursor()
type_name VARCHAR(100) NOT NULL UNIQUE);
CREATE TABLE IF NOT EXISTS Partner( cursor.executescript('''
id INTEGER PRIMARY KEY AUTOINCREMENT, CREATE TABLE IF NOT EXISTS PartnerType (
partner_type_id INTEGER REFERENCES PartnerType(id), id INTEGER PRIMARY KEY AUTOINCREMENT,
name VARCHAR(255) NOT NULL, type_name VARCHAR(100) NOT NULL UNIQUE
director VARCHAR(255), );
email VARCHAR(255),
phone VARCHAR(50), CREATE TABLE IF NOT EXISTS Partner (
legal_address TEXT, id INTEGER PRIMARY KEY AUTOINCREMENT,
inn VARCHAR(20) UNIQUE, partner_type_id INTEGER REFERENCES PartnerType(id),
rating DECIMAL(3,2) DEFAULT 0.0, name VARCHAR(255) NOT NULL,
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP director VARCHAR(255),
email VARCHAR(255),
phone VARCHAR(50),
legal_address TEXT,
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, 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),
product_name VARCHAR(255) NOT NULL, product_name VARCHAR(255) NOT NULL,
article VARCHAR(100) UNIQUE NOT NULL, article VARCHAR(100) UNIQUE NOT NULL,
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),
partner_id INTEGER REFERENCES Partner(id), partner_id INTEGER REFERENCES Partner(id),
quantity INTEGER NOT NULL CHECK (quantity > 0), quantity INTEGER NOT NULL CHECK (quantity > 0),
sale_date DATE NIT NULL, sale_date DATE NOT NULL,
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);
@ -63,171 +60,28 @@ CREATE TABLE IF NOT EXISTS PartnerType(
CREATE INDEX IF NOT EXISTS idx_sale_product ON Sale(product_id); 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);
''') ''')
db_connection.commit()
print("✅ Таблицы успешно созданы")
partner_types = [ 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:
cursor.executemany('INSERT OR IGNORE INTO PartnerType (type_name) VALUES(?)', partner_types) return 5, '5% (10000-50000 ед.)'
else:
partners = [ return 0, '0% (до 10000 ед.)'
(1, 'База Строитель', 'Иванова Александра Ивановна', 'aleksandraivanova@ml.ru',
'493 123 45 67', 'lspci -nn 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, 'МонтажПро', 'Cтепанов Степан Сергеевич', 'stepananov@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 = [
('Паркетная доска Ясень темный однополосная 14 мм', 'База Строитель', 15500, '2023-03-23'),
('Ламинат Дуб дымчато-белый 33 класс 12 мм', 'База Строитель', 12350, '2023-12-18'),
('Ламинат Дуб серый 32 класс 8 мм с фаской', 'База Строитель', 37400, '2024-06-07'),
('Инженерная доска Дуб Французкая елка однополосная 12 мм', 'Паркет 29', 3500, '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):
cursor.execute('SELECT ID FROM Product WHERE product_name = ?',
(product_name,))
product_id = cursor.fetchone()[0]
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
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 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 print_partner_cards(partners_df):
""" Функция для вывода партнеров в виде карточек с выравниванием"""
if partners_df is None or partners_df.empty:
print(" Нет данных о партнерах")
return
for index, row in partners_df.iterrows():
print("╔" + "═" * 78 + "╗")
print(f"║ {'КАРТОЧКА ПАРТНЕРА':^76} ║")
print("╠" + "═" * 78 + "╣")
# Используем разные кавычки для избежания конфликта
type_text = f"Тип: {row['Тип партнера']}"
name_text = f"Наименование: {row['Наименование партнера']}"
director_text = f"Директор: {row['Директор']}"
rating_text = f" Рейтинг: {row['Рейтинг']}"
email_text = f"Контакты: {row['Электронная почта']}"
phone_text = f"Телефон:{row['Телефон']}"
inn_text = f"ИНН: {row['ИНН']}"
address_text = f"Адрес: {row['Юридическое лицо']}"
print(f"║ {type_text:<77}║")
print(f"║ {name_text:<77}║")
print(f"║ {director_text:<77}║")
print(f"║ {rating_text:<77}║")
print(f"║ {'':<77}║")
print(f"║ {email_text:<77}║")
print(f"║ {phone_text:<77}║")
print(f"║ {'':<77}║")
print(f"║ {inn_text:<77}║")
print(f"║ {address_text:<77}║")
print("╚" + "=" * 78 + "╝")
print()
def get_partners(db_path='partners.db', sort_by='rating', ascending=False): def get_partners(db_path='partners.db', sort_by='rating', ascending=False):
"""Получает данные о партнерах с возможностью сортировки"""
try: try:
# Подключаемся к базе данных
conn = sqlite3.connect(db_path) conn = sqlite3.connect(db_path)
# SQL запрос для получения данных о партнерах
query = """ query = """
SELECT SELECT
pt.type_name AS "Тип партнера", pt.type_name AS "Тип партнера",
@ -235,33 +89,169 @@ def get_partners(db_path='partners.db', sort_by='rating', ascending=False):
p.director AS "Директор", p.director AS "Директор",
p.email AS "Электронная почта", p.email AS "Электронная почта",
p.phone AS "Телефон", p.phone AS "Телефон",
p.legal_address AS "Юридическое лицо", p.legal_address AS "Юридический адрес",
p.inn AS "ИНН", p.inn AS "ИНН",
p.rating AS "Рейтинг", p.rating AS "Рейтинг",
p.created_at AS "Дата создания" p.created_at AS "Дата создания",
COALESCE(SUM(s.quantity), 0) as total_quantity
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
GROUP BY p.id
""" """
# Добавляем сортировку # Добавляем сортировку
if sort_by == 'rating': if sort_by == 'discount':
query += "ORDER BY p.rating" query += " ORDER BY total_quantity"
elif sort_by == 'rating':
query += " ORDER BY p.rating"
elif sort_by == 'name': elif sort_by == 'name':
query += "ORDER BY pt.type_name, p.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" query += " ORDER BY p.rating"
query += " DESC" if not ascending else " ASC" 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: except sqlite3.Error as e:
print(f"Ошибка базы данных: {e}") print(f"❌ Ошибка базы данных: {e}")
return None
except Exception as e:
print(f"Общая ошибка:{e}")
return None return None
except Exception as e:
print(f"❌ Общая ошибка: {e}")
return None
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): def simple_choice(options):
"""Упрощенная функция выбора"""
print("\n" + "=" * 50) print("\n" + "=" * 50)
print("ВЫБЕРИТЕ ОПЦИЮ:") print("ВЫБЕРИТЕ ОПЦИЮ:")
print("=" * 50) print("=" * 50)
@ -275,47 +265,181 @@ def simple_choice(options):
choice_num = int(choice) choice_num = int(choice)
if 1 <= choice_num <= len(options): if 1 <= choice_num <= len(options):
return options[choice_num - 1] return options[choice_num - 1]
print(f"Пожалуйста, введите число от 1 до {len(options)}") print(f"❌ Пожалуйста, введите число от 1 до {len(options)}")
except (ValueError, EOFError): except (ValueError, EOFError):
print("Неверный ввод. Попробуйте еще раз.") print("❌ Неверный ввод. Попробуйте еще раз.")
except KeyboardInterrupt: except KeyboardInterrupt:
print("\nПрограмма прервана пользователем.") print("\n👋 Программа прервана пользователем.")
sys.exit(0) 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():
"""Загружает данные из Excel файлов в базу данных"""
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(): def display_menu():
"""Функция для отображения главного меню """ """Главное меню программы"""
menu_options = [ menu_options = [
'КАРТОЧКИ ПАРТНЕРОВ (сортировка по рейтингу)', 'КАРТОЧКИ ПАРТНЕРОВ (сортировка по рейтингу)',
'пункт дом', 'КАРТОЧКИ ПАРТНЕРОВ (сортировка по скидкам)',
'ВЫХОД' 'ИСТОРИЯ РЕАЛИЗАЦИИ',
'ЗАПОЛНЕНИЕ БД из файлов',
'ВЫХОД'
] ]
while True: while True:
choice = simple_choice(menu_options) choice = simple_choice(menu_options)
if choice == 'КАРТОЧКИ ПАРТНЕРОВ (сортировка по рейтингу)': if choice == 'КАРТОЧКИ ПАРТНЕРОВ (сортировка по рейтингу)':
print("\n КАРТОЧКИ ПАРТНЕРОВ (сортировка по рейтингу)") print("\n🎯 КАРТОЧКИ ПАРТНЕРОВ (сортировка по рейтингу)")
print("=" * 70) 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)
elif choice == 'пункт дом':
print("\nДомой") elif choice == 'КАРТОЧКИ ПАРТНЕРОВ (сортировка по скидкам)':
break print("\n📊 КАРТОЧКИ ПАРТНЕРОВ (сортировка по скидкам)")
print("═" * 70)
partners = get_partners(sort_by='discount', ascending=False)
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 == 'ВЫХОД': elif choice == 'ВЫХОД':
print("\nДо свидания!") print("\n👋 До свидания!")
break break
if __name__ == "__main__":
print("Данные успешно добавлены в базу данных!")
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} записей ")
conn.close() if __name__ == "__main__":
display_menu() display_menu()
# print("1. КАРТОЧКИ ПАРТНЕРОВ ( сортировка по рейтингу")
# print("=" * 70)
# partners = get_partners(sort_by='rating', ascending=False)
# print_partner_cards(partners)

4
read.txt Normal file
View file

@ -0,0 +1,4 @@
Файл с партнерами: Partners_import.xlsx
Файл с типами продукции: Product_type_import.xlsx
Файл с продуктами: Products_import.xlsx
Файл с продажами: Partner_products_import.xlsx