Скачиваний:
5
Добавлен:
29.03.2025
Размер:
53.77 Кб
Скачать
# -*- coding: utf-8 -*-

import sqlite3

path = "C:/Users/admin/source/repos/UNIVERSITY_repos/3 course/6 sem/mispris/2/furniture.db"
# path = "/home/user/repos/mispris/2/furniture.db"

conn = sqlite3.connect(path)
cursor = conn.cursor()
print("\n\nDB has been created/connected\n")

conn.execute("PRAGMA foreign_keys = ON")

# def create_category_table():
# cursor.execute('''CREATE TABLE IF NOT EXISTS category
# (id INTEGER PRIMARY KEY AUTOINCREMENT,
# id_parent INTEGER,
# terminal_classifier_name TEXT UNIQUE,
# FOREIGN KEY (id_parent) REFERENCES category(id))''')
# conn.commit()


def create_product_table():
cursor.execute('''
CREATE TABLE IF NOT EXISTS product (
id INTEGER PRIMARY KEY AUTOINCREMENT,
id_category INTEGER,
product_name TEXT NOT NULL,
quantity INTEGER DEFAULT 0,
price REAL DEFAULT 0,
unit_value REAL,
FOREIGN KEY (id_category) REFERENCES terminal_classifier(id)
)
''')
conn.commit()


########################## second lab ##########################

def create_terminal_classifier_table():
cursor.execute('''CREATE TABLE IF NOT EXISTS terminal_classifier
(id INTEGER PRIMARY KEY AUTOINCREMENT,
id_parent INTEGER,
terminal_classifier_name TEXT NOT NULL,
terminal_classifier_short_name TEXT NOT NULL,
id_unit INTEGER,
FOREIGN KEY (id_unit) REFERENCES unit(id),
FOREIGN KEY (id_parent) REFERENCES terminal_classifier(id))''')
conn.commit()


def create_enum_table():
cursor.execute('''CREATE TABLE IF NOT EXISTS enum
(id INTEGER PRIMARY KEY AUTOINCREMENT,
enum_id INTEGER,
priority INTEGER NOT NULL DEFAULT 0,
enum_name TEXT NOT NULL,
enum_short_name TEXT NOT NULL,
enum_description TEXT,
real_value REAL,
int_value INTEGER,
pic_value TEXT,
FOREIGN KEY (enum_id) REFERENCES terminal_classifier(id))''')
conn.commit()


def create_unit_table():
cursor.execute('''CREATE TABLE IF NOT EXISTS unit
(id INTEGER PRIMARY KEY AUTOINCREMENT,
unit_name TEXT UNIQUE,
unit_name_short TEXT UNIQUE)''')
conn.commit()



def creating_tables():
# create_category_table()
create_product_table()
create_terminal_classifier_table()
create_enum_table()
create_unit_table()

creating_tables()



def delete_data(tables_order):
try:
for table_name in tables_order:
cursor.execute(f"DELETE FROM {table_name};")

conn.commit()
print("Данные успешно удалены")
except sqlite3.Error as _ex:
print(f"Произошла ошибка при удалении данных: {_ex}")


def dropAllTables():
try:
# Список таблиц в порядке, в котором их нужно удалить (сначала те, которые не содержат внешних ключей)
tables_order = [ 'product', 'enum', 'terminal_classifier', 'unit']

delete_data(tables_order)

for table_name in tables_order:
cursor.execute(f"DROP TABLE IF EXISTS {table_name};")

conn.commit()
print("Таблицы успешно удалены")

except Exception as _ex:
print(f"Произошла ошибка при удалении таблиц: {_ex}")


def is_category_exists(terminal_classifier_name):
cursor.execute("SELECT COUNT(*) FROM terminal_classifier WHERE terminal_classifier_name = ?", (terminal_classifier_name,))
count = cursor.fetchone()[0]
res = count > 0

return res


def is_product_exist(product_name):
cursor.execute("SELECT COUNT(*) FROM product WHERE product_name = ?", (product_name,))
count = cursor.fetchone()[0]
res = count > 0

return res

def is_unit_exist(unit_name_short):
cursor.execute("SELECT COUNT(*) FROM unit WHERE unit_name_short = ?", (unit_name_short,))
count = cursor.fetchone()[0]
res = count > 0

return res


def add_unit(unit_name, unit_name_short):
try:
if not is_unit_exist(unit_name_short):
cursor.execute('''
INSERT INTO unit (unit_name, unit_name_short)
VALUES (?, ?)
''', (unit_name, unit_name_short))
conn.commit()

print('Единица измерения успешно добавлена.')
else:
print('Единица измерения с таким названием уже существует.')

except Exception as _ex:
print(f'Error {_ex} - произошла ошибка при добавлении единицы измерения')


def add_category(terminal_classifier_name, terminal_classifier_short_name, id_parent=None, id_unit=None):
# try:
if is_category_exists(terminal_classifier_name):
print(f'Категория с названием {terminal_classifier_name} уже существует.')
return
if id_unit:
cursor.execute("SELECT COUNT(*) FROM unit WHERE id = ?", (id_unit,))
unit_check_exist = cursor.fetchone()[0]
if not unit_check_exist:
print(f'Указанного id единицы измерения не существует!')
return

