# -*- coding:utf-8 -*-
"""
@Author  : xuxingchen
@Contact : xuxingchen@sinochem.com
@Desc    : 内置的产品信息表，在初始化时会初始化csv进行覆盖
"""
import csv
import os
from datetime import datetime

from pydantic import BaseModel

from utils import logger
from utils.database import BaseTable
from utils.misc import now_datetime_second


class Product(BaseModel):
    product_id: str
    product_name: str
    node_type: str


class ProductsTable(BaseTable):
    @staticmethod
    def check(table_handler: BaseTable):
        """检测是否存在当前表"""
        init_config_path = os.path.join(os.path.dirname(os.path.abspath("__file__")), "data/InitialData/products.csv")
        if os.path.exists(init_config_path):
            table_handler.execute("DROP TABLE IF EXISTS products;")
        table_handler.execute(
            f"""
            CREATE TABLE products (
                product_id TEXT,
                product_name TEXT,
                node_type TEXT,
                update_datetime TEXT,
                PRIMARY KEY (product_id)
            )
            """
        )
        if os.path.exists(init_config_path):
            with open(init_config_path, newline='', encoding='utf8') as csvfile:
                csvreader = csv.reader(csvfile)
                head = next(csvreader)
                data = []
                if len(head) == 3:
                    for row in csvreader:
                        product_id = row[0].strip()
                        product_name = row[1].strip()
                        node_type = row[2].strip()
                        update_datetime = datetime.now().strftime("%Y-%m-%d %H:%M:%S.%f")
                        data.append((product_id, product_name, node_type, update_datetime))
                    table_handler.executemany(
                        f"""
                            INSERT INTO products 
                            (product_id, product_name, node_type, update_datetime) 
                            VALUES (?, ?, ?, ?)
                            """,
                        data
                    )

        table_handler.query("SELECT name FROM sqlite_master WHERE type='table' AND name='brands'")
        if table_handler.cursor.fetchone() is None:
            table_handler.execute(
                f"""
                CREATE TABLE brands (
                    brand_id INT,
                    brand_name TEXT,
                    status TEXT,
                    add_datetime TEXT,
                    update_datetime TEXT,
                    PRIMARY KEY (brand_id)
                )
                """
            )
        init_config_path = os.path.join(os.path.dirname(os.path.abspath("__file__")), "data/InitialData/brands.csv")
        if os.path.exists(init_config_path):
            with open(init_config_path, newline='', encoding='utf8') as csvfile:
                csvreader = csv.reader(csvfile)
                head = next(csvreader)
                data = []
                if len(head) == 3:
                    for row in csvreader:
                        brand_id = row[0].strip()
                        brand_name = row[1].strip()
                        status = row[2].strip()
                        data.append((brand_id, brand_name, status, now_datetime_second(), now_datetime_second()))
                    table_handler.executemany(
                        f"""
                        INSERT INTO brands 
                        (brand_id, brand_name, status, add_datetime, update_datetime) 
                        VALUES (?, ?, ?, ?, ?)
                        ON CONFLICT (brand_id) DO NOTHING
                        """,
                        data
                    )

    @staticmethod
    def get_product_info(table_handler: BaseTable, product_id: str):
        table_handler.query(
            """
            SELECT product_name, node_type
            FROM products
            WHERE product_id = ?
            """,
            (product_id,)
        )
        res = table_handler.cursor.fetchall()
        if res:
            return Product(product_id=product_id, product_name=res[0][0], node_type=res[0][1])
        else:
            return None