From e3f0c4973ae218dc96638741c2bfe6f46c2be40a Mon Sep 17 00:00:00 2001
From: Kirill <kremizov3@gmail.com>
Date: Sat, 29 Nov 2025 14:34:21 +0300
Subject: [PATCH] feat(DB): Added utils file for better initializing db

---
 db/DBDataUtils.py | 113 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 113 insertions(+)
 create mode 100644 db/DBDataUtils.py

diff --git a/db/DBDataUtils.py b/db/DBDataUtils.py
new file mode 100644
index 0000000..a35b598
--- /dev/null
+++ b/db/DBDataUtils.py
@@ -0,0 +1,113 @@
+import re
+from loguru import logger as log
+
+def add_category(conn, name, description=None, parent_id=None):
+    cursor = conn.cursor()
+    
+    try:
+        cursor.execute('''
+            INSERT OR IGNORE INTO categories (name, description, parent_id)
+            VALUES (?, ?, ?)
+        ''', (name, description, parent_id))
+
+        cursor.execute('SELECT id FROM categories WHERE name = ?', (name,))
+        result = cursor.fetchone()
+        category_id = result[0] if result else None
+        
+        return category_id
+    except sqlite3.Error as e:
+        log.error(f"Ошибка при добавлении категории: {e}")
+        return None
+
+def add_filter(conn, name, filter_key, filter_type='single'):
+    cursor = conn.cursor()
+    
+    cursor.execute('''
+        INSERT OR IGNORE INTO filters (name, filter_key, filter_type)
+        VALUES (?, ?, ?)
+    ''', (name, filter_key, filter_type))
+
+def parse_product_data(product_text):
+    application_match = re.search(r'(ВНУТРЕННЕЙ|НАРУЖНОЙ)', product_text)
+    application_type = "внутренняя" if application_match and application_match.group(1) == "ВНУТРЕННЕЙ" else "наружная"
+
+    color_match = re.search(r'цвет\s+([^)]+)', product_text)
+    color = color_match.group(1).strip() if color_match else None
+
+    stiffness_match = re.search(r'SN(\d+)', product_text)
+    stiffness_class = f"SN{stiffness_match.group(1)}" if stiffness_match else None
+
+    size_match = re.search(r'(\d+)х(\d+[,.]?\d*)х(\d+)', product_text.replace(',', '.'))
+    if size_match:
+        diameter = int(size_match.group(1))
+        wall_thickness = float(size_match.group(2))
+        length = int(size_match.group(3))
+    else:
+        diameter = wall_thickness = length = None
+
+    short_name = f"Труба ПВХ {diameter}мм {application_type} {length}мм"
+    
+    return {
+        'short_name': short_name,
+        'full_name': product_text,
+        'diameter': diameter,
+        'wall_thickness': wall_thickness,
+        'length': length,
+        'color': color,
+        'application_type': application_type,
+        'stiffness_class': stiffness_class
+    }
+
+def add_product(conn, product_text, category_id):
+    """Добавление товара с парсингом параметров"""
+    cursor = conn.cursor()
+
+    product_data = parse_product_data(product_text)
+    
+    try:
+        cursor.execute('''
+            INSERT INTO products (
+                name, full_name, description, category_id,
+                diameter, wall_thickness, length, color, 
+                application_type, stiffness_class
+            ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+        ''', (
+            product_data['short_name'],
+            product_data['full_name'],
+            f"Труба ПВХ для {product_data['application_type']} канализации",
+            category_id,
+            product_data['diameter'],
+            product_data['wall_thickness'],
+            product_data['length'],
+            product_data['color'],
+            product_data['application_type'],
+            product_data['stiffness_class']
+        ))
+        
+        product_id = cursor.lastrowid
+
+        filters_to_add = [
+            ('diameter', str(product_data['diameter'])),
+            ('length', str(product_data['length'])),
+            ('application_type', product_data['application_type']),
+            ('color', product_data['color']),
+        ]
+        
+        if product_data['wall_thickness']:
+            filters_to_add.append(('wall_thickness', str(product_data['wall_thickness'])))
+        
+        if product_data['stiffness_class']:
+            filters_to_add.append(('stiffness_class', product_data['stiffness_class']))
+        
+        for filter_key, value in filters_to_add:
+            if value:  
+                cursor.execute('''
+                    INSERT OR IGNORE INTO product_filters (product_id, filter_key, value)
+                    VALUES (?, ?, ?)
+                ''', (product_id, filter_key, value))
+        
+        return product_id
+        
+    except sqlite3.Error as e:
+        log.error(f"Ошибка при добавлении товара: {e}")
+        return None