Add an optional table name postfix to the metadata builder

python/felis/metadata.py

@@ -134,6 +134,7 @@ def __init__(
         schema: Schema,
         apply_schema_to_metadata: bool = True,
         ignore_constraints: bool = False,
+        table_name_postfix: str = "",
     ) -> None:
         """Initialize the metadata builder."""
         self.schema = schema
@@ -142,6 +143,7 @@ def __init__(
         self.metadata = MetaData(schema=schema.name if apply_schema_to_metadata else None)
         self._objects: dict[str, Any] = {}
         self.ignore_constraints = ignore_constraints
+        self.table_name_postfix = table_name_postfix
 
     def build(self) -> MetaData:
         """Build the SQLAlchemy tables and constraints from the schema.
@@ -225,7 +227,7 @@ def build_table(self, table_obj: datamodel.Table) -> None:
         description = table_obj.description
         columns = [self.build_column(column) for column in table_obj.columns]
         table = Table(
-            name,
+            name + self.table_name_postfix,
             self.metadata,
             *columns,
             comment=description,

python/felis/tap_schema.py

@@ -91,9 +91,15 @@ def __init__(
         self.table_name_postfix = table_name_postfix
         self.apply_schema_to_metadata = apply_schema_to_metadata
         self.schema_name = schema_name or TableManager._SCHEMA_NAME_STD
+        self.table_name_postfix = table_name_postfix
 
         if is_valid_engine(engine):
             assert isinstance(engine, Engine)
+            if table_name_postfix != "":
+                logger.warning(
+                    "Table name postfix '%s' will be ignored when reflecting TAP_SCHEMA database",
+                    table_name_postfix,
+                )
             logger.debug(
                 "Reflecting TAP_SCHEMA database from existing database at %s",
                 engine.url._replace(password="***"),
@@ -131,7 +137,9 @@ def _load_yaml(self) -> None:
             self.schema_name = self.schema.name
 
         self._metadata = MetaDataBuilder(
-            self.schema, apply_schema_to_metadata=self.apply_schema_to_metadata
+            self.schema,
+            apply_schema_to_metadata=self.apply_schema_to_metadata,
+            table_name_postfix=self.table_name_postfix,
         ).build()
 
         logger.debug("Loaded TAP_SCHEMA '%s' from YAML resource", self.schema_name)

tests/test_metadata.py

@@ -231,6 +231,15 @@ def test_ignore_constraints(self) -> None:
                 msg=f"Table {table.name} has non-primary key constraints defined",
             )
 
+    def test_table_name_postfix(self) -> None:
+        """Test that table name postfixes are correctly applied."""
+        schema = Schema.model_validate(self.yaml_data)
+        schema.name = "main"
+        builder = MetaDataBuilder(schema, table_name_postfix="_test")
+        md = builder.build()
+        for table in md.tables.values():
+            self.assertTrue(table.name.endswith("_test"))
+
 
 if __name__ == "__main__":
     unittest.main()

tests/test_tap_schema.py

@@ -60,6 +60,12 @@ def test_create_table_manager(self) -> None:
         # already been created.
         mgr = TableManager()
 
+    def test_table_name_postfix(self) -> None:
+        """Test the table name postfix."""
+        mgr = TableManager(table_name_postfix="_test")
+        for table_name in mgr.metadata.tables:
+            self.assertTrue(table_name.endswith("_test"))
+
 
 class DataLoaderTestCase(unittest.TestCase):
     """Test the `DataLoader` class."""