Skip to content

Commit

Permalink
add connection id
Browse files Browse the repository at this point in the history
  • Loading branch information
Hazelmat committed Feb 7, 2024
1 parent 47c60a6 commit 65532e6
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions dags/preprocess_actors.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,14 +5,13 @@
from airflow.providers.postgres.hooks.postgres import PostgresHook

from utils.utils import load_table, normalize_nom, normalize_url, normalize_email, normalize_phone_number, \

apply_normalization, save_to_database


def normalize_data():

# Setup the PostgresHook connection
pg_hook = PostgresHook(postgres_conn_id='your_connection_id')
pg_hook = PostgresHook(postgres_conn_id='lvao-preprod')

# Use the hook to get the engine
engine = pg_hook.get_sqlalchemy_engine()
Expand Down

0 comments on commit 65532e6

Please sign in to comment.