sws-spark-dissemination-helper 0.0.95__py3-none-any.whl → 0.0.96__py3-none-any.whl

sws_spark_dissemination_helper/SWSBronzeIcebergSparkHelper.py

@@ -1,4 +1,5 @@
 import logging
+import time
 from copy import copy
 from typing import Dict, List, Tuple
 
@@ -389,6 +390,15 @@ class SWSBronzeIcebergSparkHelper:
         self.spark.sql(
             f"ALTER TABLE {self.iceberg_tables.BRONZE.iceberg_id} CREATE TAG `{self.tag_name}`"
         )
+        while (
+            self.spark.sql(
+                f"SELECT * FROM {self.iceberg_tables.BRONZE.iceberg_id}.refs"
+            )
+            .filter((col("type") == lit("TAG")) & (col("name") == lit(self.tag_name)))
+            .count()
+        ) == 0:
+            logging.info(f"Waiting for the tag {self.tag_name} to be created")
+            time.sleep(2)
 
         logging.info(f"bronze tag '{self.tag_name}' created")
 
@@ -447,14 +457,31 @@ class SWSBronzeIcebergSparkHelper:
     def write_bronze_disseminated_tag_data_to_iceberg_and_csv(
         self, dimensions: Dict[str, List[str]]
     ) -> DataFrame:
-        
-        refs = self.spark.sql(f"SELECT * FROM {self.iceberg_tables.BRONZE.iceberg_id}.refs").collect()
+
+        refs = self.spark.sql(
+            f"SELECT * FROM {self.iceberg_tables.BRONZE.iceberg_id}.refs"
+        ).collect()
         logging.info(f"bronze refs: {refs}")
 
         create_branch_query = f"ALTER TABLE {self.iceberg_tables.BRONZE.iceberg_id}.`tag_{self.tag_name}` CREATE OR REPLACE BRANCH `diss_tag_{self.tag_name}`"  # AS OF VERSION `{tag_name}`
         logging.info(f"create_branch_query: {create_branch_query}")
         create_branch_query_result = self.spark.sql(create_branch_query).collect()
 
+        while (
+            self.spark.sql(
+                f"SELECT * FROM {self.iceberg_tables.BRONZE.iceberg_id}.refs"
+            )
+            .filter(
+                (col("type") == lit("BRANCH"))
+                & (col("name") == lit(f"diss_tag_{self.tag_name}"))
+            )
+            .count()
+        ) == 0:
+            logging.info(
+                f"Waiting for the branch {self.tag_name} diss_tag_{self.tag_name} to be created"
+            )
+            time.sleep(2)
+
         logging.info(f"result of create_branch_query: {create_branch_query_result}")
 
         self.disseminated_tag_df = self.spark.read.option(

{sws_spark_dissemination_helper-0.0.95.dist-info → sws_spark_dissemination_helper-0.0.96.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sws-spark-dissemination-helper
-Version: 0.0.95
+Version: 0.0.96
 Summary: A Python helper package providing streamlined Spark functions for efficient data dissemination processes
 Project-URL: Repository, https://bitbucket.org/cioapps/sws-it-python-spark-dissemination-helper
 Author-email: Daniele Mansillo <danielemansillo@gmail.com>

{sws_spark_dissemination_helper-0.0.95.dist-info → sws_spark_dissemination_helper-0.0.96.dist-info}/RECORD

@@ -1,11 +1,11 @@
-sws_spark_dissemination_helper/SWSBronzeIcebergSparkHelper.py,sha256=ja7zXIJNYdzvSiHdMB8oeKSQkLVu-FSUhy0vn_myhnU,21086
+sws_spark_dissemination_helper/SWSBronzeIcebergSparkHelper.py,sha256=R9FsWyoX_iE2re5BUXer6QGDR45LJ_Z4P4lLET9Ij88,21964
 sws_spark_dissemination_helper/SWSGoldIcebergSparkHelper.py,sha256=ZC7hxkppo6qmfCc2z5vm2Y2iH1901F-rx9Er9cxuzP4,16037
 sws_spark_dissemination_helper/SWSPostgresSparkReader.py,sha256=ja7AbOfbmC_EXHCJk7UMDzzbA-LRxzPkaaUmuvcihJ8,17449
 sws_spark_dissemination_helper/SWSSilverIcebergSparkHelper.py,sha256=zEppNq5shiHZH2yt5faWGsb5QEmpAQS0ToIrG6fmv6o,22231
 sws_spark_dissemination_helper/__init__.py,sha256=Efjoe9V4vGXWVp-DY5P6NbRwIUr_zkZJkDmMi-lf5Bc,262
 sws_spark_dissemination_helper/constants.py,sha256=hpHHlbojShMWRfyIelXz6c5BqFzO48Oap1zmztlMMrs,11349
 sws_spark_dissemination_helper/utils.py,sha256=6SzrXX0xhvynRyv-vRFDbc6V4UNe_RzKKETZAtefnhg,21341
-sws_spark_dissemination_helper-0.0.95.dist-info/METADATA,sha256=5oamHa9qI0T6_rNyZc4NLUM4kPUaCGfmHOypaNyky8s,2823
-sws_spark_dissemination_helper-0.0.95.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-sws_spark_dissemination_helper-0.0.95.dist-info/licenses/LICENSE,sha256=zFzeb_j_6pXEHwH8Z0OpIkKFJk7vmhZjdem-K0d4zU4,1073
-sws_spark_dissemination_helper-0.0.95.dist-info/RECORD,,
+sws_spark_dissemination_helper-0.0.96.dist-info/METADATA,sha256=ugZY_gSFF51c4V65tLQAe4ERI8Cu-WgBY2VxuWJB3Ow,2823
+sws_spark_dissemination_helper-0.0.96.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+sws_spark_dissemination_helper-0.0.96.dist-info/licenses/LICENSE,sha256=zFzeb_j_6pXEHwH8Z0OpIkKFJk7vmhZjdem-K0d4zU4,1073
+sws_spark_dissemination_helper-0.0.96.dist-info/RECORD,,