Merge pull request #5 from geo-engine/ui

feat: UI

Author: michaelmattig

api-client/post-process.py

@@ -16,16 +16,22 @@
 
 
 def file_modifications() -> Generator[tuple[Path, FileModifier], None, None]:
+    """Return a generator of file paths and their corresponding modification functions."""
+
     yield Path("types/ObjectParamAPI.ts"), object_param_api_ts
     yield Path("types/PromiseAPI.ts"), promise_api_ts
+    yield Path("models/ObjectSerializer.ts"), object_serializer_ts
+    yield Path("models/Results.ts"), results_ts
 
 
 def main():
+    """Main function to perform file modifications."""
+
     logging.basicConfig(level=logging.INFO, format="[%(levelname)s] %(message)s")
 
     subdir = Path("typescript")
     for file_path, modify_fn in file_modifications():
-        logging.info(f"Modifying {file_path}…")
+        logging.info("Modifying %s…", file_path)
 
         file_path = subdir / file_path
 
@@ -36,8 +42,8 @@ def main():
             with open(file_path, "w", encoding="utf-8") as f:
                 for line in modify_fn(file_contents):
                     f.write(line)
-        except Exception as e:
-            logging.error(f"Error modifying {file_path}: {e}")
+        except Exception as e:  # pylint: disable=broad-exception-caught
+            logging.error("Error modifying %s: %s", file_path, e)
 
 
 def object_param_api_ts(file_contents: list[str]) -> Generator[str, None, None]:
@@ -56,5 +62,21 @@ def promise_api_ts(file_contents: list[str]) -> Generator[str, None, None]:
         yield line
 
 
+def results_ts(file_contents: list[str]) -> Generator[str, None, None]:
+    """Modify the Results.ts file."""
+    for line in file_contents:
+        if dedent(line).startswith("import { HttpFile } from '../http/http';"):
+            line = line + "import { InlineOrRefData } from './InlineOrRefData';\n"
+        yield line
+
+
+def object_serializer_ts(file_contents: list[str]) -> Generator[str, None, None]:
+    """Modify the ObjectSerializer.ts file."""
+    for line in file_contents:
+        if dedent(line).startswith('"Results": ResultsClass,'):
+            line = ""  # lead to call of missing `getAttributeTypeMap` method
+        yield line
+
+
 if __name__ == "__main__":
     main()