1 files changed, 23 insertions, 0 deletions
diff --git a/python/cru/system.py b/python/cru/system.py
new file mode 100644
index 0000000..f321717
--- /dev/null
+++ b/python/cru/system.py
@@ -0,0 +1,23 @@
+import os.path
+import re
+
+
+def check_debian_derivative_version(name: str) -> None | str:
+    if not os.path.isfile("/etc/os-release"):
+        return None
+    with open("/etc/os-release", "r") as f:
+        content = f.read()
+        if f"ID={name}" not in content:
+            return None
+        m = re.search(r'VERSION_ID="(.+)"', content)
+        if m is None:
+            return None
+        return m.group(1)
+
+
+def check_ubuntu_version() -> None | str:
+    return check_debian_derivative_version("ubuntu")
+
+
+def check_debian_version() -> None | str:
+    return check_debian_derivative_version("debian")