diff --git a/main.py b/main.py index a0524cc..64418e3 100644 --- a/main.py +++ b/main.py @@ -1,13 +1,15 @@ import sqlite3 - -import datetime +# import datetime import pandas as pd - -#ghfdl -# jhbjhg +import os -#fg hgfjhgfhfgh vfv +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() @@ -163,4 +165,51 @@ for sale in sales_data: print(f"Ошибка при добавлении продажи: {sale}, ошибка: {e}") conn.commit() + +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['Рейтинг']}" + 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"║ {'':<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() + +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()