Add version parcing for mariadb utility

pleskdistup/common/src/mariadb.py

@@ -56,6 +56,13 @@ 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"
+        major_part, minor_part, patch_part = util_output.split("from ")[1].split("-MariaDB")[0].split(".")[:3]
+        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 +73,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."""

pleskdistup/common/tests/mariadbtests.py

@@ -17,6 +17,12 @@ def test_parse_utility_output(self):
         self.assertEqual(php.minor, 6)
         self.assertEqual(php.patch, 12)
 
+    def test_parse_utility_output_no_distrib(self):
+        php = mariadb.MariaDBVersion("mariadb from 11.6.2-MariaDB, client 15.2 for Linux (x86_64) using  EditLine wrapper")
+        self.assertEqual(php.major, 11)
+        self.assertEqual(php.minor, 6)
+        self.assertEqual(php.patch, 2)
+
     def test_parse_wrong_string(self):
         with self.assertRaises(ValueError):
             mariadb.MariaDBVersion("nothing")