Rubén Briones
03/15/2023, 6:12 PMPostgresHook(conn_id='db').get_pandas_df(query)
in Dagster?yuhan
03/16/2023, 4:45 AMyuhan
03/16/2023, 4:48 AMfrom dagster import asset, Definitions, ConfigurableResource
class MyDBResource(ConfigurableResource):
conn_id: str
def get_pandas_df(self) -> Response:
...
@asset
def data_from_db(my_conn: MyDBResource) -> Dict[str, Any]:
return my_conn.get_pandas_df(query)
defs = Definitions(
assets=[data_from_db],
resources={
"my_conn": MyDBResource(conn_id="..."),
},
)