cursor.execute('''
INSERT INTO terminal_classifier (terminal_classifier_name, terminal_classifier_short_name, id_parent, id_unit)
VALUES (?, ?, ?, ?)
''', (terminal_classifier_name, terminal_classifier_short_name, id_parent, id_unit))

conn.commit()
print('Категория успешно добавлена.')
# except Exception as _ex:
# print(f'Error {_ex} - ошибка при добавлении категории.')


def add_subcategory(terminal_classifier_name, subterminal_classifier_name, subterminal_classifier_short_name, id_unit=None):
try:

if is_category_exists(terminal_classifier_name):
# Получаем Id категории
cursor.execute("SELECT id FROM terminal_classifier WHERE terminal_classifier_name = ?", (terminal_classifier_name,))
id_parent_to_set = cursor.fetchone()[0]

add_category(subterminal_classifier_name, subterminal_classifier_short_name, id_parent_to_set, id_unit)
conn.commit()
print("Подкатегория успешно добавлена.")

else:
print("Указанная категория/подкатегория не существует.")
except Exception as _ex:
print(f"Произошла ошибка при добавлении подкатегории: {_ex}")


def edit_category(id, terminal_classifier_name_new, terminal_classifier_short_name_new, id_unit=None):
try:
cursor.execute('SELECT COUNT(*) FROM terminal_classifier WHERE id = ?', (id,))
if cursor.fetchone()[0] == 0:
raise ValueError(f"Запись с id={id} не найдена в таблице terminal_classifier.")

if id_unit is not None:
cursor.execute('SELECT COUNT(*) FROM unit WHERE id = ?', (id_unit,))
if cursor.fetchone()[0] == 0:
raise ValueError(f"Запись с id_unit={id_unit} не найдена в таблице unit.")

cursor.execute('SELECT COUNT(*) FROM terminal_classifier WHERE terminal_classifier_name = ? AND id != ?', (terminal_classifier_name_new, id))
if cursor.fetchone()[0] > 0:
raise ValueError(f"Новое полное имя '{terminal_classifier_name_new}' уже занято.")

cursor.execute('SELECT COUNT(*) FROM terminal_classifier WHERE terminal_classifier_short_name = ? AND id != ?', (terminal_classifier_short_name_new, id))
if cursor.fetchone()[0] > 0:
raise ValueError(f"Новое короткое имя '{terminal_classifier_short_name_new}' уже занято.")

cursor.execute('''
UPDATE terminal_classifier
SET terminal_classifier_name = ?,
terminal_classifier_short_name = ?,
id_unit = ?
WHERE id = ?
''', (terminal_classifier_name_new, terminal_classifier_short_name_new, id_unit, id))

# Подтверждаем изменения
conn.commit()

except Exception as _ex:
print(f"Произошла ошибка при изменении категории: {_ex}")


def add_product(product_name, terminal_classifier_name, quantity, price, unit_value ):
try:
if is_category_exists(terminal_classifier_name):

# Получаем id подкатегории по ее названию
cursor.execute("SELECT id FROM terminal_classifier WHERE terminal_classifier_name = ?", (terminal_classifier_name,))
category_id = cursor.fetchone()[0]

# Добавляем новое изделие в таблицу Изделие
cursor.execute("INSERT INTO product (id_category, product_name, quantity, price, unit_value) VALUES (?, ?, ?, ?, ?)",
(category_id, product_name, quantity, price, unit_value))
conn.commit()
print("Изделие успешно добавлено.")

else:
print("Указанная подкатегория не существует.")

except Exception as _ex:
print(f"Произошла ошибка при добавлении изделия: {_ex}")


def get_all_categories():
try:
cursor.execute("SELECT * FROM terminal_classifier")
categories_names = cursor.fetchall()

return categories_names

except Exception as _ex:
print(f"Произошла ошибка при получении категорий: {_ex}")
return None


def get_all_products():
try:
cursor.execute("SELECT * FROM product")
products_names = cursor.fetchall()

return products_names

except Exception as _ex:
print(f"Произошла ошибка при получении товаров: {_ex}")
return None


def product_update_price(product_name, new_price):
try:

if is_product_exist(product_name):
cursor.execute("UPDATE product SET price = ? WHERE product_name = ?", (new_price, product_name))
conn.commit()
print('Цена изделия успешно обновлена.')
else:
print('Указанное изделие не существует.')

except Exception as _ex:
print(f"Произошла ошибка при обновлении цены изделия: {_ex}")


def product_update_quantity(product_name, new_quantity):
try:

if is_product_exist(product_name):
cursor.execute("UPDATE product SET quantity = ? WHERE product_name = ?", (new_quantity, product_name))
conn.commit()
print('Количество изделия успешно обновлено.')
else:
print('Указанное изделие не существует.')

except Exception as _ex:
print(f"Произошла ошибка при обновлении цены изделия: {_ex}")


def change_product_category(product_name, new_category):
try:

if is_category_exists(new_category):
# Получаем идентификатор новой категории
cursor.execute("SELECT id FROM terminal_classifier WHERE terminal_classifier_name = ?", (new_category,))
new_category_id = cursor.fetchone()[0]

