fix(mysql): add odd_metadata property to Column entity: (#215)

opendatadiscovery · Sep 4, 2023 · 423b557 · 423b557
1 parent 1658dcc
commit 423b557
Show file tree

Hide file tree

Showing 4 changed files with 39 additions and 15 deletions.
diff --git a/odd_collector/models/column.py b/odd_collector/models/column.py
@@ -1,9 +1,11 @@
 from dataclasses import dataclass, field
 from typing import Any, Optional
 
+from odd_collector_sdk.utils.metadata import HasMetadata
+
 
 @dataclass(frozen=True)
-class Column:
+class Column(HasMetadata):
     table_catalog: str
     table_schema: str
     table_name: str
@@ -15,3 +17,7 @@ class Column:
     is_primary_key: bool = False
     is_sort_key: bool = False
     metadata: dict[str, Any] = field(default_factory=dict)
+
+    @property
+    def odd_metadata(self):
+        return self.metadata
diff --git a/odd_collector/models/table.py b/odd_collector/models/table.py
@@ -1,13 +1,15 @@
 import dataclasses
 from typing import Any, Optional
 
+from odd_collector_sdk.utils.metadata import HasMetadata
+
 from odd_collector.helpers.datetime import Datetime
 
 from .column import Column
 
 
 @dataclasses.dataclass
-class Table:
+class Table(HasMetadata):
     catalog: str
     schema: str
     name: str

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -48,7 +48,7 @@ llvmlite = "^0.39.1"
 numba = "^0.56.4"
 mlflow = "^2.2.2"
 sql-metadata = "^2.6.0"
-odd-collector-sdk = "^0.3.42"
+odd-collector-sdk = "^0.3.43"
 clickhouse-connect = "^0.5.14"
 odd-models = "^2.0.28"
 couchbase = "^4.1.3"