from os import getenv, environ
import hvac
from oracledb import Error, create_pool, init_oracle_client, CLOB
from fastapi import HTTPException
from dateutil import parser

init_oracle_client()

class SimpleDB:
    def __init__(self):
        self._connect_hvac()
        self.pool 
        init_oracle_client()
    
    def _connect_hvac(self):
        """
        Подключение к Vault и получение параметров БД
        """
        try:
            client = hvac.Client(
                url='https://vlt.dataekb.ru:8222',
                token=environ.get('VAULT_TOKEN'),
            )
            
            read_response = client.secrets.kv.v2.read_secret_version(
                path='oracledb', 
                mount_point='kv'
            )
            
            # Создаем пул соединений
            self.pool = create_pool(
                user=read_response['data']['data']['user'],
                password=read_response['data']['data']['password'],
                dsn=read_response['data']['data']['cs'],
                min=2,
                max=10,
                increment=1
            )
        except Exception as e:
            raise HTTPException(status_code=500, detail=f"Database connection failed: {e}")
    
    def data_transfer_in_database(self, dict_data: dict):
        """
        Передача данных в базу
        Процедура на вставку: 

        P_RK_GOVERNMENT_REPORTS_INSERS
        (
        ID IN NUMBER,
        ORGAN IN VARCHAR2,
        NAMES IN VARCHAR2,
        DATE_FROM in date,
        DATE_TO   in date,
        VERS  IN VARCHAR2)
        """
        if not self.pool:
            raise HTTPException(status_code=500, detail="Database pool not initialized")
        
        if not dict_data:
            raise ValueError(dict_data, "No data to process")

        try:
            with self.pool.acquire() as connection: 
                with connection.cursor() as cursor:
                    # Узнать сигнатуру процедуры
                    cursor.execute("""
                        SELECT 
                            argument_name,
                            data_type,
                            in_out,
                            position,
                            data_length,
                            data_precision,
                            data_scale
                        FROM all_arguments 
                        WHERE object_name = 'P_RK_SEND_JSON_LIST_FACEACC'
                        ORDER BY position
                    """)
                    
                    print("Сигнатура процедуры P_RK_SEND_JSON_LIST_FACEACC:")
                    print("Pos | Name | Type | Direction | Length")
                    print("-" * 50)
                    for arg in cursor:
                        pos, name, dtype, direction, length, precision, scale = arg
                        print(f"{pos:3d} | {name or 'RETURN':15} | {dtype:15} | {direction:8} | {length or 'N/A':5}")

                    for id, value in dict_data.items():
                        if len(value) != 5:
                            continue    
                        organ, names, date_from_str, date_to_str, ver = value
                        # Парсим даты
                        date_from = parser.parse(date_from_str).date()
                        date_to = parser.parse(date_to_str).date() if date_to_str else None
                        print("Вывод отправляемых агрументов: ", int(id), organ, names, date_from, date_to, ver)
                        print(type(int(id)), type(organ), type(names), type(date_from), type(date_to), type(ver))
                        result_var = cursor.var(CLOB)
                        cursor.callproc('P_RK_SEND_JSON_LIST_FACEACC', [
                            int(id), 
                            organ, 
                            names, 
                            date_from, 
                            date_to, ver,
                            result_var  # OUT-параметр всегда последний
                            ])
                        json_result = result_var.getvalue()
                        print("JSON результат:", json_result)
        except Error as e:
            raise HTTPException(status_code=500, detail=f"Database error: {e}")