import requests
import json
import pandas as pd
import numpy as np
from datetime import datetime
from requests.auth import HTTPBasicAuth
from sqlalchemy import create_engine
from airflow import DAG
from airflow.operators.python import PythonOperator

default_args = {
    'owner': 'airflow',
    'depends_on_past': False,
    'start_date': datetime(2023, 1, 1),
    'retries': 1,
}

def get_db_engine():
    """Создает подключение к PostgreSQL"""
    DF_CONFIG = {
        'dbname': "postgres",
        'user': "postgres",
        'password': "4a00d4b90cd830da0796",
        'host': "postgresql", 
        'port': "5432"
    }
    return create_engine(
        f"postgresql+psycopg2://{DF_CONFIG['user']}:{DF_CONFIG['password']}@"
        f"{DF_CONFIG['host']}:{DF_CONFIG['port']}/{DF_CONFIG['dbname']}",
        pool_size=10,
        max_overflow=20
    )

def read_data_1C(**kwargs):
    params = {"СписокСчетов": ["66","66.01","66.02", "66.03","66.04","66.21","66.22","66.23","66.24", "67","67.01","67.02", "67.03","67.04","67.21","67.22","67.23","67.24"]}
    query = """ВЫБРАТЬ
	ОстаткиОбороты.Счет,
	ОстаткиОбороты.Субконто1,
	ОстаткиОбороты.Субконто2,
	ОстаткиОбороты.Организация,
	ОстаткиОбороты.СуммаОборот,
	ОстаткиОбороты.СуммаОборотДт,
	ОстаткиОбороты.СуммаОборотКт,
	ОстаткиОбороты.СуммаКонечныйОстаток,
	ОстаткиОбороты.СуммаКонечныйОстатокДт,
	ОстаткиОбороты.СуммаКонечныйОстатокКт,
	ОстаткиОбороты.СуммаКонечныйРазвернутыйОстатокДт,
	ОстаткиОбороты.СуммаКонечныйРазвернутыйОстатокКт,
	// Добавляем контрагента из договора
	ДоговорыКонтрагентов.Владелец КАК Контрагент,
	ДоговорыКонтрагентов.Номер КАК НомерДоговора,
	ДоговорыКонтрагентов.Дата КАК ДатаДоговора,
	ДоговорыКонтрагентов.СрокДействия КАК СрокДействияДоговора
ИЗ 
	РегистрБухгалтерии.Хозрасчетный.ОстаткиИОбороты КАК ОстаткиОбороты
	ЛЕВОЕ СОЕДИНЕНИЕ Справочник.ДоговорыКонтрагентов КАК ДоговорыКонтрагентов
	ПО ОстаткиОбороты.Субконто2 = ДоговорыКонтрагентов.Ссылка
ГДЕ
	ОстаткиОбороты.Счет.Код В (&СписокСчетов)
    """

    auth = HTTPBasicAuth('obmen', 'bOR2W7w4')
    response = requests.post(
        # http://адрес сервера/название БД/hs/services/query?ProcessPostedRequest
        #
        url=r'http://192.168.1.75/chupd/hs/services/query?ProcessPostedRequest', 
        json={"query":query, "params": params},
        auth=auth,
        verify=False
    )

    data_from_1c = response.json()
    df = pd.DataFrame(data_from_1c['data'])
    engine = get_db_engine()
    table_name = 'oborotno_salbdovaya_vedomostb'

    field_mapping = {
        # Основные поля
        'Счет': 'schet',
        'Субконто1': 'subkonto1',
        'Субконто2': 'subkonto2',
        'Организация': 'organizaciya',
        'Контрагент': 'kontagent',
        'НомерДоговора': 'nomer',
        'ДатаДоговора': 'date_bedin',
        'СрокДействияДоговора': 'date_end',
        
        # Суммовые остатки и обороты
        'СуммаОборот': 'summa_oborot',
        'СуммаОборотДт': 'summa_oborot_dt',
        'СуммаОборотКт': 'summa_oborot_kt',
        'СуммаКонечныйОстаток': 'summa_konechnyy_ostatok',
        'СуммаКонечныйОстатокДт': 'summa_konechnyy_ostatok_dt',
        'СуммаКонечныйОстатокКт': 'summa_konechnyy_ostatok_kt',
        'СуммаКонечныйРазвернутыйОстатокДт': 'summa_konechnyy_razvernutyy_ostatok_dt',
        'СуммаКонечныйРазвернутыйОстатокКт': 'summa_konechnyy_razvernutyy_ostatok_kt'
    }

    df = df.rename(columns=field_mapping)
    with engine.begin() as conn:
        if not df.empty:
            conn.execute(f"CREATE TEMP TABLE temp_{table_name} AS SELECT * FROM public.{table_name} WHERE 1 = 0")
            df.to_sql(
                f'temp_{table_name}',
                con=conn,
                if_exists='append',
                index=False,
                method='multi'
            )
            conn.execute(f"DELETE FROM public.{table_name} where get_date = CURRENT_DATE::date::timestamp")
            conn.execute(f"""
            INSERT INTO public.{table_name}
            SELECT  
                  schet
                , subkonto1
                , subkonto2
                , organizaciya
                , kontagent
                , nomer
                , date_bedin
                , date_end
                , summa_oborot
                , summa_oborot_dt
                , summa_oborot_kt
                , summa_konechnyy_ostatok
                , summa_konechnyy_ostatok_dt
                , summa_konechnyy_ostatok_kt
                , summa_konechnyy_razvernutyy_ostatok_dt
                , summa_konechnyy_razvernutyy_ostatok_kt
                , CURRENT_DATE::date::timestamp 
            FROM temp_{table_name}
            --ON CONFLICT (schet, subkonto1, subkonto2, organizaciya)
            """
            )

with DAG(
    dag_id='data_download_from_1C_source',
    default_args=default_args,
    description='Выгрузка данных из 1С',
    schedule_interval=None, #"0,30 01-10 * * *",
    catchup=False,
    tags=['sigma'],
) as dag:

    read_data_1C_task = PythonOperator(
        task_id="read_data_1C",
        python_callable=read_data_1C,
        provide_context=True
    )

read_data_1C_task