fix paths to storage and logs

This commit is contained in:
2025-07-22 12:03:03 +02:00
parent 5b5e962773
commit be8fae969c
8 changed files with 16 additions and 17 deletions

View File

@@ -1,4 +1,3 @@
import pandas as pd
import polars as pl
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(
io_manager_key="polars_parquet_io_manager",
partitions_def=partitions_def_single,
@@ -58,7 +47,6 @@ def asset_single_1(context):
def asset_multi_1(context):
ic()
ic(context.partition_key)
return pl.DataFrame(
[{"date": context.partition_key, "data": f"Data for {context.partition_key}"}]
)

View File

@@ -9,7 +9,7 @@ install()
# Define a job that includes both assets
daily_job = define_asset_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(
@@ -20,6 +20,8 @@ definitions = Definitions(
)
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],
)

View File

@@ -11,7 +11,6 @@ from dagster._core.instance import DynamicPartitionsStore
from dagster._serdes import whitelist_for_serdes
# @whitelist_for_serdes
class LatestTwoPartitionsMapping(PartitionMapping):
def get_upstream_mapped_partitions_result_for_partitions(
self,

View File

@@ -110,7 +110,7 @@ def deals(context):
@asset(deps=[deals], io_manager_key="polars_parquet_io_manager")
def new_deals(context: OpExecutionContext) -> pl.DataFrame:
ic()
storage_dir = context.instance.storage_directory()
storage_dir = context.resources.polars_parquet_io_manager.base_dir
asset_key = "deals"
with duckdb.connect() as con:

View File

@@ -29,7 +29,7 @@ definitions = Definitions(
for asset in load_assets_from_modules([assets])
],
resources={
"polars_parquet_io_manager": PolarsParquetIOManager(),
"polars_parquet_io_manager": PolarsParquetIOManager(base_dir="/storage"),
"duckdb_io_manager": PandasDuckDBIOManager(database="vinyl"),
},
jobs=[deals_job, check_partititions_job, musicbrainz_lookup_job],

View File

@@ -25,6 +25,7 @@ services:
DAGSTER_CURRENT_IMAGE: user_code_vinyl
volumes:
- /opt/dagster/apps/:/apps/:ro
- /opt/dagster/logs/:/logs:rw
- /opt/dagster/storage/import/:/storage/import/:ro
- /opt/dagster/storage/deals/:/storage/deals/:rw
networks:
@@ -45,5 +46,6 @@ services:
DAGSTER_CURRENT_IMAGE: user_code_other
volumes:
- /opt/dagster/apps/:/apps:ro
- /opt/dagster/logs/:/logs:rw
networks:
- dagster

View File

@@ -19,6 +19,7 @@ x-volumes: &volumes
- /opt/dagster/dagster.yaml:/opt/dagster/home/dagster.yaml:ro
- /opt/dagster/workspace.yaml:/opt/dagster/home/workspace.yaml:ro
- /opt/dagster/storage/:/storage/:rw
- /opt/dagster/logs/:/logs:rw
- /var/run/docker.sock:/var/run/docker.sock:rw
services:

View File

@@ -23,6 +23,7 @@ run_launcher:
volumes:
- /opt/dagster/apps/:/apps:ro
- /opt/dagster/storage/:/storage/:rw
- /opt/dagster/logs/:/logs:rw
run_storage:
module: dagster_postgres.run_storage
@@ -71,3 +72,9 @@ event_log_storage:
env: DAGSTER_POSTGRES_PASSWORD
db_name:
env: DAGSTER_POSTGRES_DB
compute_logs:
module: dagster.core.storage.local_compute_log_manager
class: LocalComputeLogManager
config:
base_dir: /logs