cmsdials 0.0.0__py3-none-any.whl

cmsdials/utils/base_model.py

@@ -0,0 +1,33 @@
+from importlib import util as importlib_util
+from typing import Optional
+
+from pydantic import BaseModel
+
+
+if importlib_util.find_spec("pandas"):
+    import pandas as pd
+
+    PANDAS_NOT_INSTALLED = False
+else:
+    PANDAS_NOT_INSTALLED = True
+
+
+class OBaseModel(BaseModel):
+    def cleandict(self):
+        return {
+            key: ",".join(str(v) for v in value) if isinstance(value, (list, tuple)) else value
+            for key, value in self.model_dump().items()
+            if value is not None
+        }
+
+
+class PaginatedBaseModel(BaseModel):
+    exc_type: Optional[str] = None
+    exc_formatted: Optional[str] = None
+
+    def to_pandas(self):
+        if PANDAS_NOT_INSTALLED:
+            raise RuntimeError(
+                "The 'pandas' package is not installed, you can re-install cmsdials specifying the pandas extra: pip install cmsdials[pandas]"
+            )
+        return pd.DataFrame([res.__dict__ for res in self.results])

cmsdials/utils/logger.py

@@ -0,0 +1,6 @@
+import logging
+
+
+logging.basicConfig(format="[%(asctime)s] %(levelname)s: %(message)s", level=logging.INFO)
+
+logger = logging.getLogger(__name__)