# Обновляем категорию у изделия
cursor.execute("UPDATE product SET id_category = ? WHERE product_name = ?", (new_category_id, product_name))
conn.commit()
print('Категория изделия успешно изменена.')
else:
print('Указанная новая подкатегория не существует.')

except Exception as _ex:
print(f"Произошла ошибка при изменении подкатегории изделия: {_ex}")





def change_product_unit_value(product_name, unit_value):
try:
if is_product_exist(product_name):
cursor.execute("UPDATE product SET unit_value = ? WHERE product_name = ?", (unit_value, product_name))
conn.commit()
print('Значение единицы измерения изделия успешно изменено.')

else:
print('Указанное изделие не существует.')

except Exception as _ex:
print(f"Произошла ошибка при изменении значения единицы измерения изделия: {_ex}")


def delete_product_unit_value(product_name):
try:
if is_product_exist(product_name):
# cursor.execute("SELECT id_unit FROM product WHERE product_name = ?", (product_name,))
# delete_unit_id = cursor.fetchone()[0]

cursor.execute("UPDATE product SET unit_value = ? WHERE product_name = ?", (None, product_name))
conn.commit()
print('Значение единицы измерения изделия успешно удалено.')

else:
print('Указанное изделие не существует.')

except Exception as _ex:
print(f"Произошла ошибка при удалении значения единицы измерения изделия: {_ex}")



def is_enum_exist(terminal_classifier_name):
try:
cursor.execute("SELECT id FROM terminal_classifier WHERE terminal_classifier_name = ?", (terminal_classifier_name,))
check_if_exist = cursor.fetchone()

return check_if_exist

except Exception as _ex:
print(f"Произошла ошибка при проверке перечисления на существование: {_ex}")


def is_enum_pos_exist(enum_name):
try:
cursor.execute("SELECT id FROM enum WHERE enum_name = ?", (enum_name,))
check_if_exist = cursor.fetchone()

if check_if_exist:
return True
else:
return False

except Exception as _ex:
print(f"Произошла ошибка при проверке перечисления на существование: {_ex}")



def add_enum_pos(enum_id, enum_name, enum_short_name, enum_description=None, real_value=None, int_value=None, pic_value=None, priority=None):

try:
# Check if enum already exists in terminal_classifier
if is_enum_pos_exist(enum_name):
print("Указанное перечисление уже существует.")
return

# Get next priority if not provided
if priority is None:
cursor.execute("SELECT COALESCE(MAX(priority), 0) + 1 FROM enum", ())
priority = cursor.fetchone()[0]

# UPDATE via id
cursor.execute("INSERT INTO enum (enum_id, enum_name, enum_short_name, enum_description, real_value, int_value, pic_value, priority) VALUES (?, ?, ?, ?, ?, ?, ?, ?)",
(enum_id, enum_name, enum_short_name, enum_description, real_value, int_value, pic_value, priority))

conn.commit()

print('Перечисление успешно добавлено.')

except Exception as _ex:
print(f"Произошла ошибка при добавлении перечисления: {_ex}")


def delete_enum_classification(terminal_classifier_name):
try:

if is_enum_exist(terminal_classifier_name):
# Получаем идентификатор удаляемого класса перечислений
cursor.execute("SELECT id FROM terminal_classifier WHERE terminal_classifier_name = ?", (terminal_classifier_name,))
delete_id = cursor.fetchone()[0]

# Перемещаем все классы перечисления удаляемого класса перечисления в класс-перечисление "неопределенные"
cursor.execute("UPDATE terminal_classifier SET id_parent = (SELECT id FROM terminal_classifier WHERE terminal_classifier_name = 'неопределенные') WHERE id_parent = ?", (delete_id,))

# Перемещаем все перечисления удаляемой категории в категорию "неопределенные"
cursor.execute("UPDATE enum SET enum_id = (SELECT id FROM terminal_classifier WHERE terminal_classifier_name = 'неопределенные') WHERE enum_id = ?", (delete_id,))

# Удаляем класс-перечисление
cursor.execute("DELETE FROM terminal_classifier WHERE id = ?", (delete_id,))
conn.commit()

print('Класс перечисления успешно удален.')
else:
print('Указанный класс перечисления не существует.')

except Exception as _ex:
print(f"Произошла ошибка при удалении класса перечисления: {_ex}")


def delete_enum_pos(enum_name):
try:

# Удаление перечисления
cursor.execute("DELETE FROM enum WHERE enum_name = ?", (enum_name,))
rows_deleted = cursor.rowcount

if rows_deleted > 0:
print('Перечисление успешно удалено.')
else:
print('Перечисление с указанным названием не найдено.')

conn.commit()

except Exception as _ex:
print(f"Произошла ошибка при удалении перечисления: {_ex}")



def change_enum_pos(id, new_enum_name, new_enum_short_name, new_enum_description=None, new_real_value=None, new_int_value=None, new_pic_value=None):

try:

cursor.execute("SELECT enum_name FROM enum WHERE id = ?", (id,))
old_name_to_check = cursor.fetchone()[0]

if is_enum_pos_exist(old_name_to_check):

