0
respostas

[Dúvida] O código não sai da etapa extrai_dados, fica como "running"

Eu não sei qual o problema, a principio até tentei mudar as datas para esse mês apenas para ver se ia, mas ele passa da Etapa de criar a pasta, corretamente, e fica na Etapa de extrai_dados com a descrição "Running", mas não sai dela.

Insira aqui a descrição dessa imagem para ajudar na acessibilidade

o código:

from airflow import DAG 
from airflow.operators.bash import BashOperator
from airflow.operators.python import PythonOperator
from airflow.macros import ds_add
import pendulum
from os.path import join
import pandas as pd

with DAG(
    "dados_climaticos_2",
    start_date=pendulum.datetime(2022, 8, 22, tz="UTC"),
    schedule_interval='0 0 * * 1', # executar toda segunda feira
) as dag:

    tarefa_1  = BashOperator(
        task_id = 'cria_pasta',
        bash_command= 'mkdir -p "/Users/marciofernandes/Documents/airflow/semana={{data_interval_end.strftime("%Y-%m-%d")}}"'
    )

    def extrai_dados(data_interval_end):
        city = 'Boston'
        key = 'YMQUCLMC2X2EMY5PRPKVQ6ZFK'

        URL = join('https://weather.visualcrossing.com/VisualCrossingWebServices/rest/services/timeline/',
            f'{city}/{data_interval_end}/{ds_add(data_interval_end, 7)}?unitGroup=metric&include=days&key={key}&contentType=csv')

        dados = pd.read_csv(URL)

        file_path = f'/Users/marciofernandes/Documents/airflow/semana={data_interval_end}/'

        dados.to_csv(file_path + 'dados_brutos.csv')
        dados[['datetime','tempmin', 'temp', 'tempmax']].to_csv(file_path + 'temperaturas.csv')
        dados[['datetime', 'description', 'icon']].to_csv(file_path + 'condicoes.csv')

    tarefa_2 = PythonOperator(
        task_id = 'extrai_dados',
        python_callable = extrai_dados,
        op_kwargs = {'data_interval_end': '{{data_interval_end.strftime("%Y-%m-%d")}}'}
    )

    tarefa_1 >> tarefa_2