Skip to main content

How to add task list to existing DAG

I'm trying to automate the creation of batch pipelines from backend mySQLL to BiqQuery and I've ran into issue with my script.

It initialises a DAG with it's own class (DagCreator) and then passes it as a instance variable in my MySQLBatchPipelines class.

The tasks and the order of the tasks are contained in methods inside my MySQLBatchPipelines instance.

How would I add those tasks and the tasks list to my existing DAG?

Here's some example code so you understand my query:

class DagCreator:

    def __init__(
        self,
        dag_id,
        start_date,
        time_delay: int
    ):
        self.dag_id = dag_id,
        self.start_date = start_date,
        self.time_delay = time_delay

        print(self.dag_id)

    def DagToReturn(self):
        args = {
            'owner': 'Data Lake',
            'depends_on_past': True,
            'start_date': self.start_date,
            'email_on_failure': False,
            'email_on_retry': False,
            'on_failure_callback': 'failed_dag_slack_alert',
            'retries': 0,
            'max_active_runs': 1,
            'max_file_size': int(50e6)
        }
        dag_to_return = DAG(
            dag_id=f'{self.dag_id}_batch',
            schedule_interval=f'{self.time_delay} * * *',
            concurrency=4,
            default_args=args
        )
        return dag_to_return

dag = DagCreator('table_name', dt.datetime(2022, 1, 15), 7)
dag_returned = dag.DagToReturn()
dag_returned

This is the second class and it's methods:


class MySQLBatchPipeline:
    '''
    Class to generate MySQL batch pipelines that store CSV's 
    in GCS then import them into BigQuery
    '''
    export_format='CSV' 

    def __init__(
        self,
        dag,
    ....

    # lots of methods that return tasks like these
    def uk_source_table_extract_task(self):
        uk_source_table_extract_task = MySqlToGoogleCloudStorageOperator(
            sql=self.queries['new_rows_batch_query'], # <--- needs testing
            bucket=self.gcs_bucket,
            filename=self.extract_gcs_path,
            approx_max_file_size_bytes=MAX_FILE_SIZE, # <-- is max_file_size necc? is in dag arg's is it necc here
            mysql_conn_id=self.mysql_connection_id,
            export_format=self.export_format,
            google_cloud_storage_conn_id=self.gcs_connection_id,
            params={
                'export_format': self.export_format,
                'country': 'uk',
                'time_delay': self.time_delay
            },
            task_id='uk_source_table_extract_task',
            dag=self.dag
      return uk_expected_stg_row_count_extract_task


    # this the the workflow I want to add to the existing DAG I've passed a instance attribute
        def uk_workflow(self):
            uk_source_table_extract_task.set_downstream(
                [
                    uk_table_name_load_task,
                    uk_truncate_table_name_staging_task,
                    uk_expected_stg_row_count_extract_task
                ]
            )
            uk_truncate_table_name_staging_task.set_downstream(
            uk_table_name_load_task
            # etc etc until the final task


source https://stackoverflow.com/questions/70756217/how-to-add-task-list-to-existing-dag

Comments

Popular posts from this blog

ValueError: X has 10 features, but LinearRegression is expecting 1 features as input

So, I am trying to predict the model but its throwing error like it has 10 features but it expacts only 1. So I am confused can anyone help me with it? more importantly its not working for me when my friend runs it. It works perfectly fine dose anyone know the reason about it? cv = KFold(n_splits = 10) all_loss = [] for i in range(9): # 1st for loop over polynomial orders poly_order = i X_train = make_polynomial(x, poly_order) loss_at_order = [] # initiate a set to collect loss for CV for train_index, test_index in cv.split(X_train): print('TRAIN:', train_index, 'TEST:', test_index) X_train_cv, X_test_cv = X_train[train_index], X_test[test_index] t_train_cv, t_test_cv = t[train_index], t[test_index] reg.fit(X_train_cv, t_train_cv) loss_at_order.append(np.mean((t_test_cv - reg.predict(X_test_cv))**2)) # collect loss at fold all_loss.append(np.mean(loss_at_order)) # collect loss at order plt.plot(np.log(al...

Sorting large arrays of big numeric stings

I was solving bigSorting() problem from hackerrank: Consider an array of numeric strings where each string is a positive number with anywhere from to digits. Sort the array's elements in non-decreasing, or ascending order of their integer values and return the sorted array. I know it works as follows: def bigSorting(unsorted): return sorted(unsorted, key=int) But I didnt guess this approach earlier. Initially I tried below: def bigSorting(unsorted): int_unsorted = [int(i) for i in unsorted] int_sorted = sorted(int_unsorted) return [str(i) for i in int_sorted] However, for some of the test cases, it was showing time limit exceeded. Why is it so? PS: I dont know exactly what those test cases were as hacker rank does not reveal all test cases. source https://stackoverflow.com/questions/73007397/sorting-large-arrays-of-big-numeric-stings

How to load Javascript with imported modules?

I am trying to import modules from tensorflowjs, and below is my code. test.html <!DOCTYPE html> <html lang="en"> <head> <meta charset="UTF-8"> <title>Document</title </head> <body> <script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@2.0.0/dist/tf.min.js"></script> <script type="module" src="./test.js"></script> </body> </html> test.js import * as tf from "./node_modules/@tensorflow/tfjs"; import {loadGraphModel} from "./node_modules/@tensorflow/tfjs-converter"; const MODEL_URL = './model.json'; const model = await loadGraphModel(MODEL_URL); const cat = document.getElementById('cat'); model.execute(tf.browser.fromPixels(cat)); Besides, I run the server using python -m http.server in my command prompt(Windows 10), and this is the error prompt in the console log of my browser: Failed to loa...