cursor.execute("UPDATE enum SET enum_name = ?, enum_short_name = ?, enum_description = ?, real_value = ?, int_value = ?, pic_value = ? WHERE id = ?",
(new_enum_name, new_enum_short_name, new_enum_description, new_real_value, new_int_value, new_pic_value, id))
conn.commit()

print('Перечисление изменено.')

else:
print('Указаное перечисление не найдено.')
return

except Exception as _ex:
print(f"Произошла ошибка при изменении перечисления: {_ex}")


def edit_enum_pos_priority(enum_pos_id, new_priority):
try:
# Получаем enum_id для указанного enum_pos_id
cursor.execute("SELECT enum_id FROM enum WHERE id = ?", (enum_pos_id,))
result = cursor.fetchone()
if result is None:
print('Указанное enum_pos_id не найдено.')
return None
enum_id = result[0]

# Проверяем, существует ли уже запись с таким priority для данного enum_id
cursor.execute("""
SELECT count(*)
FROM enum
WHERE priority = ?
AND enum_id = ?
""", (new_priority, enum_id))
count = cursor.fetchone()[0]

if count == 0:
# Обновляем priority
cursor.execute("""
UPDATE enum
SET priority = ?
WHERE id = ?
""", (new_priority, enum_pos_id))
else:
# Смещаем приоритеты и обновляем нужную запись
cursor.execute("""
UPDATE enum
SET priority = priority + 1
WHERE priority >= ?
AND enum_id = ?
""", (new_priority, enum_id))

cursor.execute("""
UPDATE enum
SET priority = ?
WHERE id = ?
""", (new_priority, enum_pos_id))

conn.commit()

print("Приоритет перечисления изменен")
return new_priority



except Exception as _ex:
print(f"Произошла ошибка при изменении приоритета перечисления: {_ex}")


def find_enums(enum_class_id):
try:

cursor.execute("""
SELECT ep.id, ep.enum_name, ep.enum_short_name, ep.enum_id, ep.priority, ep.real_value, ep.int_value, ep.pic_value
FROM enum ep
WHERE ep.enum_id = ?
ORDER BY ep.priority ASC
""", (enum_class_id,))

results = cursor.fetchall()

for row in results:
print(row)

return results

except Exception as _ex:
print(f"Произошла ошибка при получении списка перечислений заданного класса: {_ex}")


def add_enum_in_classifier(terminal_classifier_name, terminal_classifier_short_name, enum_parent_name, id_unit=None):
# try:
cursor.execute("SELECT id FROM terminal_classifier WHERE terminal_classifier_name = ?", (enum_parent_name,))
id_parent_check = cursor.fetchone()
if id_parent_check is not None:
id_parent = id_parent_check[0]
else:
id_parent = None
if id_unit:
cursor.execute("SELECT COUNT(*) FROM unit WHERE id = ?", (id_unit,))
count = cursor.fetchone()[0]
if (not count):
print(f'Произошла ошибка при добавлении записи в enum_classifier! Указанного id_unit не существует.')
cursor.execute('''
INSERT INTO terminal_classifier (terminal_classifier_name, terminal_classifier_short_name, id_parent, id_unit)
VALUES (?, ?, ?, ?)
''', (terminal_classifier_name, terminal_classifier_short_name, id_parent, id_unit))
conn.commit()

print('Класс перечисления успешно добавлен в основную базу.')


# except Exception as _ex:
# print(f'Error {_ex} - ошибка при добавлении enum_in_classifier.')


def change_enum_priority(enum_name, priority):
try:
if (priority >= 0):
cursor.execute("UPDATE enum SET priority = ? WHERE enum_name = ?", (priority, enum_name))
conn.commit()
else:
print('Приоритет должен быть неотрицательным целым числом.')
return

except Exception as _ex:
print(f'Error {_ex} - ошибка при изменении приоритета конкретного перечисления.')


def check_cyclic_dependency_terminal_classifier(new_parent_terminal_classifier_id, terminal_classifier_id):
try:
def fetch_parent_terminal_classifier(enum_id):
cursor.execute("SELECT id_parent FROM terminal_classifier WHERE id = ?", (enum_id,))
return cursor.fetchone()[0]

current_id = new_parent_terminal_classifier_id
while current_id is not None:
if current_id == terminal_classifier_id:
return True
current_id = fetch_parent_terminal_classifier(current_id)
return False
except Exception as _ex:
print(f'Error {_ex} - ошибка при проверке классов перечисления на цикл.')


def change_parent_of_terminal_classifier(terminal_classifier_name, new_parent_terminal_classifier=None):
try:
print(f'введенные категории - для {terminal_classifier_name} назначить родителем {new_parent_terminal_classifier}')

if (is_enum_exist(terminal_classifier_name)):
cursor.execute("SELECT id FROM terminal_classifier WHERE terminal_classifier_name = ?", (terminal_classifier_name,))
id_enum = cursor.fetchone()[0]

if (new_parent_terminal_classifier is not None):
if (is_enum_exist(new_parent_terminal_classifier)):

if (terminal_classifier_name == new_parent_terminal_classifier):
print('Нельзя назначить классу самого себя в родители.')
return

cursor.execute("SELECT id FROM terminal_classifier WHERE terminal_classifier_name = ?", (new_parent_terminal_classifier,))
id_new_parent_terminal_classifier = cursor.fetchone()[0]

