InfoTracker 0.3.1__py3-none-any.whl → 0.4.0__py3-none-any.whl

infotracker/models.py

@@ -29,6 +29,8 @@ class TransformationType(Enum):
     DATE_FUNCTION_AGGREGATION = "DATE_FUNCTION_AGGREGATION"
     CASE_AGGREGATION = "CASE_AGGREGATION"
     EXEC = "EXEC"
+    CONSTANT = "CONSTANT"
+    UNKNOWN = "UNKNOWN"
 
 
 @dataclass
@@ -83,6 +85,8 @@ class ObjectInfo:
     schema: TableSchema
     lineage: List[ColumnLineage] = field(default_factory=list)
     dependencies: Set[str] = field(default_factory=set)  # Tables this object depends on
+    is_fallback: bool = field(default=False)  # Whether this was created by fallback parsing
+    no_output_reason: Optional[str] = field(default=None)  # Reason for no persistent output
 
 
 class SchemaRegistry:

infotracker/openlineage_utils.py

@@ -163,3 +163,19 @@ def qualify_identifier(identifier: str, default_database: Optional[str] = None)
     else:
         # Already fully qualified
         return identifier
+
+
+def sanitize_name(name: str) -> str:
+    """Sanitize object name by removing trailing semicolons and whitespace.
+    
+    Args:
+        name: Object name to sanitize
+        
+    Returns:
+        Sanitized name
+    """
+    if not name:
+        return name
+    
+    # Remove trailing semicolons and whitespace
+    return name.rstrip(';').strip()