From c3e26ce08967d1c998b77d1e8377d44226a5fb41 Mon Sep 17 00:00:00 2001
From: Mikhail Sandakov <msandakov@plesk.com>
Date: Tue, 2 Jan 2024 15:22:17 +0200
Subject: [PATCH] Support vz format of the kernel version information

Unfortiantly it have no information about arch, so we have to skip it.
---
 pleskdistup/common/src/version.py        | 17 +++++++++++++----
 pleskdistup/common/tests/versiontests.py |  3 +++
 2 files changed, 16 insertions(+), 4 deletions(-)

diff --git a/pleskdistup/common/src/version.py b/pleskdistup/common/src/version.py
index 0b15e62..37a4d29 100644
--- a/pleskdistup/common/src/version.py
+++ b/pleskdistup/common/src/version.py
@@ -19,7 +19,11 @@ def _extract_with_build(self, version: str) -> None:
             if secondary_part[iter].isalpha():
                 self.build = secondary_part[:iter - 1]
                 suffix = secondary_part[iter:]
-                self.distro, self.arch = suffix.split(".")
+                # There is no information about arch when we have vzX suffix
+                if suffix.startswith("vz"):
+                    self.distro = suffix.split(".")[0]
+                else:
+                    self.distro, self.arch = suffix.split(".")
                 break
 
     def _extract_no_build(self, version: str) -> None:
@@ -45,10 +49,15 @@ def __repr__(self) -> str:
         return f"{self.__class__.__name__}({attrs})"
 
     def __str__(self) -> str:
-        if self.build == "":
-            return f"{self.major}.{self.minor}.{self.patch}.{self.distro}.{self.arch}"
+        result = f"{self.major}.{self.minor}.{self.patch}"
+        if self.build != "":
+            result += f"-{self.build}"
 
-        return f"{self.major}.{self.minor}.{self.patch}-{self.build}.{self.distro}.{self.arch}"
+        result += f".{self.distro}"
+        if self.arch != "":
+            result += f".{self.arch}"
+
+        return result
 
     def __lt__(self, other) -> bool:
         if int(self.major) < int(other.major) or int(self.minor) < int(other.minor) or int(self.patch < other.patch):
diff --git a/pleskdistup/common/tests/versiontests.py b/pleskdistup/common/tests/versiontests.py
index 1ddaa08..8bbdd68 100644
--- a/pleskdistup/common/tests/versiontests.py
+++ b/pleskdistup/common/tests/versiontests.py
@@ -22,6 +22,9 @@ def test_kernel_parse_large_build(self):
     def test_kernel_parse_no_build(self):
         self._check_parse("3.10.0.el7.x86_64", "3.10.0.el7.x86_64")
 
+    def test_kernel_parse_virtuozo(self):
+        self._check_parse("3.10.0-1160.90.1.vz7.200.7", "3.10.0-1160.90.1.vz7")
+
     def test_compare_simple_equal(self):
         kernel1 = version.KernelVersion("3.10.0-1160.95.1.el7.x86_64")
         kernel2 = version.KernelVersion("3.10.0-1160.95.1.el7.x86_64")