Пример #1
0
def metatable_update(dag, tablename, schedule, **kwargs):
    dataset = tablename.split('.')[1]
    table_name = tablename.split('.')[2]
    bq_hook = BigQueryHook(bigquery_conn_id='bigquery_default',
                           location='europe-west3',
                           use_legacy_sql=False)
    pg_conn = config[schedule][tablename][1]
    ti = kwargs['ti']
    pgmax_ts = str(
        ti.xcom_pull(task_ids='get_maxts_{}_{}'.format(tablename, pg_conn))[1])
    bq_hook.run(
        sql=
        "update bqadmin.tablesync_meta set max_value='{}' where tablename='{}' and datasource_dbconn='{}'"
        .format(pgmax_ts, dataset + '.' + table_name, pg_conn))
    return 'Executed the update query'
Пример #2
0
 def execute(self, context):
     logging.info('Executing: %s', str(self.bql))
     hook = BigQueryHook(bigquery_conn_id=self.bigquery_conn_id,
                         delegate_to=self.delegate_to)
     hook.run(self.bql, self.destination_dataset_table,
              self.write_disposition)
Пример #3
0
 def execute(self, context):
     logging.info('Executing: %s', str(self.bql))
     hook = BigQueryHook(bigquery_conn_id=self.bigquery_conn_id)
     hook.run(self.bql, self.destination_dataset_table)
Пример #4
0
 def execute(self, context):
     logging.info('Executing: %s', str(self.bql))
     hook = BigQueryHook(bigquery_conn_id=self.bigquery_conn_id)
     hook.run(self.bql, self.destination_dataset_table)
Пример #5
0
 def execute(self, context):
     logging.info('Executing: %s', str(self.bql))
     hook = BigQueryHook(bigquery_conn_id=self.bigquery_conn_id, delegate_to=self.delegate_to)
     hook.run(self.bql, self.destination_dataset_table, self.write_disposition)