cursor.execute("SELECT id_parent FROM terminal_classifier WHERE terminal_classifier_name = ?", (terminal_classifier_name,))
id_parent_class_of_class = cursor.fetchone()[0]

# Check for cyclic dependency using the recursive function
if check_cyclic_dependency_terminal_classifier(id_new_parent_terminal_classifier, id_enum):

if (id_parent_class_of_class):
# if parent class has had their own parent then would set it

cursor.execute("UPDATE terminal_classifier SET id_parent = ? WHERE id = ?", (id_new_parent_terminal_classifier, id_enum))

conn.commit()

cursor.execute("UPDATE terminal_classifier SET id_parent = ? WHERE id = ?", (id_parent_class_of_class, id_new_parent_terminal_classifier ))

conn.commit()

print('Родитель у подкласса был успешно изменен.')

else:
# if not then class becomes root
cursor.execute("UPDATE terminal_classifier SET id_parent = ? WHERE terminal_classifier_name = ?", (None, new_parent_terminal_classifier))

conn.commit()
print(f'Подкласс c terminal_classifier_name - {new_parent_terminal_classifier} стала родительской.')

# Update the parent class
cursor.execute("UPDATE terminal_classifier SET id_parent = ? WHERE id = ?", (id_new_parent_terminal_classifier, id_enum ))

conn.commit()

else:
print('Новая родительский класс не существует.')

else:
# code for new_parent_class=None
cursor.execute("UPDATE terminal_classifier SET id_parent = ? WHERE id = ?", (None, id_enum))
conn.commit()

print('Класс стал корневым / одним из корневых')

else:
print('Указанного класса не существует.')

except Exception as _ex:
conn.rollback()
print(f"Ошибка в изменении родителя класса (terminal_classifier): {_ex}")










def delete_product(product_name):
try:

# Удаление изделия
cursor.execute("DELETE FROM product WHERE product_name = ?", (product_name,))
rows_deleted = cursor.rowcount

if rows_deleted > 0:
print('Изделие успешно удалено.')
else:
print('Изделие с указанным названием не найдено.')

conn.commit()

except Exception as _ex:
print(f"Произошла ошибка при удалении изделия: {_ex}")


def delete_category(terminal_classifier_name):
try:

if is_category_exists(terminal_classifier_name):
# Получаем идентификатор удаляемой категории
cursor.execute("SELECT id FROM terminal_classifier WHERE terminal_classifier_name = ?;", (terminal_classifier_name,))
delete_id = cursor.fetchone()[0]

# Перемещаем все подкатегории удаляемой категории в категорию "неопределенные"
cursor.execute("UPDATE category SET id_parent = (SELECT id FROM terminal_classifier WHERE terminal_classifier_name = 'неопределенные') WHERE id_parent = ?", (delete_id,))

# Перемещаем все изделия удаляемой категории в категорию "неопределенные"
cursor.execute("UPDATE product SET id_category = (SELECT id FROM terminal_classifier WHERE terminal_classifier_name = 'неопределенные') WHERE id_category = ?", (delete_id,))

# Удаляем категорию
cursor.execute("DELETE FROM terminal_classifier WHERE id = ?", (delete_id,))
conn.commit()

print('Категория успешно удалена.')
else:
print('Указанная категория не существует.')

except Exception as _ex:
print(f"Произошла ошибка при удалении категории: {_ex}")



def delete_subcategory(subterminal_classifier_name):
try:

if is_category_exists(subterminal_classifier_name):
# Получаем идентификатор удаляемой подкатегории
cursor.execute("SELECT id FROM terminal_classifier WHERE terminal_classifier_name = ?", (subterminal_classifier_name,))
subcategory_id = cursor.fetchone()[0]

# Перемещаем связанные изделия в категорию "неопределенные"
cursor.execute("UPDATE product SET id_category = (SELECT id FROM terminal_classifier WHERE terminal_classifier_name = 'неопределенные') WHERE id_category = ?", (subcategory_id,))

# Перемещаем связанные подкатегории в категорию "неопределенные"
cursor.execute("UPDATE category SET id_parent = (SELECT id FROM terminal_classifier WHERE terminal_classifier_name = 'неопределенные') WHERE id_parent = ?", (subcategory_id,))

# Удаляем подкатегорию
cursor.execute("DELETE FROM terminal_classifier WHERE id = ?", (subcategory_id,))
conn.commit()

print('Подкатегория успешно удалена.')
else:
print('Указанная подкатегория не существует.')

except Exception as _ex:
print(f"Произошла ошибка при удалении подкатегории: {_ex}")


def get_subcategory_parent(subterminal_classifier_name):
try:

if is_category_exists(subterminal_classifier_name):
cursor.execute("SELECT id_parent FROM terminal_classifier WHERE terminal_classifier_name = ?", (subterminal_classifier_name,))
id_parent_category_of_subcategory = cursor.fetchone()[0]

cursor.execute("SELECT terminal_classifier_name FROM terminal_classifier WHERE id = ?", (id_parent_category_of_subcategory,))
parent_terminal_classifier_name = cursor.fetchone()[0]

