add main clause

dlt-hub · Oct 2, 2023 · c6a503c · c6a503c
1 parent ceb3fa1
commit c6a503c
Show file tree

Hide file tree

Showing 4 changed files with 36 additions and 30 deletions.
diff --git a/docs/examples/transformers/pokemon.py b/docs/examples/transformers/pokemon.py
@@ -52,11 +52,12 @@ def species(pokemon: TDataItem):
 def source():
     return [pokemon_list | pokemon, pokemon_list | pokemon | species]
 
-# build duck db pipeline
-pipeline = dlt.pipeline(
-    pipeline_name="pokemon", destination="duckdb", dataset_name="pokemon_data"
-)
-
-# the pokemon_list resource does not need to be loaded
-load_info = pipeline.run(source())
-print(load_info)
+if __name__ == "__main__":
+    # build duck db pipeline
+    pipeline = dlt.pipeline(
+        pipeline_name="pokemon", destination="duckdb", dataset_name="pokemon_data"
+    )
+
+    # the pokemon_list resource does not need to be loaded
+    load_info = pipeline.run(source())
+    print(load_info)
diff --git a/docs/website/docs/examples/incremental_loading/code/zendesk-snippets.py b/docs/website/docs/examples/incremental_loading/code/zendesk-snippets.py
@@ -120,13 +120,15 @@ def get_pages(
 
 
     # @@@DLT_SNIPPET_START markdown_pipeline
-    # create dlt pipeline
-    pipeline = dlt.pipeline(
-        pipeline_name="zendesk", destination="duckdb", dataset_name="zendesk_data"
-    )
-
-    load_info = pipeline.run(zendesk_support())
-    print(load_info)
+    __name__ = "__main__" # @@@DLT_REMOVE
+    if __name__ == "__main__":
+        # create dlt pipeline
+        pipeline = dlt.pipeline(
+            pipeline_name="zendesk", destination="duckdb", dataset_name="zendesk_data"
+        )
+
+        load_info = pipeline.run(zendesk_support())
+        print(load_info)
     # @@@DLT_SNIPPET_END markdown_pipeline
     # @@@DLT_SNIPPET_END example
 

diff --git a/docs/website/docs/examples/transformers/code/pokemon-snippets.py b/docs/website/docs/examples/transformers/code/pokemon-snippets.py
@@ -56,14 +56,16 @@ def species(pokemon: TDataItem):
     def source():
         return [pokemon_list | pokemon, pokemon_list | pokemon | species]
 
-    # build duck db pipeline
-    pipeline = dlt.pipeline(
-        pipeline_name="pokemon", destination="duckdb", dataset_name="pokemon_data"
-    )
+    __name__ = "__main__" # @@@DLT_REMOVE
+    if __name__ == "__main__":
+        # build duck db pipeline
+        pipeline = dlt.pipeline(
+            pipeline_name="pokemon", destination="duckdb", dataset_name="pokemon_data"
+        )
 
-    # the pokemon_list resource does not need to be loaded
-    load_info = pipeline.run(source())
-    print(load_info)
+        # the pokemon_list resource does not need to be loaded
+        load_info = pipeline.run(source())
+        print(load_info)
     # @@@DLT_SNIPPET_END example
 
     # test assertions

diff --git a/docs/website/docs/examples/transformers/index.md b/docs/website/docs/examples/transformers/index.md
@@ -81,14 +81,15 @@ def species(pokemon: TDataItem):
 def source():
     return [pokemon_list | pokemon, pokemon_list | pokemon | species]
 
-# build duck db pipeline
-pipeline = dlt.pipeline(
-    pipeline_name="pokemon", destination="duckdb", dataset_name="pokemon_data"
-)
-
-# the pokemon_list resource does not need to be loaded
-load_info = pipeline.run(source())
-print(load_info)
+if __name__ == "__main__":
+    # build duck db pipeline
+    pipeline = dlt.pipeline(
+        pipeline_name="pokemon", destination="duckdb", dataset_name="pokemon_data"
+    )
+
+    # the pokemon_list resource does not need to be loaded
+    load_info = pipeline.run(source())
+    print(load_info)
 ```
 <!--@@@DLT_SNIPPET_END ./code/pokemon-snippets.py::example-->