import sys
import logging
from os import path
from datetime import datetime, timedelta
import sys

from airflow.decorators import dag, task
from airflow.operators.python import get_current_context
from airflow.operators.bash_operator import BashOperator
from airflow.models.param import Param

logging.basicConfig(stream=sys.stdout, level=logging.INFO)

ARGS = { 
    'owner': 'apentori',
    'depends_on_past': False,
    'start_date': datetime(2023,6,1),
    'email': ['alexis@status.im'],
    'email_on_failure': False,
    'email_on_retry': False,
    'retries': 0,
    'retry_delay': timedelta(minutes=10),
}

@dag('dbt_models', default_args=ARGS, schedule_interval=None)
def dbt_models():
    
    BashOperator(
        task_id = 'dbt_postgres_debug',
        bash_command='dbt debug --profiles-dir /dbt --project-dir /dbt/dbt-models'
    )
    BashOperator(
        task_id = 'dbt_postgres_test',
        bash_command='dbt test --profiles-dir /dbt --project-dir /dbt/dbt-models'
    )
    BashOperator(
       task_id='dbt_postgres_run',
       bash_command='dbt run --profiles-dir /dbt --project-dir /dbt/dbt-models --select github'
    )

dbt_models()

@dag('dbt_select_models', params={ "model": Param("status_bi", type="string")} , default_args=ARGS,schedule_interval=None)
def dbt_select_models():
    BashOperator(
       task_id='dbt_postgres_run',
       bash_command='dbt run --profiles-dir /dbt --project-dir /dbt/dbt-models --select {{ params.model}}',
    )
dbt_select_models()