return parent_terminal_classifier_name
else:
print('Указанной подкатегории не существует!')

except Exception as _ex:
print(f"Произошла ошибка при получении родительской категории подкатегории: {_ex}")


def check_cyclic_dependency(new_parent_id, category_id):
def fetch_parent_category(cat_id):
cursor.execute("SELECT id_parent FROM terminal_classifier WHERE id = ?", (cat_id,))
return cursor.fetchone()[0]

current_id = new_parent_id
while current_id is not None:
if current_id == category_id:
return True
current_id = fetch_parent_category(current_id)
return False


def change_parent_of_category(terminal_classifier_name, new_parent_category=None):
try:
print(f'введенные категории - для {terminal_classifier_name} назначить родителем {new_parent_category}')

if (is_category_exists(terminal_classifier_name)):
cursor.execute("SELECT id FROM terminal_classifier WHERE terminal_classifier_name = ?", (terminal_classifier_name,))
id_category = cursor.fetchone()[0]

if (new_parent_category is not None):
if (is_category_exists(new_parent_category)):

if (terminal_classifier_name == new_parent_category):
print('Нельзя назначить категории саму себя в родители.')
return

cursor.execute("SELECT id FROM terminal_classifier WHERE terminal_classifier_name = ?", (new_parent_category,))
id_new_parent_category = cursor.fetchone()[0]

cursor.execute("SELECT id_parent FROM terminal_classifier WHERE terminal_classifier_name = ?", (terminal_classifier_name,))
id_parent_category_of_category = cursor.fetchone()[0]

# Check for cyclic dependency using the recursive function
if check_cyclic_dependency(id_new_parent_category, id_category):

if (id_parent_category_of_category):
# if parent category has had their own parent then would set it

cursor.execute("UPDATE category SET id_parent = ? WHERE id = ?", (id_new_parent_category, id_category))

conn.commit()

cursor.execute("UPDATE category SET id_parent = ? WHERE id = ?", (id_parent_category_of_category, id_new_parent_category ))

conn.commit()

print('Родитель у подкатегории был успешно изменен.')

else:
# if not then category becomes root
cursor.execute("UPDATE category SET id_parent = ? WHERE terminal_classifier_name = ?", (None, new_parent_category))

conn.commit()
print(f'Подкатегория c terminal_classifier_name {new_parent_category} стала родительской.')

# Update the parent category
cursor.execute("UPDATE category SET id_parent = ? WHERE id = ?", (id_new_parent_category, id_category ))

conn.commit()

else:
print('Новая родительская категория не существует.')

else:
# code for new_parent_category=None
cursor.execute("UPDATE category SET id_parent = ? WHERE id = ?", (None, id_category))
conn.commit()

print('Категория стала корневой / одной из корневых')

else:
print('Указанной категории не существует.')

except Exception as _ex:
conn.rollback()
print(f"Ошибка в изменении родителя категории: {_ex}")


def change_terminal_classifier_name(old_name, new_name):
try:

if is_category_exists(old_name):
if not is_category_exists(new_name):
cursor.execute("SELECT id FROM terminal_classifier WHERE terminal_classifier_name = ?", (old_name,))
terminal_classifier_name_id = cursor.fetchone()[0]

cursor.execute("UPDATE category SET terminal_classifier_name = ? WHERE id = ?", (new_name, terminal_classifier_name_id))
print('Имя категории успешно изменено.')
else:
print('Указанное новое имя категории уже существует!')

else:
print('Указанная категория не существует.')

conn.commit()

except Exception as _ex:
print(f"Произошла ошибка при изменении имени категории: {_ex}")


def change_product_name(old_name, new_name):
try:

if is_product_exist(old_name):
if not is_product_exist(new_name):
cursor.execute("SELECT id FROM product WHERE product_name = ?", (old_name,))
product_name_id = cursor.fetchone()[0]

cursor.execute("UPDATE product SET product_name = ? WHERE id = ?", (new_name, product_name_id))
print('Имя изделия успешно изменено.')
else:
print('Указанное новое имя изделия уже существует!')

else:
print('Указанное изделие не существует.')

conn.commit()

except Exception as _ex:
print(f"Произошла ошибка при изменении имени изделия: {_ex}")


def get_category_parents(terminal_classifier_name):

if is_category_exists(terminal_classifier_name):
terminal_classifier_name_copy = terminal_classifier_name
try:

cursor.execute("SELECT id FROM terminal_classifier WHERE terminal_classifier_name = ?", (terminal_classifier_name,))
category_id = cursor.fetchone()[0]

# function for recursive search parents
def find_parents(category_id, parents_list):
id_parent = 0

cursor.execute("SELECT EXISTS(SELECT 1 FROM terminal_classifier WHERE id = ?)", (category_id,))
record_exists = cursor.fetchone()[0]

if record_exists:
cursor.execute("SELECT id_parent FROM terminal_classifier WHERE id = ?", (category_id,))
id_parent = cursor.fetchone()[0]

if id_parent == 0:
return

if category_id != 0 and id_parent != 0:
if id_parent != 0:
cursor.execute("SELECT terminal_classifier_name FROM terminal_classifier WHERE id = ?", (id_parent,))
parent_name = cursor.fetchone()
parents_list.append(parent_name)
find_parents(id_parent, parents_list)

