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

import sqlite3

path = "C:/Users/admin/source/repos/UNIVERSITY_repos/3 course/6 sem/mispris/3/furniture.db"
# path = "/home/user/repos/mispris/3/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_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 create_terminal_classifier_table():
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS terminal_classifier (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            id_parent INTEGER CHECK (id_parent != id),
            terminal_classifier_name TEXT NOT NULL,
            terminal_classifier_short_name TEXT NOT NULL,
            id_unit INTEGER,
            priority INTEGER NOT NULL DEFAULT 0,
            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,
            enum_name TEXT NOT NULL,
            enum_short_name TEXT NOT NULL,
            enum_description TEXT,
            real_value REAL,
            int_value INTEGER,
            pic_value TEXT,
            priority INTEGER NOT NULL DEFAULT 0,
            FOREIGN KEY (enum_id) REFERENCES terminal_classifier(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()

def create_param_table():
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS param (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            param_name TEXT NOT NULL,
            param_short_name TEXT NOT NULL,
            unit_id INTEGER,
            min_val REAL,
            max_val REAL,
            FOREIGN KEY (unit_id) REFERENCES unit(id)
        )
    ''')
    conn.commit()

def create_class_param_table():
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS class_param (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            param_id INTEGER NOT NULL,
            class_param_id INTEGER,
            param_class_name TEXT NOT NULL,
            param_class_short_name TEXT NOT NULL,
            priority INTEGER NOT NULL,
            FOREIGN KEY (param_id) REFERENCES param(id),
            FOREIGN KEY (class_param_id) REFERENCES terminal_classifier(id) ON DELETE CASCADE
        )
    ''')
    conn.commit()

def create_prod_param_table():
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS prod_param (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            prod_id INTEGER NOT NULL,
            param_id INTEGER NOT NULL,
            int_value INTEGER,
            str_value TEXT,
            real_value REAL,
            enum_id INTEGER,
            UNIQUE(prod_id, param_id),
            FOREIGN KEY (enum_id) REFERENCES enum(id),
            FOREIGN KEY (prod_id) REFERENCES product(id),
            FOREIGN KEY (param_id) REFERENCES param(id)
        )
    ''')
    conn.commit()
    # было UNIQUE(prod_id, param_id),
    # было UNIQUE(prod_id, param_id, int_value, str_value, real_value, enum_id),
    # было UNIQUE(prod_id, param_id, enum_id),



# def create_agr_class_table():
#     cursor.execute('''
#         CREATE TABLE IF NOT EXISTS agr_class (
#             id INTEGER PRIMARY KEY AUTOINCREMENT,
#             agr_class_name TEXT NOT NULL
#         )
#     ''')
#     conn.commit()

# def create_agr_pos_table():
#     cursor.execute('''
#         CREATE TABLE IF NOT EXISTS agr_pos (
#             id INTEGER PRIMARY KEY AUTOINCREMENT,
#             agr_pos_name TEXT NOT NULL,
#             id_agr_class INTEGER NOT NULL,
#             id_prod INTEGER NOT NULL,
#             FOREIGN KEY (id_agr_class) REFERENCES agr_class(id),
#             FOREIGN KEY (id_prod) REFERENCES product(id)
#         )
#     ''')
#     conn.commit()

# def create_agr_pos_list_table():
#     cursor.execute('''
#         CREATE TABLE IF NOT EXISTS agr_pos_list (
#             id INTEGER PRIMARY KEY AUTOINCREMENT,
#             id_prod_param INTEGER NOT NULL,
#             id_agr_class INTEGER NOT NULL,
#             FOREIGN KEY (id_agr_class) REFERENCES agr_class(id),
#             FOREIGN KEY (id_prod_param) REFERENCES prod_param(prod_id)
#         )
#     ''')
#     conn.commit()

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

    create_param_table()
    create_class_param_table()
    create_prod_param_table()

    # create_agr_class_table()
    # create_agr_pos_table()
    # create_agr_pos_list_table()

creating_tables()



def delete_data(tables_order):
    try:
        for table_name in tables_order:
            cursor.execute(f"DELETE FROM {table_name};")
        conn.commit()
    except sqlite3.Error as ex:
        print(f"Произошла ошибка при удалении данных: {ex}")


def dropAllTables():
    try:
        tables_order = ['prod_param', 'class_param', 'product', 'enum', 'param','terminal_classifier', 'unit']

        # Удаляем данные из таблиц, которые ссылаются на другие таблицы
        delete_data(['prod_param', 'class_param'])


        tables_order2 = ['product', 'enum', 'param','terminal_classifier', 'unit']


        # Удаляем остальные данные
        delete_data(tables_order2)

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

        conn.commit()
        print("Данные успешно удалены")
        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(id):
            cursor.execute("SELECT id_parent FROM terminal_classifier WHERE id = ?", (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 0

                    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('Родитель у подкласса был успешно изменен.')
                            return 1

                        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} стала родительской.')
                            return 1

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

                    conn.commit()
                    return 1

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

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

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

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

    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 terminal_classifier 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 terminal_classifier SET id_parent = ? WHERE id = ?", (id_new_parent_category, id_category))

                            conn.commit()

                            cursor.execute("UPDATE terminal_classifier 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 terminal_classifier 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 terminal_classifier 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 terminal_classifier 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('Категории не существует.')














##################################### third lab #####################################


def create_param(new_name, new_short_name, new_unit_id, new_min_val, new_max_val):

    try:
        # Вставка новой записи в таблицу param
        cursor.execute('''SELECT unit_name FROM unit WHERE id = ?''', (new_unit_id,))
        is_unit_exist = cursor.fetchone()

        if not is_unit_exist:
            print(f'Ошибка в create_param: Единицы измерения не сушествует.')

        cursor.execute('''
            INSERT INTO param(param_name, param_short_name, unit_id, min_val, max_val)
            VALUES (?, ?, ?, ?, ?)
        ''', (new_name, new_short_name, new_unit_id, new_min_val, new_max_val))

        # Проверка количества вставленных строк
        if cursor.rowcount == 1:
            conn.commit()
            print(f'параметр с именем {new_name} успешно добавлен.')
            return 1
        else:
            conn.rollback()
            return 0
    except sqlite3.Error as e:
        # В случае ошибки откатываем транзакцию
        print(f"An error occurred in create_param: {e}")
        conn.rollback()
        return 0


def edit_param(param_id, new_name, new_short_name, new_unit_id, new_min_val, new_max_val):
    try:
        # Начало транзакции
        conn.execute('BEGIN TRANSACTION')

        cursor.execute('''SELECT param_name FROM param WHERE id = ? ''', (param_id,))
        is_param_exist = cursor.fetchone()

        if not is_param_exist:
            print(f'Ошибка в edit_param: Параметра с id = {param_id} не существует.')
            return 0


        cursor.execute('''SELECT unit_name FROM unit WHERE id = ? ''', (new_unit_id,))
        is_unit_exist = cursor.fetchone()

        if not is_unit_exist:
            print(f'Ошибка в edit_param: Единицы измерения с id = {new_unit_id} не существует.')
            return 0


        # Выполнение запроса на обновление
        cursor.execute('''
            UPDATE param
            SET param_name = ?, param_short_name = ?, unit_id = ?, min_val = ?, max_val = ?
            WHERE id = ?
        ''', (new_name, new_short_name, new_unit_id, new_min_val, new_max_val, param_id))

        conn.commit()

        # Получение количества обновленных строк
        res = cursor.rowcount

        # Фиксация изменений
        conn.commit()

        # Возвращение результата
        if res == 1:
            print(f'Параметр с id = {param_id} успешно обновлен.')
            return 1
        else:
            return 0

    except sqlite3.Error as e:
        print(f"Ошибка в edit_param: {e}")
        conn.rollback()
        return 0


def delete_param(param_to_delete):
    cursor.execute('''SELECT id FROM param WHERE param_name = ? ''', (param_to_delete,))
    param_id = cursor.fetchone()

    if not param_id:
        print(f'Ошибка. Параметра с таким названием не существует')
        return 0
    else:
        param_id = param_id[0]
        print(f'param_id - {param_id}')

    def del_param_from__class_param():
        cursor.execute('''SELECT id FROM class_param WHERE param_id = ? ''', (param_id,))
        classes_id_to_del = cursor.fetchall()
        print(f'classes_id_to_del - {classes_id_to_del}')

        for elem_id in classes_id_to_del:
            print(f'elem_id - {elem_id[0]}')

            cursor.execute("DELETE FROM class_param WHERE id = ?", (elem_id[0],))
            conn.commit()

        print(f'Классы параметров (если были) удаляемого параметра были обновлены')


    del_param_from__class_param()

    cursor.execute("DELETE FROM param WHERE param_name = ?", (param_to_delete,))

    res = cursor.rowcount

    conn.commit()

    # Возвращение результата в зависимости от количества удаленных строк
    if res == 1:
        print(f'Параметр успешно удален.')
        return 1
    else:
        print(f'При удалении параметра что-то пошло не так.')
        return 0




def create_class_param(terminal_classifier_name_class, terminal_classifier_name_short_class, param_name, parent_class_param_name):


    try:
        cursor.execute("SELECT id FROM param WHERE param_name = ?", (param_name,))

        param_id = cursor.fetchone()
        if not param_id:
            print(f'Ошибка в create_class_param: параметра с именем {param_name} не существует в таблице param')
            return 0
        else:
            param_id = param_id[0]

        cursor.execute("SELECT id FROM terminal_classifier WHERE terminal_classifier_name = ?", (parent_class_param_name,))

        parent_class_id = cursor.fetchone()
        if not parent_class_id:
            print(f'В create_class_param: класса параметра (родителя) с именем {parent_class_param_name} не существует в таблице terminal_classifier, будет вставлено значение None')
            parent_class_id = None
        else:
            parent_class_id = parent_class_id[0]


        # Получаем максимальное значение priority для данного class_id

        cursor.execute(''' SELECT COUNT(*) FROM class_param WHERE param_class_name = ? ''', (terminal_classifier_name_class,))
        res = cursor.fetchone()[0]

        if res is None:
            res = 0
        else:
            cursor.execute(''' SELECT MAX(priority) + 1 FROM class_param WHERE param_class_name = ? ''', (terminal_classifier_name_class,))
            res = cursor.fetchone()[0]

        if res is None:
            res = 0

        cursor.execute('''SELECT unit_id FROM param WHERE param_name = ? ''', (param_name,))
        unit_id = cursor.fetchone()
        if not unit_id:
            print(f'Единицы измерения с заданным {unit_id} не найдено. Будет вставлено NULL')
        else:
            unit_id = unit_id[0]

        cursor.execute('''
            INSERT INTO terminal_classifier(id_parent, terminal_classifier_name, terminal_classifier_short_name, id_unit, priority)
            VALUES (?, ?, ?, ?, ?)
        ''', (parent_class_id, terminal_classifier_name_class, terminal_classifier_name_short_class, unit_id, res))

        cursor.execute('''
            INSERT INTO class_param(param_class_name, param_class_short_name, class_param_id, param_id, priority)
            VALUES (?, ?, ?, ?, ?)
        ''', (terminal_classifier_name_class, terminal_classifier_name_short_class, parent_class_id, param_id, res))

        conn.commit()

    except sqlite3.Error as e:
        # В случае ошибки откатываем транзакцию
        print(f"An error occurred: {e}")
        conn.rollback()
        return 0





def edit_class_param(id, new_param_class_name, new_param_class_short_name, new_param_id=None, class_param_id=None):
    try:
        # Начало транзакции
        conn.execute('BEGIN TRANSACTION')

        cursor.execute('''SELECT param_class_name FROM class_param WHERE id = ? ''', (id,))
        is_class_param_exist = cursor.fetchone()

        if not is_class_param_exist:
            print(f'Ошибка в edit_class_param: Параметра класса с id = {id} не существует.')
            return 0
        else:
            param_class_name_old = is_class_param_exist[0]

            cursor.execute('''SELECT id FROM terminal_classifier WHERE terminal_classifier_name = ? ''', (param_class_name_old,))
            id_class_param_in_tc = cursor.fetchone()[0]

        if new_param_id:
            cursor.execute('''SELECT param_name FROM param WHERE id = ? ''', (new_param_id,))
            is_param_exist = cursor.fetchone()

            if not is_param_exist:
                print(f'Ошибка в edit_class_param: Параметра с id = {new_param_id} не существует.')
                return 0
        else:
            cursor.execute('''SELECT id FROM param WHERE param_name = ? ''', ('неопределенный параметр',))
            new_param_id = cursor.fetchone()[0]



        if class_param_id:
            cursor.execute('''SELECT terminal_classifier_name FROM terminal_classifier WHERE id = ? ''', (class_param_id,))
            is_terminal_class_name_exist = cursor.fetchone()

            if check_cyclic_dependency_terminal_classifier(class_param_id, param_class_name_old):
                print("Ошибка в edit_class_param: родитель и потомок образуют цикл. ")
                return 0

            if (is_terminal_class_name_exist[0] == param_class_name_old):
                print(f'_terminal_class_name_ = {is_terminal_class_name_exist[0]}, param_class_name_old = {param_class_name_old} =>', sep='')
                print(f'Нельзя назначить классу в родителя самого себя')
                return 0

            if not is_terminal_class_name_exist:
                    print(f'Ошибка в edit_class_param: Параметра класса с id = {id} не существует.')
                    return 0

            if change_parent_of_category(param_class_name_old, is_terminal_class_name_exist[0]):


                cursor.execute('''SELECT id FROM terminal_classifier WHERE terminal_classifier_name = ? ''', ('неопределенный параметр класса',))
                class_param_id = cursor.fetchone()[0]

                cursor.execute("UPDATE class_param SET param_class_name = ?, param_class_short_name = ?, param_id = ?, class_param_id = ? WHERE id = ?", (new_param_class_name, new_param_class_short_name, new_param_id, class_param_id, id))


                cursor.execute("UPDATE terminal_classifier SET terminal_classifier_name = ?, terminal_classifier_short_name = ?,  id_parent = ? WHERE id = ?", (new_param_class_name, new_param_class_short_name, class_param_id, id_class_param_in_tc))



                res = cursor.rowcount

                # Фиксация изменений
                conn.commit()

                # Возвращение результата
                if res == 1:
                    print(f'Параметр с id = {id} успешно обновлен.')
                    return 1
                else:
                    return 0

        else:
            cursor.execute('''SELECT id FROM terminal_classifier WHERE terminal_classifier_name = ? ''', ('неопределенный параметр класса',))
            class_param_id = cursor.fetchone()[0]

            cursor.execute("UPDATE class_param SET param_class_name = ?, param_class_short_name = ?, param_id = ?, class_param_id = ? WHERE id = ?", (new_param_class_name, new_param_class_short_name, new_param_id, class_param_id, id))


            cursor.execute("UPDATE terminal_classifier SET terminal_classifier_name = ?, terminal_classifier_short_name = ?,  id_parent = ? WHERE id = ?", (new_param_class_name, new_param_class_short_name, class_param_id, id_class_param_in_tc))

            res = cursor.rowcount

            # Фиксация изменений
            conn.commit()

            # Возвращение результата
            if res == 1:
                print(f'Параметр класса с id = {id} успешно обновлен.')
                return 1
            else:
                return 0


    except sqlite3.Error as e:
        print(f"Ошибка в edit_class_param: {e}")
        conn.rollback()
        return 0



def delete_class_param(class_to_delete):
    try:
        cursor.execute('''SELECT id FROM terminal_classifier WHERE terminal_classifier_name = ? ''', (class_to_delete,))
        exists = cursor.fetchone()

        if not exists:
            print(f'Ошибка в delete_class_param: Класса параметра с таким названием не существует')
            return 0

        if (class_to_delete == 'неопределенный параметр класса' or class_to_delete == 'параметр класса'):
            print(f"Нельзя удалить параметр класса \'неопределенный параметр класса\' или \'параметр класса\'")
            return 0

        # Проверка на наличие дочерних элементов
        is_children_exist = get_category_children(class_to_delete)

        print(f'is_children_exist - {is_children_exist}')

        if not is_children_exist:
            cursor.execute("DELETE FROM class_param WHERE param_class_name = ?", (class_to_delete,))
            conn.commit()

            cursor.execute("DELETE FROM terminal_classifier WHERE terminal_classifier_name = ?", (class_to_delete,))
            conn.commit()
        else:
            cursor.execute('''SELECT id FROM terminal_classifier WHERE terminal_classifier_name = ? ''', (class_to_delete,))
            id_class_to_delete = cursor.fetchone()[0]

            print(f'class_to_delete - {class_to_delete}')
            print(f'id_class_to_delete - {id_class_to_delete}')

            cursor.execute("SELECT id FROM class_param WHERE class_param_id = ? ", (id_class_to_delete,))
            children = cursor.fetchall()

            print(f'children (id) - {children}')

            cursor.execute('''SELECT id FROM terminal_classifier WHERE terminal_classifier_name = ? ''', ('неопределенный параметр класса',))
            undef_class_id = cursor.fetchone()[0]

            children_info_rows = []


            for child in children:
                print(f'child - {child}')

                cursor.execute('''SELECT param_class_name FROM class_param WHERE id = ? ''', (child[0],))
                child_name = cursor.fetchone()[0]
                print(f'child_name - {child_name}')

                cursor.execute('''SELECT * FROM class_param WHERE id = ?''', (child[0],))
                child_row = cursor.fetchone()[0]

                children_info_rows.append(child_row)



                cursor.execute("UPDATE class_param SET class_param_id = ? WHERE id = ?", (undef_class_id, child[0]))

                change_parent_of_terminal_classifier(child_name, 'неопределенный параметр класса')
                conn.commit()



                cursor.execute("DELETE FROM class_param WHERE param_class_name = ?", (class_to_delete,))
                conn.commit()

                cursor.execute("DELETE FROM terminal_classifier WHERE terminal_classifier_name = ?", (class_to_delete,))
                conn.commit()

            print(f'children_info_rows - {children_info_rows}')

        conn.commit()
        print("Удаление завершено успешно")
        return 1

    except sqlite3.IntegrityError as e:
        print(f"Ошибка в delete_class_param: {e}")
        conn.rollback()
        return 0





# inherit_class_parameters('длина_new', 'длн_new', 'длина')

def inherit_class_parameters(new_class_to_inherit__name, new_class_to_inherit__short_name, parent_class__name):
    try:
        # Получаем parent_id
        cursor.execute('''
            SELECT id FROM class_param WHERE param_class_name = ?
        ''', (parent_class__name,))
        parent_id_row = cursor.fetchone()

        if parent_id_row is None:
            print(f'Ошибка в inherit_class_parameters: класса параметра с именем для наследования не найдено')
            return 0  # Если id родителя в таблице class_param не найдено


        # Получаем параметры из CLASS_PARAM для parent_id
        cursor.execute('''
            SELECT class_param_id, param_id FROM class_param WHERE param_class_name = ? ORDER BY priority
        ''', (parent_class__name,))

        param_rows = cursor.fetchall()

        print(f'param_rows - {param_rows}')

        for param_row in param_rows:
            class_param_id, param_id = param_row

            # Проверяем, существует ли уже параметр для данного cl_id
            cursor.execute('''
                SELECT 1 FROM class_param WHERE param_class_name = ? AND param_id = ?
            ''', (new_class_to_inherit__name, param_id))

            exists = cursor.fetchone()

            cursor.execute('''
                SELECT param_name FROM param WHERE id = ?''', (param_id,))
            param_name = cursor.fetchone()[0]

            print(f'exists ? - {exists}')

            if exists:
                print(f'Класс параметра с именем {new_class_to_inherit__name} уже существует.')

            if not exists:

                create_class_param(new_class_to_inherit__name, new_class_to_inherit__short_name, param_name, parent_class__name)


        conn.commit()


    except sqlite3.Error as e:
        print(f"An error occurred in inherit_class_parameters: {e}")
        conn.rollback()
        return 0




def edit_class_param_output_priority(class_id, id, priority):
    try:
        # Выполнение запроса для получения количества записей с указанным priority и id класса
        cursor.execute("""
            SELECT count(*)
            FROM class_param
            WHERE priority = ? AND class_param_id = ?
        """, (priority, class_id))

        # Получение результата запроса
        cnt = cursor.fetchone()[0]

        if cnt == 0:
            # Если таких записей нет, обновляем priority для указанной записи
            cursor.execute("""
                UPDATE class_param
                SET priority = ?
                WHERE class_param_id = ? AND id = ?
            """, (priority, class_id, id))
        else:
            # Если такие записи есть, увеличиваем output_priority для всех записей с priority >= переданного в функцию priority
            cursor.execute("""
                UPDATE class_param
                SET priority = priority + 1
                WHERE priority >= ? AND class_param_id = ?
            """, (priority, class_id))

            # Затем обновляем priority для указанной записи
            cursor.execute("""
                UPDATE class_param
                SET priority = ?
                WHERE class_param_id = ? AND id = ?
            """, (priority, class_id, id))

        conn.commit()
        print(f'Приоритет изменен.')
        return priority

    except sqlite3.Error as e:
        print(f"An error occurred in edit_class_param_output_priority: {e}")
        conn.rollback()
        return 0





def create_prod_param(new_prod_id, new_param_id, new_int_val, new_str_val, new_real_val, new_enum_id):
    try:
        # проверка на существование new_prod_id
        cursor.execute("SELECT product_name FROM product WHERE id = ?", (new_prod_id,))
        is_new_product_name_exist = cursor.fetchone()

        print(f'new_prod_id - {new_prod_id}')
        print(f'is_new_product_name_exist - {is_new_product_name_exist}')

        if is_new_product_name_exist is None:
            print(f'Ошибка в create_prod_param: изделия с id {new_prod_id} не найдено')
            return 0


        # проверка на существование new_param_id
        cursor.execute('''
            SELECT param_name FROM param WHERE id = ?
        ''', (new_param_id,))
        is_new_param_name_exist = cursor.fetchone()

        if is_new_param_name_exist is None:
            print(f'Ошибка в create_prod_param: параметра с id {new_param_id} не найдено')
            return 0


        # проверка на существование new_enum_val
        cursor.execute('''
            SELECT enum_name FROM enum WHERE id = ?
        ''', (new_enum_id,))
        is_new_enum_id_exist = cursor.fetchone()

        if is_new_enum_id_exist is None:
            print(f'Ошибка в create_prod_param: перечисления с id {new_enum_id} не найдено')
            return 0

        if new_int_val:
            cursor.execute('''SELECT min_val, max_val FROM param WHERE id = ?''', (new_param_id,))
            limits = cursor.fetchall()[0]

            print(f'у new_int_val limits = {limits}')

            if not limits:
                min_val, max_val = None
            else:
                min_val, max_val = limits
                if not ((min_val <= new_int_val) and (new_int_val <= max_val)):
                    print(f'значение {new_int_val} не попадает в промежуток границ параметра [{min_val} ; {max_val}]')

        if new_real_val:
            cursor.execute('''SELECT min_val, max_val FROM param WHERE id = ?''', (new_param_id,))
            print(f'new_param_id = {new_param_id}')
            limits = cursor.fetchall()[0]

            print(f'у new_real_val limits = {limits}')

            if not limits:
                min_val, max_val = None
            else:
                min_val, max_val = limits
                print(f'min_val - {min_val} \n max_val - {max_val}')
                if not ((min_val <= new_real_val ) and (max_val >= new_real_val)):
                    print(f'значение {new_real_val} не попадает в промежуток границ параметра [{min_val} ; {max_val}]')

        # Выполнение запроса на вставку данных
        cursor.execute('''
            INSERT INTO prod_param (prod_id, param_id, int_value, str_value, real_value, enum_id)
            VALUES (?, ?, ?, ?, ?, ?)
        ''', (new_prod_id, new_param_id, new_int_val, new_str_val, new_real_val, new_enum_id,))

        # Получение количества вставленных строк
        res = cursor.rowcount

        # Фиксация изменений
        conn.commit()

        # Возвращение результата
        if res == 1:
            print(f'Добавлена запись в таблицу prod_param')
            return 1
        else:
            return 0

    except sqlite3.Error as e:
        print(f"Ошибка в create_prod_param: {e}")
        conn.rollback()
        return 0





def edit_prod_param(id, new_int_val, new_str_val, new_real_val, new_enum_val):
    try:
        # Начало транзакции
        conn.execute('BEGIN TRANSACTION')

        new_enum_val
        # проверка на существование new_enum_val
        cursor.execute('''
            SELECT id FROM enum WHERE enum_name = ?
        ''', (new_enum_val,))
        is_enum_exist = cursor.fetchall()

        print(f'is_enum_exist - {is_enum_exist}')

        if is_enum_exist is None:
            print(f'Ошибка в edit_prod_param: записи с new_enum_val {new_enum_val} в enum не найдено')
            return 0
        else:
            new_enum_id = is_enum_exist[0]


        # проверка на существование id
        cursor.execute('''
            SELECT prod_id, param_id FROM prod_param WHERE id = ?
        ''', (id,))
        is_id_exist = cursor.fetchall()

        print(f'is_id_exist - {is_id_exist}')

        if is_id_exist is None:
            print(f'Ошибка в edit_prod_param: записи с id {id} в prod_param не найдено')
            return 0



        print(f'enum_id - {new_enum_id[0]}')
        # Выполнение запроса на обновление данных
        cursor.execute('''
            UPDATE prod_param
            SET int_value = ?, str_value = ?, enum_id = ?, real_value = ?
            WHERE id = ?
        ''', (new_int_val, new_str_val, new_enum_id[0], new_real_val, id))

        # Получение количества обновленных строк
        res = cursor.rowcount

        # Фиксация изменений
        conn.commit()

        # Возвращение результата
        if res == 1:
            return 1
        else:
            return 0

    except sqlite3.Error as e:
        print(f"Ошибка в edit_prod_param: {e}")
        conn.rollback()
        return 0




def find_param_id_from_prod(prod_id):
    try:

        cursor.execute('SELECT product_name FROM product WHERE id = ?', (prod_id,))
        product_name = cursor.fetchone()
        if product_name:
            product_name = product_name[0]
        else:
            print(f'изделия с id - {prod_id} не существует.')
            return 0

        # Начало транзакции
        conn.execute('BEGIN TRANSACTION')

        # Получение class_id из таблицы obj
        cursor.execute('SELECT id_category FROM product WHERE id = ?', (prod_id,))
        id_parent = cursor.fetchone()
        print(f'id_parent = {id_parent}')

        if id_parent is None:
            print(f'Ошибка: объект с id {prod_id} не найден')
            return []

        id_parent = id_parent[0]

        # Получение param_id из таблицы class_param, упорядоченных по output_priority
        cursor.execute('''
            SELECT param_id FROM class_param
            WHERE class_param_id = ?
            ORDER BY priority
        ''', (id_parent,))

        param_ids = cursor.fetchall()

        # Фиксация изменений
        conn.commit()

        # Возвращение результата в виде списка
        return [row[0] for row in param_ids]

    except sqlite3.Error as e:
        print(f"Ошибка в find_param_id_from_prod: {e}")
        conn.rollback()
        return []


def find_products_of_class_param(id_of_class_param_el):

    cursor.execute('SELECT param_class_name FROM class_param WHERE id = ?', (id_of_class_param_el,))
    name_of_class_param_el = cursor.fetchone()

    print(f'id_of_class_param_el - {name_of_class_param_el}')

    if name_of_class_param_el:
        name_of_class_param_el = name_of_class_param_el[0]
    else:
        print(f'класса параметра с названием - {name_of_class_param_el} не существует.')
        return 0

    cursor.execute('SELECT class_param_id FROM class_param WHERE id = ?', (id_of_class_param_el,))
    class_param_id = cursor.fetchone()

    print(f'name_of_class_param_el - {name_of_class_param_el}')

    if class_param_id:
        class_param_id = class_param_id[0]
    else:
        print(f'класс параметра с id категории изделий - {class_param_id} не существует.')
        return 0

    cursor.execute('SELECT * FROM product WHERE id_category = ?', (class_param_id,))
    products = cursor.fetchall()

    if products:
        products = products[0]
    else:
        print(f'изделия с id классом параметра - {class_param_id} не существует.')
        return 0

    return products



def find_products_in_interval_param(param_id, type_value, start_val, end_val):

    if 'int' in type_value.lower():
        cursor.execute('SELECT prod_id FROM prod_param WHERE param_id = ? AND int_value >= ? AND int_value <= ?', (param_id, start_val, end_val))
        prod_ids = cursor.fetchall()

        if prod_ids:
            prod_ids = prod_ids
            return prod_ids
        else:
            print(f'изделий с param_id = {param_id}, type_value = {type_value}, start_val = {start_val}, end_val = {end_val} нет в таблице product')
            return

    if 'real' in type_value.lower():
        cursor.execute('SELECT prod_id FROM prod_param WHERE param_id = ? AND real_value >= ? AND real_value <= ?', (param_id, start_val, end_val))
        prod_ids = cursor.fetchall()

        if prod_ids:
            prod_ids = prod_ids
            return prod_ids
        else:
            print(f'изделий с param_id = {param_id}, type_value = {type_value}, start_val = {start_val}, end_val = {end_val} нет в таблице product')
            return

##################################### third lab #####################################








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_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, 'перечисление ширина ножки стула', 'п-шрн-нжк-стл', 'ширина ножки стула дачного', None, None, None, 1)
    add_enum_pos(11, 'перечисление высота ножки стула', 'п-вст-нжк-стл', 'ширина ножки стула дачного', None, None, None, 2)
    add_enum_pos(12, 'перечисление длина ножки стула', 'п-длн-нжк-стл', 'ширина ножки стула дачного', None, None, None, 3)

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







# dropAllTables()

# fill_data_example()




# create_param('неопределенный параметр', 'н-п', 10, None, None)

# create_param('длина', 'длн в мм', 2, 0, 2500)

# create_param('ширина', 'шрн в мм', 2, 0, 1500)


# create_class_param('неопределенный параметр класса', 'н-п-к', 'неопределенный параметр', None)

# create_class_param('параметр класса', 'п-к', 'неопределенный параметр', 'неопределенный параметр класса')

# create_class_param('длина изделия', 'длн', 'длина', 'категория столы-металл')
# create_class_param('ширина изделия', 'шрн', 'ширина', 'категория столы-металл')


# create_class_param('длина изделия', 'длн', 'длина', 'категория стулья-металл')
# create_class_param('ширина изделия', 'шрн', 'ширина', 'категория стулья-металл')



# inherit_class_parameters('ширина изделия наследованная', 'шрн нслдвн', 'ширина изделия')



# delete_class_param('ширина изделия наследованная')


# edit_class_param_output_priority(8, 6, 4)


# edit_param(2, 'dlina_new', 'dln_new', 4, 0, 5000)

# delete_param('dlina_new')



# edit_class_param(2, 'параметр класса_new', 'п-к_n', 3, None)

# delete_class_param('параметр класса_new')

# edit_class_param(2, 'параметр класса_new', 'п-к_n', 3, 25) # проверку на цикл самого в себя проходит


######### here #########



'''(function) def create_prod_param(
    new_prod_id: Any,
    new_param_id: Any,
    new_int_val: Any,
    new_str_val: Any,
    new_real_val: Any,
    new_enum_id: Any
) -> Any'''

# create_prod_param(1, 4, None, None, 70.0, 1) # стол arizone

# create_prod_param(7, 4, None, None, 45.0, 4) # стул arizone

# edit_prod_param(1, None, None, 80.0, 'перечисление длина ножки стола')


# prod_id = 1
# print(f'идентификаторы параметров (из таблицы «class_param») базового класса объекта таблицы «product» c product_id = {prod_id}', find_param_id_from_prod(prod_id))

# create_class_param('длина изделия', 'длн', 'длина', 'категория столы-металл')
# create_class_param('ширина изделия', 'шрн', 'ширина', 'категория столы-металл')

# print(f'изделия заданного класса параметра : {find_products_of_class_param(3)}')

# print(f"id изделий с заданными значениями: {find_products_in_interval_param(4, 'real', 0, 75)}")




#######################################################################################################################

# delete_class_param('ширина some изделия 2') # работает, тк нет детей

# delete_class_param('ширина some изделия') # работает

# edit_class_param_output_priority(25, 3, 0)


# edit_param(2, 'dlina_new', 'dln_new', 4, 0, 5000)


# edit_class_param(2, 'параметр класса_new', 'п-к_n', 3, None)

# edit_class_param(2, 'параметр класса_new', 'п-к_n', 3, 25) # проверку на цикл самого в себя проходит


# edit_class_param(1, 'неопределенный параметр класса_new', 'н-п-к_n', 3, 26) # вроде ворк (проверка на цикл - ребенок-родитель меняет устраняет цикл просто изменением порядка категорий (почти поворот дерева))



# create_prod_param(1, 3, None, None, 999, 3)

# create_prod_param(1, 3, None, None, 999, 3)
# create_prod_param(1, 3, None, None, 999, 3)


# edit_prod_param(1, 985, None, None, 'перечисление длина ножки стула')





# inherit_class_parameters('длина наследованная', 'длн_нслдвн', 'длина some изделия')

# inherit_class_parameters('длина глубочайшее наследование', 'длн_глбчш_нслдвн', 'длина наследованная')


# delete_class_param('длина глубочайшее наследование')


# delete_param('длина')










# 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)


################################################ previous sem  ################################################







################################################ 2.1  ################################################









################################################ 2.1  ################################################











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