This commit is contained in:
user_3d 2025-11-06 09:58:42 +05:00
parent 197868c744
commit ff22a6bde0

39
main.py
View file

@ -2,10 +2,11 @@ import sqlite3
import pandas as pd import pandas as pd
import os import os
import sys import sys
#import openpyxl import openpyxl
def create_tables(db_connection): def create_tables(db_connection):
#Создает все необходимые таблицы в базе данных
cursor = db_connection.cursor() cursor = db_connection.cursor()
cursor.executescript(''' cursor.executescript('''
@ -97,7 +98,7 @@ def get_partners(db_path='partners.db', sort_by='rating', ascending=False):
GROUP BY p.id GROUP BY p.id
""" """
#Сортировку
if sort_by == 'discount': if sort_by == 'discount':
query += " ORDER BY total_quantity" query += " ORDER BY total_quantity"
elif sort_by == 'rating': elif sort_by == 'rating':
@ -118,7 +119,7 @@ def get_partners(db_path='partners.db', sort_by='rating', ascending=False):
df = pd.read_sql_query(query, conn) df = pd.read_sql_query(query, conn)
conn.close() conn.close()
#Колонки со скидкой
if not df.empty: 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)[0])
df['Текст_скидки'] = df['total_quantity'].apply(lambda x: calculate_discount(x)[1]) df['Текст_скидки'] = df['total_quantity'].apply(lambda x: calculate_discount(x)[1])
@ -134,6 +135,7 @@ def get_partners(db_path='partners.db', sort_by='rating', ascending=False):
def get_sales_history(db_path='partners.db', limit=50, sort_by='date_desc'): def get_sales_history(db_path='partners.db', limit=50, sort_by='date_desc'):
#Получаем историю реализации продукции с разными вариантами сортировки
try: try:
conn = sqlite3.connect(db_path) conn = sqlite3.connect(db_path)
query = """ query = """
@ -152,7 +154,7 @@ def get_sales_history(db_path='partners.db', limit=50, sort_by='date_desc'):
JOIN ProductType pt ON pr.product_type_id = pt.id JOIN ProductType pt ON pr.product_type_id = pt.id
""" """
#Сортировка по дате
if sort_by == 'date_desc': if sort_by == 'date_desc':
query += " ORDER BY s.sale_date DESC, s.created_at DESC" query += " ORDER BY s.sale_date DESC, s.created_at DESC"
elif sort_by == 'date_asc': elif sort_by == 'date_asc':
@ -174,6 +176,7 @@ def get_sales_history(db_path='partners.db', limit=50, sort_by='date_desc'):
return None 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("📭 Нет данных о партнерах")
@ -216,7 +219,7 @@ def print_partner_cards(partners_df):
def print_sales_cards(sales_df): def print_sales_cards(sales_df):
#Выводим карточку исории реализации
if sales_df is None or sales_df.empty: if sales_df is None or sales_df.empty:
print("📭 Нет данных о продажах") print("📭 Нет данных о продажах")
return return
@ -225,14 +228,14 @@ def print_sales_cards(sales_df):
print(f"║ {'ИСТОРИЯ РЕАЛИЗАЦИИ ПРОДУКЦИИ':^145} ║") print(f"║ {'ИСТОРИЯ РЕАЛИЗАЦИИ ПРОДУКЦИИ':^145} ║")
print("╠" + "═" * 147 + "╣") print("╠" + "═" * 147 + "╣")
#Заголовок таблицы
headers = ["Дата", "Партнер", "Продукция", "Артикул", "Кол-во", "Цена", "Сумма"] headers = ["Дата", "Партнер", "Продукция", "Артикул", "Кол-во", "Цена", "Сумма"]
print("║ {:<12} {:<18} {:<59} {:<3} {:>9} {:>8} {:>13} ║".format( print("║ {:<12} {:<18} {:<59} {:<3} {:>9} {:>8} {:>13} ║".format(
headers[0], headers[1], headers[2], headers[3], headers[4], headers[5], headers[6] headers[0], headers[1], headers[2], headers[3], headers[4], headers[5], headers[6]
)) ))
print("╠" + "═" * 147 + "╣") print("╠" + "═" * 147 + "╣")
#Данные таблицы
for index, row in sales_df.iterrows(): for index, row in sales_df.iterrows():
date = str(row['Дата продажи'])[:10] date = str(row['Дата продажи'])[:10]
partner = row['Партнер'] partner = row['Партнер']
@ -246,12 +249,12 @@ def print_sales_cards(sales_df):
print("╚" + "═" * 147 + "╝") print("╚" + "═" * 147 + "╝")
#Итоги
total_quantity = sales_df['Количество'].sum() total_quantity = sales_df['Количество'].sum()
total_amount = sales_df['Общая сумма'].sum() total_amount = sales_df['Общая сумма'].sum()
print(f"📊 ИТОГО: {len(sales_df)} продаж, {total_quantity:,} единиц, общая сумма: {total_amount:,.2f} руб.") 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)
@ -274,7 +277,7 @@ def simple_choice(options):
def check_database(): def check_database():
#Проверка файлов базы двнных
conn = sqlite3.connect('partners.db') conn = sqlite3.connect('partners.db')
try: try:
cursor = conn.cursor() cursor = conn.cursor()
@ -289,26 +292,26 @@ def check_database():
def load_data_from_files(): def load_data_from_files():
#Загружает файлы в базу данных
conn = sqlite3.connect('partners.db') conn = sqlite3.connect('partners.db')
cursor = conn.cursor() cursor = conn.cursor()
try: try:
#Запрос файлов
print("\n📁 Введите имена файлов:") print("\n📁 Введите имена файлов:")
partners_file = input("Файл с партнерами: ") partners_file = input("Файл с партнерами: ")
product_type_file = input("Файл с типами продукции: ") product_type_file = input("Файл с типами продукции: ")
products_file = input("Файл с продуктами: ") products_file = input("Файл с продуктами: ")
sales_file = input("Файл с продажами: ") sales_file = input("Файл с продажами: ")
# 1. Загрузка типов партнеров #Загрузка типов партнеров
print("\n1. 📊 Загрузка типов партнеров...") print("\n1. 📊 Загрузка типов партнеров...")
partners_df = pd.read_excel(partners_file) partners_df = pd.read_excel(partners_file)
partner_types = partners_df['Тип партнера'].unique() partner_types = partners_df['Тип партнера'].unique()
for partner_type in partner_types: for partner_type in partner_types:
cursor.execute("INSERT OR IGNORE INTO PartnerType (type_name) VALUES (?)", (partner_type,)) cursor.execute("INSERT OR IGNORE INTO PartnerType (type_name) VALUES (?)", (partner_type,))
# 2. Загрузка партнеров #Загрузка партнеров
print("2. 👥 Загрузка партнеров...") print("2. 👥 Загрузка партнеров...")
for _, row in partners_df.iterrows(): for _, row in partners_df.iterrows():
cursor.execute("SELECT id FROM PartnerType WHERE type_name = ?", (row['Тип партнера'],)) cursor.execute("SELECT id FROM PartnerType WHERE type_name = ?", (row['Тип партнера'],))
@ -322,7 +325,7 @@ def load_data_from_files():
row['Юридический адрес партнера'], row['ИНН'], row['Рейтинг']) row['Юридический адрес партнера'], row['ИНН'], row['Рейтинг'])
) )
# 3. Загрузка типов продукции #Загрузка типов продукции
print("3. 📦 Загрузка типов продукции...") print("3. 📦 Загрузка типов продукции...")
product_type_df = pd.read_excel(product_type_file) product_type_df = pd.read_excel(product_type_file)
for _, row in product_type_df.iterrows(): for _, row in product_type_df.iterrows():
@ -330,7 +333,7 @@ def load_data_from_files():
(row['Тип продукции'], row['Коэффициент типа продукции']) (row['Тип продукции'], row['Коэффициент типа продукции'])
) )
# 4. Загрузка продуктов #Загрузка продуктов
print("4. 🏷️ Загрузка продуктов...") print("4. 🏷️ Загрузка продуктов...")
products_df = pd.read_excel(products_file) products_df = pd.read_excel(products_file)
for _, row in products_df.iterrows(): for _, row in products_df.iterrows():
@ -344,7 +347,7 @@ def load_data_from_files():
row['Артикул'], row['Минимальная стоимость для партнера']) row['Артикул'], row['Минимальная стоимость для партнера'])
) )
# 5. Загрузка продаж #Загрузка продаж
print("5. 💰 Загрузка продаж...") print("5. 💰 Загрузка продаж...")
sales_df = pd.read_excel(sales_file) sales_df = pd.read_excel(sales_file)
loaded_sales = 0 loaded_sales = 0
@ -394,7 +397,7 @@ def load_data_from_files():
def display_menu(): def display_menu():
#Главное меню программы
menu_options = [ menu_options = [
'КАРТОЧКИ ПАРТНЕРОВ (сортировка по рейтингу)', 'КАРТОЧКИ ПАРТНЕРОВ (сортировка по рейтингу)',
'КАРТОЧКИ ПАРТНЕРОВ (сортировка по скидкам)', 'КАРТОЧКИ ПАРТНЕРОВ (сортировка по скидкам)',