fix paths to storage and logs
This commit is contained in:
@@ -1,4 +1,3 @@
|
|||||||
import pandas as pd
|
|
||||||
import polars as pl
|
import polars as pl
|
||||||
|
|
||||||
from dagster import (
|
from dagster import (
|
||||||
@@ -23,16 +22,6 @@ partitions_def_multi = MultiPartitionsDefinition(
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
@asset(
|
|
||||||
# tags={
|
|
||||||
# "dagster/executor": "other_executor"
|
|
||||||
# },
|
|
||||||
)
|
|
||||||
def dummy_asset():
|
|
||||||
"""A dummy asset to ensure the module is recognized by Dagster."""
|
|
||||||
return pd.DataFrame({"dummy": [1, 2, 3]})
|
|
||||||
|
|
||||||
|
|
||||||
@asset(
|
@asset(
|
||||||
io_manager_key="polars_parquet_io_manager",
|
io_manager_key="polars_parquet_io_manager",
|
||||||
partitions_def=partitions_def_single,
|
partitions_def=partitions_def_single,
|
||||||
@@ -58,7 +47,6 @@ def asset_single_1(context):
|
|||||||
def asset_multi_1(context):
|
def asset_multi_1(context):
|
||||||
ic()
|
ic()
|
||||||
ic(context.partition_key)
|
ic(context.partition_key)
|
||||||
|
|
||||||
return pl.DataFrame(
|
return pl.DataFrame(
|
||||||
[{"date": context.partition_key, "data": f"Data for {context.partition_key}"}]
|
[{"date": context.partition_key, "data": f"Data for {context.partition_key}"}]
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -9,7 +9,7 @@ install()
|
|||||||
# Define a job that includes both assets
|
# Define a job that includes both assets
|
||||||
daily_job = define_asset_job(
|
daily_job = define_asset_job(
|
||||||
"daily_job",
|
"daily_job",
|
||||||
selection=[assets.dummy_asset, assets.asset_multi_1, assets.asset_multi_2],
|
selection=[assets.asset_multi_1, assets.asset_multi_2],
|
||||||
)
|
)
|
||||||
|
|
||||||
definitions = Definitions(
|
definitions = Definitions(
|
||||||
@@ -20,6 +20,8 @@ definitions = Definitions(
|
|||||||
)
|
)
|
||||||
for asset in load_assets_from_modules([assets])
|
for asset in load_assets_from_modules([assets])
|
||||||
],
|
],
|
||||||
resources={"polars_parquet_io_manager": PolarsParquetIOManager()},
|
resources={
|
||||||
|
"polars_parquet_io_manager": PolarsParquetIOManager(base_dir="/storage")
|
||||||
|
},
|
||||||
jobs=[daily_job],
|
jobs=[daily_job],
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -11,7 +11,6 @@ from dagster._core.instance import DynamicPartitionsStore
|
|||||||
from dagster._serdes import whitelist_for_serdes
|
from dagster._serdes import whitelist_for_serdes
|
||||||
|
|
||||||
|
|
||||||
# @whitelist_for_serdes
|
|
||||||
class LatestTwoPartitionsMapping(PartitionMapping):
|
class LatestTwoPartitionsMapping(PartitionMapping):
|
||||||
def get_upstream_mapped_partitions_result_for_partitions(
|
def get_upstream_mapped_partitions_result_for_partitions(
|
||||||
self,
|
self,
|
||||||
|
|||||||
@@ -110,7 +110,7 @@ def deals(context):
|
|||||||
@asset(deps=[deals], io_manager_key="polars_parquet_io_manager")
|
@asset(deps=[deals], io_manager_key="polars_parquet_io_manager")
|
||||||
def new_deals(context: OpExecutionContext) -> pl.DataFrame:
|
def new_deals(context: OpExecutionContext) -> pl.DataFrame:
|
||||||
ic()
|
ic()
|
||||||
storage_dir = context.instance.storage_directory()
|
storage_dir = context.resources.polars_parquet_io_manager.base_dir
|
||||||
asset_key = "deals"
|
asset_key = "deals"
|
||||||
|
|
||||||
with duckdb.connect() as con:
|
with duckdb.connect() as con:
|
||||||
|
|||||||
@@ -29,7 +29,7 @@ definitions = Definitions(
|
|||||||
for asset in load_assets_from_modules([assets])
|
for asset in load_assets_from_modules([assets])
|
||||||
],
|
],
|
||||||
resources={
|
resources={
|
||||||
"polars_parquet_io_manager": PolarsParquetIOManager(),
|
"polars_parquet_io_manager": PolarsParquetIOManager(base_dir="/storage"),
|
||||||
"duckdb_io_manager": PandasDuckDBIOManager(database="vinyl"),
|
"duckdb_io_manager": PandasDuckDBIOManager(database="vinyl"),
|
||||||
},
|
},
|
||||||
jobs=[deals_job, check_partititions_job, musicbrainz_lookup_job],
|
jobs=[deals_job, check_partititions_job, musicbrainz_lookup_job],
|
||||||
|
|||||||
@@ -25,6 +25,7 @@ services:
|
|||||||
DAGSTER_CURRENT_IMAGE: user_code_vinyl
|
DAGSTER_CURRENT_IMAGE: user_code_vinyl
|
||||||
volumes:
|
volumes:
|
||||||
- /opt/dagster/apps/:/apps/:ro
|
- /opt/dagster/apps/:/apps/:ro
|
||||||
|
- /opt/dagster/logs/:/logs:rw
|
||||||
- /opt/dagster/storage/import/:/storage/import/:ro
|
- /opt/dagster/storage/import/:/storage/import/:ro
|
||||||
- /opt/dagster/storage/deals/:/storage/deals/:rw
|
- /opt/dagster/storage/deals/:/storage/deals/:rw
|
||||||
networks:
|
networks:
|
||||||
@@ -45,5 +46,6 @@ services:
|
|||||||
DAGSTER_CURRENT_IMAGE: user_code_other
|
DAGSTER_CURRENT_IMAGE: user_code_other
|
||||||
volumes:
|
volumes:
|
||||||
- /opt/dagster/apps/:/apps:ro
|
- /opt/dagster/apps/:/apps:ro
|
||||||
|
- /opt/dagster/logs/:/logs:rw
|
||||||
networks:
|
networks:
|
||||||
- dagster
|
- dagster
|
||||||
|
|||||||
@@ -19,6 +19,7 @@ x-volumes: &volumes
|
|||||||
- /opt/dagster/dagster.yaml:/opt/dagster/home/dagster.yaml:ro
|
- /opt/dagster/dagster.yaml:/opt/dagster/home/dagster.yaml:ro
|
||||||
- /opt/dagster/workspace.yaml:/opt/dagster/home/workspace.yaml:ro
|
- /opt/dagster/workspace.yaml:/opt/dagster/home/workspace.yaml:ro
|
||||||
- /opt/dagster/storage/:/storage/:rw
|
- /opt/dagster/storage/:/storage/:rw
|
||||||
|
- /opt/dagster/logs/:/logs:rw
|
||||||
- /var/run/docker.sock:/var/run/docker.sock:rw
|
- /var/run/docker.sock:/var/run/docker.sock:rw
|
||||||
|
|
||||||
services:
|
services:
|
||||||
|
|||||||
@@ -23,6 +23,7 @@ run_launcher:
|
|||||||
volumes:
|
volumes:
|
||||||
- /opt/dagster/apps/:/apps:ro
|
- /opt/dagster/apps/:/apps:ro
|
||||||
- /opt/dagster/storage/:/storage/:rw
|
- /opt/dagster/storage/:/storage/:rw
|
||||||
|
- /opt/dagster/logs/:/logs:rw
|
||||||
|
|
||||||
run_storage:
|
run_storage:
|
||||||
module: dagster_postgres.run_storage
|
module: dagster_postgres.run_storage
|
||||||
@@ -71,3 +72,9 @@ event_log_storage:
|
|||||||
env: DAGSTER_POSTGRES_PASSWORD
|
env: DAGSTER_POSTGRES_PASSWORD
|
||||||
db_name:
|
db_name:
|
||||||
env: DAGSTER_POSTGRES_DB
|
env: DAGSTER_POSTGRES_DB
|
||||||
|
|
||||||
|
compute_logs:
|
||||||
|
module: dagster.core.storage.local_compute_log_manager
|
||||||
|
class: LocalComputeLogManager
|
||||||
|
config:
|
||||||
|
base_dir: /logs
|
||||||
|
|||||||
Reference in New Issue
Block a user