[docs] update docs_beta_snippets to use import dagster as dg

Author: cmpadden

examples/docs_beta_snippets/docs_beta_snippets/guides/components/shell-script-component/2-shell-command-empty.py

@@ -1,4 +1,5 @@
-from dagster import Definitions
+import dagster as dg
+
 from dagster_components import (
     Component,
     ComponentLoadContext,
@@ -25,6 +26,6 @@ def get_schema(cls):
     def get_scaffolder(cls) -> DefaultComponentScaffolder:
         return DefaultComponentScaffolder()
 
-    def build_defs(self, load_context: ComponentLoadContext) -> Definitions:
+    def build_defs(self, load_context: ComponentLoadContext) -> dg.Definitions:
         # Add definition construction logic here.
-        return Definitions()
+        return dg.Definitions()

examples/docs_beta_snippets/docs_beta_snippets/guides/components/shell-script-component/empty.py

@@ -6,12 +6,12 @@
 )
 from pydantic import BaseModel
 
-from dagster import Definitions
+import dagster as dg
 
 
 @registered_component_type(name="shell_command")
 class ShellCommand(Component):
     @classmethod
     def get_schema(cls) -> type[ResolvableSchema]: ...
 
-    def build_defs(self, load_context: ComponentLoadContext) -> Definitions: ...
+    def build_defs(self, load_context: ComponentLoadContext) -> dg.Definitions: ...

examples/docs_beta_snippets/docs_beta_snippets/guides/monitor-alert/custom-logging/asset-job-example.py

@@ -3,7 +3,6 @@
 import requests
 
 import dagster as dg
-from dagster import json_console_logger
 
 LOGGER_CONFIG = {"loggers": {"console": {"config": {"log_level": "INFO"}}}}
 
@@ -30,5 +29,5 @@ def hackernews_topstory_ids(context: dg.AssetExecutionContext) -> list[int]:
 defs = dg.Definitions(
     assets=[hackernews_topstory_ids],
     jobs=[hackernews_topstory_ids_job],
-    loggers={"console": json_console_logger},
+    loggers={"console": dg.json_console_logger},
 )

examples/docs_beta_snippets/docs_beta_snippets/guides/monitor-alert/custom-logging/ops-job-example.py

@@ -1,7 +1,6 @@
 import requests
 
 import dagster as dg
-from dagster import json_console_logger
 
 LOGGER_CONFIG = {"loggers": {"console": {"config": {"log_level": "INFO"}}}}
 
@@ -19,7 +18,7 @@ def get_hackernews_topstory_ids(context: dg.OpExecutionContext):
     return top_500_newstories
 
 
-@dg.job(logger_defs={"console": json_console_logger}, config=LOGGER_CONFIG)
+@dg.job(logger_defs={"console": dg.json_console_logger}, config=LOGGER_CONFIG)
 def hackernews_topstory_ids_job():
     get_hackernews_topstory_ids()
 

examples/docs_beta_snippets/docs_beta_snippets/guides/tutorials/multi-asset-integration/integration.py

@@ -5,16 +5,8 @@
 
 import yaml
 
-from dagster import (
-    AssetExecutionContext,
-    AssetKey,
-    AssetMaterialization,
-    AssetsDefinition,
-    AssetSpec,
-    ConfigurableResource,
-    _check as check,
-    multi_asset,
-)
+import dagster as dg
+import dagster._check as check
 from dagster._annotations import public
 
 
@@ -33,9 +25,11 @@ def load(self):
         return yaml.safe_load(Path(self.replication_configuration_yaml).read_text())
 
 
-class ReplicationResource(ConfigurableResource):
+class ReplicationResource(dg.ConfigurableResource):
     @public
-    def run(self, context: AssetExecutionContext) -> Iterator[AssetMaterialization]:
+    def run(
+        self, context: dg.AssetExecutionContext
+    ) -> Iterator[dg.AssetMaterialization]:
         metadata_by_key = context.assets_def.metadata_by_key
         first_asset_metadata = next(iter(metadata_by_key.values()))
 