parents_list = []
parents_list_res = []

# begin recursive search for parents
find_parents(category_id, parents_list)

for elem in parents_list:
if elem:
parents_list_res.append(elem[0])

if parents_list_res:
print(f'родительские элементы для категории "{terminal_classifier_name_copy}": \n {parents_list_res}')
else:
print(f'У категории "{terminal_classifier_name_copy}" нет родительских элементов')

except Exception as _ex:
print(f"Произошла ошибка при получении списка родителей категории: {_ex}")

else:
print('Категории не существует.')


def print_category_children(terminal_classifier_name, order):

if not is_category_exists(terminal_classifier_name):
print("Такой категории нет.")
return

children_list_res = get_category_children(terminal_classifier_name)
if children_list_res:
if order == 'прямой':
print(f'дочерние элементы для категории "{terminal_classifier_name}": \n {children_list_res}')
elif order == 'обратный':
print(f'дочерние элементы для категории "{terminal_classifier_name}": \n {children_list_res[::-1]}')
else:
print('Такого порядка вывода нет!')
else:
print(f'У категории "{terminal_classifier_name}" нет дочерних элементов')


def get_category_children(terminal_classifier_name):

if is_category_exists(terminal_classifier_name):
try:

cursor.execute("SELECT id FROM terminal_classifier WHERE terminal_classifier_name = ?", (terminal_classifier_name,))
category_id = cursor.fetchone()[0]

# function for recursive search subcategories
def find_children(category_id, children_list):
childrens_id = 0
childrens_id = []

cursor.execute("SELECT COUNT(*) FROM terminal_classifier WHERE id_parent = ?", (category_id,))
record_exists = cursor.fetchall()

if record_exists:
cursor.execute("SELECT id FROM terminal_classifier WHERE id_parent = ?", (category_id,))
childrens_id = cursor.fetchall()

if childrens_id == 0:
return

if category_id != 0 and childrens_id != 0:
for elem in childrens_id:
elem_id = elem[0]

cursor.execute("SELECT terminal_classifier_name FROM terminal_classifier WHERE id = ?", (elem_id,))
child_name = cursor.fetchone()
children_list.append(child_name)

find_children(elem_id, children_list)

children_list = []
children_list_res = []

# begin search subcategories
find_children(category_id, children_list)

for elem in children_list:
if elem:
children_list_res.append(elem[0])

return children_list_res

except Exception as _ex:
print(f"Произошла ошибка при получении списка дочерних категорий: {_ex}")

else:
print('Категории не существует.')



############### added after first review ###############

def print_products_of_category(terminal_classifier_name, order):
products = get_products_of_category(terminal_classifier_name)
if products:
print(f'Изделия для категории "{terminal_classifier_name}":')
print_str = ''

if order == 'прямой':
for prod in products:
print_str += '"' + prod + '"' + ' '
print(print_str)

elif order == 'обратный':
products = products[::-1]
for prod in products:
print_str += '"' + prod + '"' + ' '
print(print_str)
else:
print('Такого порядка вывода нет!')

else:
print(f'У категории "{terminal_classifier_name}" нет изделий')


def get_products_of_category(terminal_classifier_name):
if is_category_exists(terminal_classifier_name):
try:
categories_to_check = get_category_children(terminal_classifier_name)
products = []

if categories_to_check:
for category in categories_to_check:
cursor.execute("SELECT id FROM terminal_classifier WHERE terminal_classifier_name = ?", (category,))
id_of_searched_category = cursor.fetchone()[0]

cursor.execute("SELECT id FROM product WHERE id_category = ?", (id_of_searched_category,))
products_id = cursor.fetchall()


for id in products_id:
cursor.execute("SELECT product_name FROM product WHERE id = ?", (id[0],))
product_to_add = cursor.fetchone()[0]
products.append(product_to_add)


else:
cursor.execute("SELECT id FROM terminal_classifier WHERE terminal_classifier_name = ?", (terminal_classifier_name,))
id_of_searched_category = cursor.fetchone()[0]

cursor.execute("SELECT id FROM product WHERE id_category = ?", (id_of_searched_category,))
products_id = cursor.fetchall()

for id in products_id:
cursor.execute("SELECT product_name FROM product WHERE id = ?", (id[0],))
product_to_add = cursor.fetchone()[0]
products.append(product_to_add)

return products

except Exception as _ex:
print(f"Произошла ошибка при поиске изделия заданной категории: {_ex}")
else:
print('Категории не существует.')





def fill_data_example():

add_unit('килограмм', 'кг')
add_unit('штука', 'шт')
add_unit('метр', 'м')
add_unit('милиметр', 'мм')
add_unit('метр-квадратный', 'м2')
add_unit('метр-кубический', 'м3')
add_unit('секунда', 'с')
add_unit('ватт', 'вт')
add_unit('метры-в-секунду', 'м/с')

add_category("категория неопределенное", 'ктгр нпрдлн')
add_category("категория изделия", 'ктгр издл')

##
add_subcategory("категория изделия", "категория столы", 'ктгр стлы', 2)

