Skip to content

Instantly share code, notes, and snippets.

View vvgsrk's full-sized avatar

Venkata Gowri Sai Rakesh Kumar Varanasi vvgsrk

View GitHub Profile
USE ROLE USERADMIN;
-- Jamie
ALTER USER jamie UNSET TAG dp_governance.tags.employee_sensitive;
-- Avery
ALTER USER avery UNSET TAG dp_governance.tags.employee_sensitive;
-- Blake
ALTER USER blake UNSET TAG dp_governance.tags.employee_sensitive;
@vvgsrk
vvgsrk / dbt_cloud_sensor.py
Created March 6, 2025 19:13
dbt cloud sensor py file
from airflow.providers.dbt.cloud.operators.dbt import DbtCloudRunJobOperator
from airflow.hooks.base_hook import BaseHook
from config.current import DBT_CLOUD_CONNECTION_ID
class DBTCloudSensor:
def __init__(self):
self.dbt_cloud_connection = BaseHook.get_connection(DBT_CLOUD_CONNECTION_ID)
self.account_id = self.dbt_cloud_connection.extra_dejson.get('account_id')
@vvgsrk
vvgsrk / dag_data_source.py
Last active March 6, 2025 19:28
An example code to invoke dbt cloud sensor class
from airflow import DAG
from config.current.agm import PARENT_DAG_NAME
from config.current.data_source import job_id_in_dbt_cloud
with DAG(
dag_id=PARENT_DAG_NAME,
...
) as dag: