Fix description detection

H-Max · Nov 22, 2023 · e2528a5 · e2528a5
1 parent c223a06
commit e2528a5
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 7 deletions.
diff --git a/.gitignore b/.gitignore
@@ -12,6 +12,7 @@ _local
 
 # Output directory of the package itself
 **/output
+**/target/bq2dbt
 
 # Distribution / packaging
 build/

diff --git a/bq2dbt/__init__.py b/bq2dbt/__init__.py
@@ -5,5 +5,9 @@
 and basic select query with explicit columns list
 """
 
-__version__ = "0.1.0"
-__author__ = 'Henri-Maxime Ducoulombier'
+from .bq2dbt import bq2dbt
+
+
+def main():
+    """Entry point for the application script"""
+    bq2dbt()
diff --git a/bq2dbt/bq2dbt.py b/bq2dbt/bq2dbt.py
@@ -125,7 +125,7 @@ def bq2dbt():
         field_info = {
             "name": destination,
             "data_type": data_type.lower() if args.lower else data_type,
-            "description": field.description
+            "description": field.description or ""
         }
         if field.is_nullable == 'NO':
             field_info = {**field_info, **{
@@ -165,7 +165,3 @@ def bq2dbt():
         sql_file.write(sql_output.strip())
 
     logger.info("Operation complete")
-
-
-if __name__ == '__main__':
-    bq2dbt()
diff --git a/pyproject.toml b/pyproject.toml
@@ -19,6 +19,9 @@ dependencies = [
 ]
 dynamic = ["version"]
 
+[project.scripts]  # Optional
+bq2dbt = "bq2dbt:bq2dbt"
+
 [tool.setuptools]
 packages = ["bq2dbt"]