1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
import functools
import logging
from typing import Optional
from packaging import version
from cloudinit import subp
from cloudinit.subp import ProcessExecutionError
from tests.integration_tests import integration_settings
log = logging.getLogger("integration_testing")
def get_all_ubuntu_series() -> list:
"""Use distro-info-data's ubuntu.csv to get a list of Ubuntu series"""
out = ""
try:
out, _err = subp.subp(["ubuntu-distro-info", "-a"])
except ProcessExecutionError:
log.info(
"ubuntu-distro-info (from the distro-info package) must be"
" installed to guess Ubuntu os/release"
)
return out.splitlines()
def ubuntu_version_from_series(series) -> str:
try:
out, _err = subp.subp(
["ubuntu-distro-info", "--release", "--series", series]
)
except subp.ProcessExecutionError as e:
raise ValueError(
f"'{series}' is not a recognized Ubuntu release"
) from e
return out.strip().rstrip(" LTS")
@functools.total_ordering
class Release:
def __init__(
self,
os: str,
series: str,
version: str,
image_id: Optional[str] = None,
):
self.os = os
self.series = series
self.version = version
self.image_id = image_id
def __repr__(self):
return f"Release({self.os}, {self.version})"
def __lt__(self, other: "Release"):
if self.os != other.os:
raise ValueError(f"{self.os} cannot be compared to {other.os}!")
return version.parse(self.version) < version.parse(other.version)
@classmethod
def from_os_image(
cls,
os_image: str = integration_settings.OS_IMAGE,
) -> "Release":
"""Get the individual parts from an OS_IMAGE definition.
Returns a namedtuple containing id, os, and release of the image."""
parts = os_image.split("::", 3)
if len(parts) == 1:
image_id = None
os = "ubuntu"
series = parts[0]
version = ubuntu_version_from_series(series)
elif len(parts) == 4:
image_id, os, series, version = parts
else:
raise ValueError(
"OS_IMAGE must either contain release name or be in the form "
"of <image_id>[::<os>::<release>::<version>]"
)
return cls(os, series, version, image_id)
BIONIC = Release("ubuntu", "bionic", "18.04")
FOCAL = Release("ubuntu", "focal", "20.04")
JAMMY = Release("ubuntu", "jammy", "22.04")
KINETIC = Release("ubuntu", "kinetic", "22.10")
LUNAR = Release("ubuntu", "lunar", "23.04")
CURRENT_RELEASE = Release.from_os_image()
IS_UBUNTU = CURRENT_RELEASE.os == "ubuntu"
|