from os import getenv, environ
import hvac
from oracledb import Error, create_pool, init_oracle_client
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
                        FROM all_arguments 
                        WHERE object_name = 'P_RK_SEND_JSON_LIST_FACEACC'
                        ORDER BY position
                    """)
                    for arg in cursor:
                        print(f"Name: {arg[0]}, Type: {arg[1]}, Direction: {arg[2]}")
                    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))
                        respons = cursor.callproc('P_RK_SEND_JSON_LIST_FACEACC', [int(id), organ, names, date_from, date_to, ver])
                        print(respons)
                        # тестого
                        print("Ответ от БД: ", respons)
        except Error as e:
            raise HTTPException(status_code=500, detail=f"Database error: {e}")