modelbase2 0.1.79__py3-none-any.whl → 0.2.0__py3-none-any.whl

modelbase2/sbml/_name_conversion.py

@@ -0,0 +1,52 @@
+from __future__ import annotations
+
+import keyword
+import re
+
+__all__ = ["RE_FROM_SBML", "RE_KWDS", "SBML_DOT"]
+
+RE_KWDS = re.compile("|".join(f"^{i}$" for i in keyword.kwlist))
+SBML_DOT = "__SBML_DOT__"
+RE_FROM_SBML = re.compile(r"__(\d+)__")
+
+
+def _escape_keyword(re_sub: re.Match) -> str:
+    return f"{re_sub.group(0)}_"
+
+
+def _ascii_to_character(re_sub: re.Match) -> str:
+    """Convert an escaped non-alphanumeric character."""
+    return chr(int(re_sub.group(1)))
+
+
+def _name_to_py(name: str) -> str:
+    name = RE_FROM_SBML.sub(_ascii_to_character, name)
+    name = RE_KWDS.sub(_escape_keyword, name)
+    name = (
+        name.replace(SBML_DOT, ".")
+        .replace(" ", "_")
+        .replace("-", "_")
+        .replace("(", "")
+        .replace(")", "")
+        .replace("[", "")
+        .replace("]", "")
+        .replace(".", "")
+        .replace(",", "")
+        .replace(":", "")
+        .replace(";", "")
+        .replace('"', "")
+        .replace("'", "")
+        .replace("^", "")
+        .replace("|", "")
+        .replace("=", "eq")
+        .replace(">", "lg")
+        .replace("<", "sm")
+        .replace("+", "plus")
+        .replace("-", "minus")
+        .replace("*", "star")
+        .replace("/", "div")
+        # .lower()
+    )
+    if not name[0].isalpha():
+        return f"_{name}"
+    return name

modelbase2/sbml/_unit_conversion.py

@@ -0,0 +1,74 @@
+from __future__ import annotations
+
+__all__ = ["get_ast_types", "get_operator_mappings", "get_unit_conversion"]
+
+
+import libsbml
+
+unit_conversion = {
+    "UNIT_KIND_AMPERE": "AMPERE",
+    "UNIT_KIND_AVOGADRO": "AVOGADRO",
+    "UNIT_KIND_BECQUEREL": "BECQUEREL",
+    "UNIT_KIND_CANDELA": "CANDELA",
+    "UNIT_KIND_CELSIUS": "CELSIUS",
+    "UNIT_KIND_COULOMB": "COULOMB",
+    "UNIT_KIND_DIMENSIONLESS": "DIMENSIONLESS",
+    "UNIT_KIND_FARAD": "FARAD",
+    "UNIT_KIND_GRAM": "GRAM",
+    "UNIT_KIND_GRAY": "GRAY",
+    "UNIT_KIND_HENRY": "HENRY",
+    "UNIT_KIND_HERTZ": "HERTZ",
+    "UNIT_KIND_ITEM": "ITEM",
+    "UNIT_KIND_JOULE": "JOULE",
+    "UNIT_KIND_KATAL": "KATAL",
+    "UNIT_KIND_KELVIN": "KELVIN",
+    "UNIT_KIND_KILOGRAM": "KILOGRAM",
+    "UNIT_KIND_LITER": "LITER",
+    "UNIT_KIND_LITRE": "LITER",
+    "UNIT_KIND_LUMEN": "LUMEN",
+    "UNIT_KIND_LUX": "LUX",
+    "UNIT_KIND_METER": "METER",
+    "UNIT_KIND_METRE": "METRE",
+    "UNIT_KIND_MOLE": "MOLE",
+    "UNIT_KIND_NEWTON": "NEWTON",
+    "UNIT_KIND_OHM": "OHM",
+    "UNIT_KIND_PASCAL": "PASCAL",
+    "UNIT_KIND_RADIAN": "RADIAN",
+    "UNIT_KIND_SECOND": "SECOND",
+    "UNIT_KIND_SIEMENS": "SIEMENS",
+    "UNIT_KIND_SIEVERT": "SIEVERT",
+    "UNIT_KIND_STERADIAN": "STERADIAN",
+    "UNIT_KIND_TESLA": "TESLA",
+    "UNIT_KIND_VOLT": "VOLT",
+    "UNIT_KIND_WATT": "WATT",
+    "UNIT_KIND_WEBER": "WEBER",
+    "UNIT_KIND_INVALID": "INVALID",
+}
+
+
+def get_unit_conversion() -> dict[str, str]:
+    return {getattr(libsbml, k): v.lower() for k, v in unit_conversion.items()}
+
+
+ast_types = dict(
+    sorted(
+        {getattr(libsbml, i): i for i in dir(libsbml) if i.startswith("AST_")}.items()
+    )
+)
+
+
+def get_ast_types() -> dict[str, str]:
+    return ast_types
+
+
+operator_mappings = {
+    "AST_TIMES": "*",
+    "AST_PLUS": "+",
+    "AST_MINUS": "-",
+    "AST_DIVIDE": "/",
+    "AST_POWER": "**",
+}
+
+
+def get_operator_mappings() -> dict[str, str]:
+    return operator_mappings