update schema

Author: Kraysent

hyperleda/client.py

@@ -146,7 +146,7 @@ def patch_table_schema(self, table_name: str, actions: list) -> None:
             dataclasses.asdict(model.PatchTableRequestSchema(table_name, actions)),
         )
 
-    def create_homogenization_rules(self, rules: list[model.HomogenizationRule]) -> None:
+    def create_homogenization_rules(self, rules: list[model.Catalog]) -> None:
         _ = self._request(
             "POST",
             "/admin/api/v1/homogenization/rules",

hyperleda/model.py

@@ -1,17 +1,20 @@
 # generated by datamodel-codegen:
 #   filename:  swagger.json
-#   timestamp: 2025-05-07T17:31:13+00:00
+#   timestamp: 2025-05-08T08:27:46+00:00
 
 from __future__ import annotations
 
 from dataclasses import dataclass
 from enum import Enum
 from typing import Any
 
+int_aliased = int
+str_aliased = str
+
 
 @dataclass
 class AddDataRequestSchema:
-    table_id: int
+    table_id: int_aliased
     data: list[dict[str, Any]]
 
 
@@ -22,14 +25,14 @@ class AddDataResponseSchema:
 
 @dataclass
 class CreateSourceRequestSchema:
-    title: str
-    authors: list[str]
-    year: int
+    title: str_aliased
+    authors: list[str_aliased]
+    year: int_aliased
 
 
 @dataclass
 class CreateSourceResponseSchema:
-    code: str
+    code: str_aliased
 
 
 class DataType(Enum):
@@ -53,11 +56,11 @@ class DataType(Enum):
 
 @dataclass
 class ColumnDescription:
-    name: str
+    name: str_aliased
     data_type: DataType
-    unit: str | None = None
-    ucd: str | None = None
-    description: str | None = ""
+    unit: str_aliased | None = None
+    ucd: str_aliased | None = None
+    description: str_aliased | None = ""
 
 
 class Datatype(Enum):
@@ -69,27 +72,27 @@ class Datatype(Enum):
 
 @dataclass
 class CreateTableRequestSchema:
-    table_name: str
+    table_name: str_aliased
     columns: list[ColumnDescription]
-    bibcode: str
+    bibcode: str_aliased
     datatype: Datatype | None = Datatype.regular
-    description: str | None = ""
+    description: str_aliased | None = ""
 
 
 @dataclass
 class CreateTableResponseSchema:
-    id: int
+    id: int_aliased
 
 
 @dataclass
 class GetTableValidationRequestSchema:
-    table_name: str
+    table_name: str_aliased
 
 
 @dataclass
 class TableValidation:
-    message: str | None = None
-    validator: str | None = None
+    message: str_aliased | None = None
+    validator: str_aliased | None = None
 
 
 @dataclass
@@ -104,7 +107,7 @@ class PatchTableAction:
 
 @dataclass
 class PatchTableRequestSchema:
-    table_name: str
+    table_name: str_aliased
     actions: list[PatchTableAction]
 
 
@@ -115,24 +118,24 @@ class PatchTableResponseSchema:
 
 @dataclass
 class GetTaskInfoRequestSchema:
-    task_id: int
+    task_id: int_aliased
 
 
 @dataclass
 class GetTaskInfoResponseSchema:
-    id: int | None = None
-    task_name: str | None = None
-    status: str | None = None
+    id: int_aliased | None = None
+    task_name: str_aliased | None = None
+    status: str_aliased | None = None
     payload: dict[str, Any] | None = None
-    start_time: str | None = None
-    end_time: str | None = None
+    start_time: str_aliased | None = None
+    end_time: str_aliased | None = None
     message: dict[str, Any] | None = None
 
 
 @dataclass
 class LoginRequestSchema:
-    username: str
-    password: str
+    username: str_aliased
+    password: str_aliased
 
 
 @dataclass
@@ -176,32 +179,37 @@ class TableProcessResponseSchema:
 
 @dataclass
 class TableStatusStatsRequestSchema:
-    table_id: int
+    table_id: int_aliased
 
 
 @dataclass
 class TableStatusStatsResponseSchema:
-    processing: dict[str, int] | None = None
+    processing: dict[str, int_aliased] | None = None
+
+
+@dataclass
+class Parameter:
+    filters: dict[str, Any]
+    enrichment: dict[str, Any] | None = None
 
 
-class Catalog(Enum):
+class Name(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
+class Catalog:
+    name: Name
+    parameters: dict[str, Parameter]
+    key: str_aliased | None = None
+    additional_params: dict[str, Any] | None = None
 
 
 @dataclass
 class CreateHomogenizationRulesRequestSchema:
-    rules: list[HomogenizationRule]
+    catalogs: list[Catalog]
 
 
 @dataclass