added homogenization rules handler

Author: Kraysent

hyperleda/client.py

@@ -63,7 +63,11 @@ def _request(
 
         response = requests.request(method, f"{self.endpoint}{path}", **kwargs)
         if not response.ok:
-            raise error.APIError.from_dict(response.json())
+            try:
+                msg = error.APIError.from_dict(response.json())
+            except Exception:
+                msg = error.APIError(response.status_code, response.text)
+            raise msg
 
         return response.json()
 
@@ -156,3 +160,10 @@ def patch_table_schema(self, table_name: str, actions: list) -> None:
             "/admin/api/v1/table",
             dataclasses.asdict(model.PatchTableRequestSchema(table_name, actions)),
         )
+
+    def create_homogenization_rules(self, rules: list[model.HomogenizationRule]) -> None:
+        _ = self._request(
+            "POST",
+            "/admin/api/v1/table/homogenization/rules",
+            dataclasses.asdict(model.CreateHomogenizationRulesRequestSchema(rules)),
+        )

hyperleda/model.py

@@ -1,12 +1,12 @@
 # generated by datamodel-codegen:
 #   filename:  swagger.json
-#   timestamp: 2025-02-02T15:14:49+00:00
+#   timestamp: 2025-05-07T17:31:13+00:00
 
 from __future__ import annotations
 
 from dataclasses import dataclass
 from enum import Enum
-from typing import Any, Optional
+from typing import Any
 
 
 @dataclass
@@ -41,6 +41,7 @@ class DataType(Enum):
     char = "char"
     short = "short"
     int = "int"
+    long = "long"
     integer = "integer"
     smallint = "smallint"
     float = "float"
@@ -82,7 +83,7 @@ class CreateTableResponseSchema:
 
 @dataclass
 class GetTableValidationRequestSchema:
-    table_id: int
+    table_name: str
 
 
 @dataclass
@@ -181,3 +182,28 @@ class TableStatusStatsRequestSchema:
 @dataclass
 class TableStatusStatsResponseSchema:
     processing: dict[str, int] | None = None
+
+
+class Catalog(Enum):
+    icrs = "icrs"
+    designation = "designation"
+    redshift = "redshift"
+
+
+@dataclass
+class HomogenizationRule:
+    catalog: Catalog
+    parameter: str
+    filters: dict[str, Any]
+    key: str | None = None
+    enrichment: dict[str, Any] | None = None
+
+
+@dataclass
+class CreateHomogenizationRulesRequestSchema:
+    rules: list[HomogenizationRule]
+
+
+@dataclass
+class CreateHomogenizationRulesResponseSchema:
+    pass

makefile

@@ -13,7 +13,7 @@ build:
 # to use the model one needs to start the server on localhost:8080
 generate-model:
 	mkdir -p hyperleda/gen
-	curl localhost:8080/api/docs/swagger.json > hyperleda/gen/swagger.json
+	curl http://localhost:8080/admin/api/docs/swagger.json > hyperleda/gen/swagger.json
 	datamodel-codegen --input hyperleda/gen/swagger.json --output hyperleda/model.py --output-model-type dataclasses.dataclass --input-file-type openapi
 	make fix