@@ -52,16 +46,16 @@ def run(self, context: AssetExecutionContext) -> Iterator[AssetMaterialization]:
         results = replicate(Path(project.replication_configuration_yaml))
         for table in results:
             if table.get("status") == "SUCCESS":
-                yield AssetMaterialization(
+                yield dg.AssetMaterialization(
                     asset_key=translator.get_asset_key(table), metadata=table
                 )
 
 
 @dataclass
 class ReplicationTranslator:
     @public
-    def get_asset_key(self, table_definition: Mapping[str, str]) -> AssetKey:
-        return AssetKey(str(table_definition.get("name")))
+    def get_asset_key(self, table_definition: Mapping[str, str]) -> dg.AssetKey:
+        return dg.AssetKey(str(table_definition.get("name")))
 
 
 def custom_replication_assets(
@@ -70,19 +64,19 @@ def custom_replication_assets(
     name: Optional[str] = None,
     group_name: Optional[str] = None,
     translator: Optional[ReplicationTranslator] = None,
-) -> Callable[[Callable[..., Any]], AssetsDefinition]:
+) -> Callable[[Callable[..., Any]], dg.AssetsDefinition]:
     project = replication_project.load()
 
     translator = (
         check.opt_inst_param(translator, "translator", ReplicationTranslator)
         or ReplicationTranslator()
     )
 
-    return multi_asset(
+    return dg.multi_asset(
         name=name,
         group_name=group_name,
         specs=[
-            AssetSpec(
+            dg.AssetSpec(
                 key=translator.get_asset_key(table),
                 metadata={
                     "replication_project": project,

examples/docs_beta_snippets/docs_beta_snippets_tests/snippet_checks/guides/components/test_component_docs_existing_code_location/my-existing-project/my_existing_project/assets.py

@@ -1,6 +1,6 @@
-from dagster import asset
+import dagster as dg
 
 
-@asset
+@dg.asset
 def my_asset():
     pass

examples/docs_beta_snippets/docs_beta_snippets_tests/snippet_checks/guides/components/test_components_docs_migrating_definitions/my-existing-project/my_existing_project/analytics/assets.py

@@ -1,6 +1,6 @@
-from dagster import asset
+import dagster as dg
 
 
-@asset
+@dg.asset
 def my_analytics_asset():
     pass

examples/docs_beta_snippets/docs_beta_snippets_tests/snippet_checks/guides/components/test_components_docs_migrating_definitions/my-existing-project/my_existing_project/elt/assets.py

@@ -1,13 +1,13 @@
-from dagster import asset
+import dagster as dg
 
 
-@asset
+@dg.asset
 def customers_table(): ...
 
 
-@asset
+@dg.asset
 def orders_table(): ...
 
 
-@asset
+@dg.asset
 def products_table(): ...

examples/docs_beta_snippets/docs_beta_snippets_tests/test_all_files_load.py

@@ -3,9 +3,9 @@
 
 import pytest
 
-from dagster import file_relative_path
+import dagster as dg
 
-snippets_folder = file_relative_path(__file__, "../docs_beta_snippets/")
+snippets_folder = dg.file_relative_path(__file__, "../docs_beta_snippets/")
 
 EXCLUDED_FILES = {
     # see DOC-375

examples/docs_beta_snippets/docs_beta_snippets_tests/test_integration_files_load.py

@@ -3,9 +3,9 @@
 
 import pytest
 
-from dagster import file_relative_path
+import dagster as dg
 
-snippets_folder = file_relative_path(__file__, "../docs_beta_snippets/integrations")
+snippets_folder = dg.file_relative_path(__file__, "../docs_beta_snippets/integrations")
 
 EXCLUDED_FILES = {
     # exclude community integrations because they require non-editable dagster depdendencies