Add version parcing for mariadb utility

plesk · Jan 10, 2025 · 4f355d7 · 4f355d7
1 parent 9d318bf
commit 4f355d7
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 9 deletions.
diff --git a/pleskdistup/common/src/mariadb.py b/pleskdistup/common/src/mariadb.py
@@ -3,6 +3,7 @@
 import typing
 
 from . import dist, log
+import re
 
 
 def is_version_larger(left: str, right: str) -> bool:
@@ -56,6 +57,16 @@ def _extract_from_mysql_util(self, util_output: str):
         self.minor = int(minor_part)
         self.patch = int(patch_part)
 
+    def _extract_from_mariadb_util(self, util_output: str):
+        # String example: "mariadb from 11.6.2-MariaDB, client 15.2 for Linux (x86_64) using  EditLine wrapper"
+        match = re.search(r"mariadb from (\d+)\.(\d+)\.(\d+)-MariaDB", util_output)
+        if not match:
+            raise ValueError(f"Cannot extract mariadb version from '{util_output}'")
+        major_part, minor_part, patch_part = match.groups()
+        self.major = int(major_part)
+        self.minor = int(minor_part)
+        self.patch = int(patch_part)
+
     def __init__(self, to_extract: str):
         """Initialize a version object."""
         self.major = 0
@@ -66,8 +77,10 @@ def __init__(self, to_extract: str):
             self._extract_from_version_str(to_extract)
         elif "Distrib" in to_extract:
             self._extract_from_mysql_util(to_extract)
+        elif to_extract.startswith("mariadb"):
+            self._extract_from_mariadb_util(to_extract)
         else:
-            raise ValueError(f"Cannot extract php version from '{to_extract}'")
+            raise ValueError(f"Cannot extract mariadb version from '{to_extract}'")
 
     def __str__(self):
         """Return a string representation of a PHPVersion object."""

diff --git a/pleskdistup/common/tests/mariadbtests.py b/pleskdistup/common/tests/mariadbtests.py
@@ -6,16 +6,22 @@
 
 class MariaDBVersionsTests(unittest.TestCase):
     def test_parse_simple(self):
-        php = mariadb.MariaDBVersion("10.6.12")
-        self.assertEqual(php.major, 10)
-        self.assertEqual(php.minor, 6)
-        self.assertEqual(php.patch, 12)
+        mariadb_ver = mariadb.MariaDBVersion("10.6.12")
+        self.assertEqual(mariadb_ver.major, 10)
+        self.assertEqual(mariadb_ver.minor, 6)
+        self.assertEqual(mariadb_ver.patch, 12)
 
     def test_parse_utility_output(self):
-        php = mariadb.MariaDBVersion("mysql  Ver 15.1 Distrib 10.6.12-MariaDB, for debian-linux-gnu (x86_64) using  EditLine wrapper")
-        self.assertEqual(php.major, 10)
-        self.assertEqual(php.minor, 6)
-        self.assertEqual(php.patch, 12)
+        mariadb_ver = mariadb.MariaDBVersion("mysql  Ver 15.1 Distrib 10.6.12-MariaDB, for debian-linux-gnu (x86_64) using  EditLine wrapper")
+        self.assertEqual(mariadb_ver.major, 10)
+        self.assertEqual(mariadb_ver.minor, 6)
+        self.assertEqual(mariadb_ver.patch, 12)
+
+    def test_parse_utility_output_no_distrib(self):
+        mariadb_ver = mariadb.MariaDBVersion("mariadb from 11.6.2-MariaDB, client 15.2 for Linux (x86_64) using  EditLine wrapper")
+        self.assertEqual(mariadb_ver.major, 11)
+        self.assertEqual(mariadb_ver.minor, 6)
+        self.assertEqual(mariadb_ver.patch, 2)
 
     def test_parse_wrong_string(self):
         with self.assertRaises(ValueError):