add_subcategory("категория столы", "категория столы-металл", 'ктгр стлы-мтл',2)
add_product("стол_arizone", "категория столы-металл", 110, 2500, None)
add_product("стол_garden_story", "категория столы-металл", 54, 5665, None)
#
add_subcategory("категория столы", "категория столы-дерево", 'ктгр стлы-дрв', 2)
add_product("стол_эстер", "категория столы-дерево", 124, 6460, None)
add_product("стол_кентуки", "категория столы-дерево", 1340, 2400, None)

##
add_subcategory("категория изделия", "категория стулья", 'ктгр стля', 2)

add_subcategory("категория стулья", "категория стулья-пластик", 'ктгр стля-плстк', 2)
add_product("стул_rambo", "категория стулья-пластик", 435, 1580, None)
add_product("стул_keter", "категория стулья-пластик", 252, 1300, None)
#
add_subcategory("категория стулья", "категория стулья-металл", 'ктгр стля-мтл', 2)
add_product("стул_arizone", "категория стулья-пластик", 245, 3400, None)
add_product("стул_giardino", "категория стулья-пластик", 25, 2400, None)

##
add_subcategory("категория изделия", "категория кресла", 'ктгр крсл', 2)

add_subcategory("категория кресла", "категория кресла-кресло_кокон", 'ктгр крсл-ккн', 2)
add_product("кресло-m-group", "категория кресла-кресло_кокон", 24, 12000, None)
add_product("кресло-гамак", "категория кресла-кресло_кокон", 14, 15500, None)
#
add_subcategory("категория кресла", "категория кресла-двухместное_кресло", 'ктгр крсл-двмстн', 2)
add_product("кресло-gemini_promob", "категория кресла-двухместное_кресло", 4, 8999, None)
add_product("кресло-парящая_кровать", "категория кресла-двухместное_кресло", 54, 9999, None)



add_enum_in_classifier('перечисление неопределенное', 'п-нпрдлн', None, None)
add_enum_in_classifier('перечисление перечисление', 'п-enum', None, None)
add_enum_in_classifier('перечисление чисел', 'п-чсл', 'перечисление', None)

add_enum_in_classifier('перечисление целых-чисел', 'п-ц-чсл', 'перечисление чисел', None)
add_enum_in_classifier('перечисление вещественных-чисел', 'п-в-чсл', 'перечисление чисел', None)

add_enum_in_classifier('перечисление букв','п-бкв', 'перечисление перечисление', None)
add_enum_in_classifier('перечисление строк','п-стр', 'перечисление букв', None)

add_enum_in_classifier('перечисление размер', 'п-рзм', 'перечисление вещественных чисел', None)

add_enum_in_classifier('перечисление материал', 'п-мтр', 'перечисление строк', None)

add_enum_in_classifier('перечисление ширина', 'п-шрн', 'перечисление размер', 4) #10
add_enum_in_classifier('перечисление высота', 'п-вст', 'перечисление размер', 4) #11
add_enum_in_classifier('перечисление длина', 'п-длн', 'перечисление размер', 4) #12


add_enum_pos(10, 'перечисление ширина ножки стула', 'п-шрн-нжк-стл', 'ширина ножки стула дачного', 50, None, None, 2)
add_enum_pos(11, 'перечисление высота ножки стула', 'п-вст-нжк-стл', 'ширина ножки стула дачного', 500, None, None, 2)
add_enum_pos(12, 'перечисление длина ножки стула', 'п-длн-нжк-стл', 'ширина ножки стула дачного', 50, None, None, 2)




# dropAllTables()

# fill_data_example()


# add_enum_pos(10, 'ширина стул_rambo', 'шрн с_rambo', 'стул дачный', 0.555, None, None, 2, None)

# delete_enum_pos('ширина стул_rambo')

# change_enum_pos(2, 'ширина стул_rambo new', 'шрн с_rambo new', 'стул дачный new', 0.9, None, None, 2)

# print(is_enum_pos_exist('ширина стул_rambo'))

# edit_enum_pos_priority(3, 1)

# find_enums(10)

# change_enum_priority('ширина стул_arizone', 2)

# print(is_enum_pos_exist('длина'))

# delete_enum_pos('ширина')

# change_enum_pos(2, 'высота_new', 'вст_new')

# change_enum_priority('длина', 1)


# change_parent_of_terminal_classifier('материал')


# edit_enum_pos_priority(5, 0)

# add_enum_in_classifier('неопределенное', 'н', None)
# add_enum_in_classifier('перечисление', 'enum', None)
# add_enum_in_classifier('перечисление чисел', 'п-чсл', 'перечисление')


# add_enum_pos(1, 'оттенок', 'отнк', enum_description=None, real_value=None, int_value=None, pic_value=None, id_unit=None, priority=None)



# change_enum_pos(2, 'test for ottenok 2', 'test_otnk_2', new_enum_description=None, new_real_value=None, new_int_value=None, new_pic_value=None, new_id_unit=2)

# edit_enum_pos_priority(2, 0)


# change_parent_of_terminal_classifier(terminal_classifier_name)

# change_parent_of_terminal_classifier(terminal_classifier_name, new_parent_terminal_classifier=None)




conn.close()
Соседние файлы в папке 2