667 lines
28 KiB
Python
667 lines
28 KiB
Python
|
#!/usr/bin/python3
|
||
|
|
|
||
|
|
import argparse
|
||
|
|
import glob
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
|
||
|
|
ALL_DIST = 'all'
|
||
|
|
ALL_ARCH = 'all'
|
||
|
|
DEFAULT_MODULE = 'default'
|
||
|
|
DEFAULT_VERSION_PATH = 'files/build/versions'
|
||
|
|
VERSION_PREFIX="versions-"
|
||
|
|
VERSION_DEB_PREFERENCE = '01-versions-deb'
|
||
|
|
DEFAULT_OVERWRITE_COMPONENTS=['deb', 'py2', 'py3']
|
||
|
|
SLAVE_INDIVIDULE_VERSION = False
|
||
|
|
|
||
|
|
|
||
|
|
class Component:
|
||
|
|
'''
|
||
|
|
The component consists of mutiple packages
|
||
|
|
|
||
|
|
ctype -- Component Type, such as deb, py2, etc
|
||
|
|
dist -- Distribution, such as stretch, buster, etc
|
||
|
|
arch -- Architectrue, such as amd64, arm64, etc
|
||
|
|
|
||
|
|
'''
|
||
|
|
def __init__(self, versions, ctype, dist=ALL_DIST, arch=ALL_ARCH):
|
||
|
|
self.versions = versions
|
||
|
|
self.ctype = ctype
|
||
|
|
if not dist:
|
||
|
|
dist = ALL_DIST
|
||
|
|
if not arch:
|
||
|
|
arch = ALL_ARCH
|
||
|
|
self.dist = dist
|
||
|
|
self.arch = arch
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def get_versions(cls, version_file):
|
||
|
|
result = {}
|
||
|
|
if not os.path.exists(version_file):
|
||
|
|
return result
|
||
|
|
with open(version_file) as fp:
|
||
|
|
for line in fp.readlines():
|
||
|
|
offset = line.rfind('==')
|
||
|
|
if offset > 0:
|
||
|
|
package = line[:offset].strip()
|
||
|
|
version = line[offset+2:].strip()
|
||
|
|
result[package] = version
|
||
|
|
return result
|
||
|
|
|
||
|
|
def clone(self):
|
||
|
|
return Component(self.versions.copy(), self.ctype, self.dist, self.arch)
|
||
|
|
|
||
|
|
def merge(self, versions, overwritten=True):
|
||
|
|
for package in versions:
|
||
|
|
if overwritten or package not in self.versions:
|
||
|
|
self.versions[package] = versions[package]
|
||
|
|
|
||
|
|
def subtract(self, versions):
|
||
|
|
for package in versions:
|
||
|
|
if package in self.versions and self.versions[package] == versions[package]:
|
||
|
|
del self.versions[package]
|
||
|
|
|
||
|
|
def dump(self, config=False, priority=999):
|
||
|
|
result = []
|
||
|
|
for package in sorted(self.versions.keys(), key=str.casefold):
|
||
|
|
if config and self.ctype == 'deb':
|
||
|
|
lines = 'Package: {0}\nPin: version {1}\nPin-Priority: {2}\n\n'.format(package, self.versions[package], priority)
|
||
|
|
result.append(lines)
|
||
|
|
else:
|
||
|
|
result.append('{0}=={1}'.format(package, self.versions[package]))
|
||
|
|
return "\n".join(result)
|
||
|
|
|
||
|
|
def dump_to_file(self, version_file, config=False, priority=999):
|
||
|
|
if len(self.versions) <= 0:
|
||
|
|
return
|
||
|
|
with open(version_file, 'w') as f:
|
||
|
|
f.write(self.dump(config, priority))
|
||
|
|
|
||
|
|
def dump_to_path(self, file_path, config=False, priority=999):
|
||
|
|
if len(self.versions) <= 0:
|
||
|
|
return
|
||
|
|
if not os.path.exists(file_path):
|
||
|
|
os.makedirs(file_path)
|
||
|
|
filename = self.get_filename()
|
||
|
|
if config and self.ctype == 'deb':
|
||
|
|
none_config_file_path = os.path.join(file_path, filename)
|
||
|
|
self.dump_to_file(none_config_file_path, False, priority)
|
||
|
|
filename = VERSION_DEB_PREFERENCE
|
||
|
|
file_path = os.path.join(file_path, filename)
|
||
|
|
self.dump_to_file(file_path, config, priority)
|
||
|
|
|
||
|
|
# Check if the self component can be overwritten by the input component
|
||
|
|
def check_overwritable(self, component, for_all_dist=False, for_all_arch=False):
|
||
|
|
if self.ctype != component.ctype:
|
||
|
|
return False
|
||
|
|
if self.dist != component.dist and not (for_all_dist and self.dist == ALL_DIST):
|
||
|
|
return False
|
||
|
|
if self.arch != component.arch and not (for_all_arch and self.arch == ALL_ARCH):
|
||
|
|
return False
|
||
|
|
return True
|
||
|
|
|
||
|
|
# Check if the self component can inherit the package versions from the input component
|
||
|
|
def check_inheritable(self, component):
|
||
|
|
if self.ctype != component.ctype:
|
||
|
|
return False
|
||
|
|
if self.dist != component.dist and component.dist != ALL_DIST:
|
||
|
|
return False
|
||
|
|
if self.arch != component.arch and component.arch != ALL_ARCH:
|
||
|
|
return False
|
||
|
|
return True
|
||
|
|
|
||
|
|
'''
|
||
|
|
Get the file name
|
||
|
|
|
||
|
|
The file name format: versions-{ctype}-{dist}-{arch}
|
||
|
|
If {arch} is all, then the file name format: versions-{ctype}-{dist}
|
||
|
|
if {arch} is all and {dist} is all, then the file name format: versions-{ctype}
|
||
|
|
'''
|
||
|
|
def get_filename(self):
|
||
|
|
filename = VERSION_PREFIX + self.ctype
|
||
|
|
dist = self.dist
|
||
|
|
if self.arch and self.arch != ALL_ARCH:
|
||
|
|
if not dist:
|
||
|
|
dist = ALL_DIST
|
||
|
|
return filename + '-' + dist + '-' + self.arch
|
||
|
|
if dist and self.dist != ALL_DIST:
|
||
|
|
filename = filename + '-' + dist
|
||
|
|
return filename
|
||
|
|
|
||
|
|
def get_order_keys(self):
|
||
|
|
dist = self.dist
|
||
|
|
if not dist or dist == ALL_DIST:
|
||
|
|
dist = ''
|
||
|
|
arch = self.arch
|
||
|
|
if not arch or arch == ALL_ARCH:
|
||
|
|
arch = ''
|
||
|
|
return (self.ctype, dist, arch)
|
||
|
|
|
||
|
|
def clean_info(self, clean_dist=True, clean_arch=True, force=False):
|
||
|
|
if clean_dist:
|
||
|
|
if force or self.ctype != 'deb':
|
||
|
|
self.dist = ALL_DIST
|
||
|
|
if clean_arch:
|
||
|
|
self.arch = ALL_ARCH
|
||
|
|
|
||
|
|
|
||
|
|
class VersionModule:
|
||
|
|
'''
|
||
|
|
The version module represents a build target, such as docker image, host image, consists of multiple components.
|
||
|
|
|
||
|
|
name -- The name of the image, such as sonic-slave-buster, docker-lldp, etc
|
||
|
|
'''
|
||
|
|
def __init__(self, name=None, components=None):
|
||
|
|
self.name = name
|
||
|
|
self.components = components
|
||
|
|
|
||
|
|
# Overwrite the docker/host image/base image versions
|
||
|
|
def overwrite(self, module, for_all_dist=False, for_all_arch=False):
|
||
|
|
# Overwrite from generic one to detail one
|
||
|
|
# For examples: versions-deb overwrtten by versions-deb-buster, and versions-deb-buster overwritten by versions-deb-buster-amd64
|
||
|
|
components = sorted(module.components, key = lambda x : x.get_order_keys())
|
||
|
|
for merge_component in components:
|
||
|
|
merged = False
|
||
|
|
for component in self.components:
|
||
|
|
if component.check_overwritable(merge_component, for_all_dist=for_all_dist, for_all_arch=for_all_arch):
|
||
|
|
component.merge(merge_component.versions, True)
|
||
|
|
merged = True
|
||
|
|
if not merged:
|
||
|
|
tmp_component = merge_component.clone()
|
||
|
|
tmp_component.clean_info(clean_dist=for_all_dist, clean_arch=for_all_arch)
|
||
|
|
self.components.append(tmp_component)
|
||
|
|
self.adjust()
|
||
|
|
|
||
|
|
def get_config_module(self, default_module, dist, arch):
|
||
|
|
if self.is_individule_version():
|
||
|
|
return self
|
||
|
|
module = default_module
|
||
|
|
if not self.is_aggregatable_module(self.name):
|
||
|
|
module = default_module.clone(exclude_ctypes=DEFAULT_OVERWRITE_COMPONENTS)
|
||
|
|
return self._get_config_module(module, dist, arch)
|
||
|
|
|
||
|
|
def _get_config_module(self, default_module, dist, arch):
|
||
|
|
module = default_module.clone()
|
||
|
|
default_ctype_components = module._get_components_per_ctypes()
|
||
|
|
module.overwrite(self)
|
||
|
|
config_components = []
|
||
|
|
ctype_components = module._get_components_per_ctypes()
|
||
|
|
for ctype in default_ctype_components:
|
||
|
|
if ctype not in ctype_components:
|
||
|
|
ctype_components[ctype] = []
|
||
|
|
for components in ctype_components.values():
|
||
|
|
config_component = self._get_config_for_ctype(components, dist, arch)
|
||
|
|
config_components.append(config_component)
|
||
|
|
config_module = VersionModule(self.name, config_components)
|
||
|
|
return config_module
|
||
|
|
|
||
|
|
def _get_config_for_ctype(self, components, dist, arch):
|
||
|
|
result = Component({}, components[0].ctype, dist, arch)
|
||
|
|
for component in sorted(components, key = lambda x : x.get_order_keys()):
|
||
|
|
if result.check_inheritable(component):
|
||
|
|
result.merge(component.versions, True)
|
||
|
|
return result
|
||
|
|
|
||
|
|
def subtract(self, default_module):
|
||
|
|
module = self.clone()
|
||
|
|
result = []
|
||
|
|
ctype_components = module._get_components_per_ctypes()
|
||
|
|
for ctype in ctype_components:
|
||
|
|
components = ctype_components[ctype]
|
||
|
|
components = sorted(components, key = lambda x : x.get_order_keys())
|
||
|
|
for i in range(0, len(components)):
|
||
|
|
component = components[i]
|
||
|
|
base_module = VersionModule(self.name, components[0:i])
|
||
|
|
config_module = base_module._get_config_module(default_module, component.dist, component.arch)
|
||
|
|
config_components = config_module._get_components_by_ctype(ctype)
|
||
|
|
if len(config_components) > 0:
|
||
|
|
config_component = config_components[0]
|
||
|
|
component.subtract(config_component.versions)
|
||
|
|
if len(component.versions):
|
||
|
|
result.append(component)
|
||
|
|
self.components = result
|
||
|
|
|
||
|
|
def adjust(self):
|
||
|
|
result_components = []
|
||
|
|
ctype_components = self._get_components_per_ctypes()
|
||
|
|
for components in ctype_components.values():
|
||
|
|
result_components += self._adjust_components_for_ctype(components)
|
||
|
|
self.components = result_components
|
||
|
|
|
||
|
|
def _get_components_by_ctype(self, ctype):
|
||
|
|
components = []
|
||
|
|
for component in self.components:
|
||
|
|
if component.ctype == ctype:
|
||
|
|
components.append(component)
|
||
|
|
return components
|
||
|
|
|
||
|
|
def _adjust_components_for_ctype(self, components):
|
||
|
|
components = sorted(components, key = lambda x : x.get_order_keys())
|
||
|
|
result = []
|
||
|
|
for i in range(0, len(components)):
|
||
|
|
component = components[i]
|
||
|
|
inheritable_component = Component({}, component.ctype)
|
||
|
|
for j in range(0, i):
|
||
|
|
base_component = components[j]
|
||
|
|
if component.check_inheritable(base_component):
|
||
|
|
inheritable_component.merge(base_component.versions, True)
|
||
|
|
component.subtract(inheritable_component.versions)
|
||
|
|
if len(component.versions) > 0:
|
||
|
|
result.append(component)
|
||
|
|
return result
|
||
|
|
|
||
|
|
def _get_components_per_ctypes(self):
|
||
|
|
result = {}
|
||
|
|
for component in self.components:
|
||
|
|
components = result.get(component.ctype, [])
|
||
|
|
components.append(component)
|
||
|
|
result[component.ctype] = components
|
||
|
|
return result
|
||
|
|
|
||
|
|
def load(self, image_path, filter_ctype=None, filter_dist=None, filter_arch=None):
|
||
|
|
version_file_pattern = os.path.join(image_path, VERSION_PREFIX) + '*'
|
||
|
|
file_paths = glob.glob(version_file_pattern)
|
||
|
|
components = []
|
||
|
|
self.name = os.path.basename(image_path)
|
||
|
|
self.components = components
|
||
|
|
for file_path in file_paths:
|
||
|
|
filename = os.path.basename(file_path)
|
||
|
|
items = filename.split('-')
|
||
|
|
if len(items) < 2:
|
||
|
|
continue
|
||
|
|
ctype = items[1]
|
||
|
|
if filter_ctype and filter_ctype != ctype:
|
||
|
|
continue
|
||
|
|
dist = ''
|
||
|
|
arch = ''
|
||
|
|
if len(items) > 2:
|
||
|
|
dist = items[2]
|
||
|
|
if filter_dist and dist and filter_dist != dist:
|
||
|
|
continue
|
||
|
|
if len(items) > 3:
|
||
|
|
arch = items[3]
|
||
|
|
if filter_arch and arch and filter_arch != arch:
|
||
|
|
continue
|
||
|
|
versions = Component.get_versions(file_path)
|
||
|
|
component = Component(versions, ctype, dist, arch)
|
||
|
|
components.append(component)
|
||
|
|
|
||
|
|
def load_from_target(self, image_path):
|
||
|
|
post_versions = os.path.join(image_path, 'post-versions')
|
||
|
|
if os.path.exists(post_versions):
|
||
|
|
self.load(post_versions)
|
||
|
|
self.name = os.path.basename(image_path)
|
||
|
|
pre_versions = os.path.join(image_path, 'pre-versions')
|
||
|
|
if os.path.exists(pre_versions):
|
||
|
|
pre_module = VersionModule()
|
||
|
|
pre_module.load(pre_versions)
|
||
|
|
self.subtract(pre_module)
|
||
|
|
else:
|
||
|
|
self.load(image_path)
|
||
|
|
|
||
|
|
def dump(self, module_path, config=False, priority=999):
|
||
|
|
version_file_pattern = os.path.join(module_path, VERSION_PREFIX + '*')
|
||
|
|
for filename in glob.glob(version_file_pattern):
|
||
|
|
os.remove(filename)
|
||
|
|
for component in self.components:
|
||
|
|
component.dump_to_path(module_path, config, priority)
|
||
|
|
|
||
|
|
def filter(self, ctypes=[]):
|
||
|
|
if 'all' in ctypes:
|
||
|
|
return self
|
||
|
|
components = []
|
||
|
|
for component in self.components:
|
||
|
|
if component.ctype in ctypes:
|
||
|
|
components.append(component)
|
||
|
|
self.components = components
|
||
|
|
|
||
|
|
def clean_info(self, clean_dist=True, clean_arch=True, force=False):
|
||
|
|
for component in self.components:
|
||
|
|
component.clean_info(clean_dist=clean_dist, clean_arch=clean_arch, force=force)
|
||
|
|
|
||
|
|
def clone(self, ctypes=None, exclude_ctypes=None):
|
||
|
|
components = []
|
||
|
|
for component in self.components:
|
||
|
|
if exclude_ctypes and component.ctype in exclude_ctypes:
|
||
|
|
continue
|
||
|
|
if ctypes and component.ctype not in ctypes:
|
||
|
|
continue
|
||
|
|
components.append(component.clone())
|
||
|
|
return VersionModule(self.name, components)
|
||
|
|
|
||
|
|
def is_slave_module(self):
|
||
|
|
return self.name.startswith('sonic-slave-')
|
||
|
|
|
||
|
|
# Do not inherit the version from the default module
|
||
|
|
def is_individule_version(self):
|
||
|
|
return self.is_slave_module() and SLAVE_INDIVIDULE_VERSION
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def is_aggregatable_module(cls, module_name):
|
||
|
|
if module_name.startswith('sonic-slave-'):
|
||
|
|
return False
|
||
|
|
if module_name.startswith('build-sonic-slave-'):
|
||
|
|
return False
|
||
|
|
if module_name == DEFAULT_MODULE:
|
||
|
|
return False
|
||
|
|
if module_name == 'host-image' or module_name == 'host-base-image':
|
||
|
|
return False
|
||
|
|
return True
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def get_module_path_by_name(cls, source_path, module_name):
|
||
|
|
common_modules = ['default', 'host-image', 'host-base-image']
|
||
|
|
if module_name in common_modules:
|
||
|
|
return os.path.join(source_path, 'files/build/versions', module_name)
|
||
|
|
if module_name.startswith('build-sonic-slave-'):
|
||
|
|
return os.path.join(source_path, 'files/build/versions/build', module_name)
|
||
|
|
return os.path.join(source_path, 'files/build/versions/dockers', module_name)
|
||
|
|
|
||
|
|
class VersionBuild:
|
||
|
|
'''
|
||
|
|
The VersionBuild consists of multiple version modules.
|
||
|
|
|
||
|
|
'''
|
||
|
|
def __init__(self, target_path="./target", source_path='.'):
|
||
|
|
self.target_path = target_path
|
||
|
|
self.source_path = source_path
|
||
|
|
self.modules = {}
|
||
|
|
|
||
|
|
def load_from_target(self):
|
||
|
|
dockers_path = os.path.join(self.target_path, 'versions/dockers')
|
||
|
|
build_path = os.path.join(self.target_path, 'versions/build')
|
||
|
|
modules = {}
|
||
|
|
self.modules = modules
|
||
|
|
file_paths = glob.glob(dockers_path + '/*')
|
||
|
|
file_paths += glob.glob(build_path + '/build-*')
|
||
|
|
file_paths.append(os.path.join(self.target_path, 'versions/host-image'))
|
||
|
|
file_paths.append(os.path.join(self.target_path, 'versions/host-base-image'))
|
||
|
|
for file_path in file_paths:
|
||
|
|
if not os.path.isdir(file_path):
|
||
|
|
continue
|
||
|
|
module = VersionModule()
|
||
|
|
module.load_from_target(file_path)
|
||
|
|
modules[module.name] = module
|
||
|
|
self._merge_dgb_modules()
|
||
|
|
|
||
|
|
def load_from_source(self):
|
||
|
|
# Load default versions and host image versions
|
||
|
|
versions_path = os.path.join(self.source_path, 'files/build/versions')
|
||
|
|
dockers_path = os.path.join(versions_path, "dockers")
|
||
|
|
build_path = os.path.join(versions_path, "build")
|
||
|
|
paths = [os.path.join(versions_path, 'default')]
|
||
|
|
paths += glob.glob(versions_path + '/host-*')
|
||
|
|
paths += glob.glob(dockers_path + '/*')
|
||
|
|
paths += glob.glob(build_path + '/*')
|
||
|
|
modules = {}
|
||
|
|
self.modules = modules
|
||
|
|
for image_path in paths:
|
||
|
|
module = VersionModule()
|
||
|
|
module.load(image_path)
|
||
|
|
modules[module.name] = module
|
||
|
|
|
||
|
|
def overwrite(self, build, for_all_dist=False, for_all_arch=False):
|
||
|
|
for target_module in build.modules.values():
|
||
|
|
module = self.modules.get(target_module.name, None)
|
||
|
|
tmp_module = target_module.clone()
|
||
|
|
tmp_module.clean_info(for_all_dist, for_all_arch)
|
||
|
|
if module:
|
||
|
|
module.overwrite(tmp_module, for_all_dist=for_all_dist, for_all_arch=for_all_arch)
|
||
|
|
else:
|
||
|
|
self.modules[target_module.name] = tmp_module
|
||
|
|
|
||
|
|
def dump(self):
|
||
|
|
for module in self.modules.values():
|
||
|
|
module_path = self.get_module_path(module)
|
||
|
|
module.dump(module_path)
|
||
|
|
|
||
|
|
def subtract(self, default_module):
|
||
|
|
none_aggregatable_module = default_module.clone(exclude_ctypes=DEFAULT_OVERWRITE_COMPONENTS)
|
||
|
|
for module in self.modules.values():
|
||
|
|
if module.name == DEFAULT_MODULE:
|
||
|
|
continue
|
||
|
|
if module.name == 'host-base-image':
|
||
|
|
continue
|
||
|
|
if module.is_individule_version():
|
||
|
|
continue
|
||
|
|
tmp_module = default_module
|
||
|
|
if not module.is_aggregatable_module(module.name):
|
||
|
|
tmp_module = none_aggregatable_module
|
||
|
|
module.subtract(tmp_module)
|
||
|
|
|
||
|
|
def freeze(self, rebuild=False, for_all_dist=False, for_all_arch=False, ctypes=['all']):
|
||
|
|
if rebuild:
|
||
|
|
self.load_from_target()
|
||
|
|
self.filter(ctypes=ctypes)
|
||
|
|
default_module = self.get_default_module()
|
||
|
|
self._clean_component_info()
|
||
|
|
self.subtract(default_module)
|
||
|
|
self.modules[DEFAULT_MODULE] = default_module
|
||
|
|
self.dump()
|
||
|
|
return
|
||
|
|
self.load_from_source()
|
||
|
|
default_module = self.modules.get(DEFAULT_MODULE, None)
|
||
|
|
target_build = VersionBuild(self.target_path, self.source_path)
|
||
|
|
target_build.load_from_target()
|
||
|
|
target_build.filter(ctypes=ctypes)
|
||
|
|
if not default_module:
|
||
|
|
raise Exception("The default versions does not exist")
|
||
|
|
for module in target_build.modules.values():
|
||
|
|
if module.is_individule_version():
|
||
|
|
continue
|
||
|
|
tmp_module = module.clone(exclude_ctypes=DEFAULT_OVERWRITE_COMPONENTS)
|
||
|
|
default_module.overwrite(tmp_module, for_all_dist=True, for_all_arch=True)
|
||
|
|
target_build.subtract(default_module)
|
||
|
|
self.overwrite(target_build, for_all_dist=for_all_dist, for_all_arch=for_all_arch)
|
||
|
|
self.dump()
|
||
|
|
|
||
|
|
def filter(self, ctypes=[]):
|
||
|
|
for module in self.modules.values():
|
||
|
|
module.filter(ctypes=ctypes)
|
||
|
|
|
||
|
|
def get_default_module(self):
|
||
|
|
if DEFAULT_MODULE in self.modules:
|
||
|
|
return self.modules[DEFAULT_MODULE]
|
||
|
|
ctypes = self.get_component_types()
|
||
|
|
dists = self.get_dists()
|
||
|
|
components = []
|
||
|
|
for ctype in ctypes:
|
||
|
|
if ctype == 'deb':
|
||
|
|
for dist in dists:
|
||
|
|
versions = self._get_versions(ctype, dist)
|
||
|
|
common_versions = self._get_common_versions(versions)
|
||
|
|
component = Component(common_versions, ctype, dist)
|
||
|
|
components.append(component)
|
||
|
|
else:
|
||
|
|
versions = self._get_versions(ctype)
|
||
|
|
common_versions = self._get_common_versions(versions)
|
||
|
|
component = Component(common_versions, ctype)
|
||
|
|
components.append(component)
|
||
|
|
return VersionModule(DEFAULT_MODULE, components)
|
||
|
|
|
||
|
|
def get_aggregatable_modules(self):
|
||
|
|
modules = {}
|
||
|
|
for module_name in self.modules:
|
||
|
|
if not VersionModule.is_aggregatable_module(module_name):
|
||
|
|
continue
|
||
|
|
module = self.modules[module_name]
|
||
|
|
modules[module_name] = module
|
||
|
|
return modules
|
||
|
|
|
||
|
|
def get_components(self):
|
||
|
|
components = []
|
||
|
|
for module_name in self.modules:
|
||
|
|
module = self.modules[module_name]
|
||
|
|
for component in module.components:
|
||
|
|
components.append(component)
|
||
|
|
return components
|
||
|
|
|
||
|
|
def get_component_types(self):
|
||
|
|
ctypes = []
|
||
|
|
for module_name in self.modules:
|
||
|
|
module = self.modules[module_name]
|
||
|
|
for component in module.components:
|
||
|
|
if component.ctype not in ctypes:
|
||
|
|
ctypes.append(component.ctype)
|
||
|
|
return ctypes
|
||
|
|
|
||
|
|
def get_dists(self):
|
||
|
|
dists = []
|
||
|
|
components = self.get_components()
|
||
|
|
for component in components:
|
||
|
|
if component.dist not in dists:
|
||
|
|
dists.append(component.dist)
|
||
|
|
return dists
|
||
|
|
|
||
|
|
def get_archs(self):
|
||
|
|
archs = []
|
||
|
|
components = self.get_components()
|
||
|
|
for component in components:
|
||
|
|
if component.arch not in archs:
|
||
|
|
archs.append(component.arch)
|
||
|
|
return archs
|
||
|
|
|
||
|
|
def get_module_path(self, module):
|
||
|
|
return self.get_module_path_by_name(module.name)
|
||
|
|
|
||
|
|
def get_module_path_by_name(self, module_name):
|
||
|
|
return VersionModule.get_module_path_by_name(self.source_path, module_name)
|
||
|
|
|
||
|
|
def _merge_dgb_modules(self):
|
||
|
|
dbg_modules = []
|
||
|
|
for module_name in self.modules:
|
||
|
|
if not module_name.endswith('-dbg'):
|
||
|
|
continue
|
||
|
|
dbg_modules.append(module_name)
|
||
|
|
base_module_name = module_name[:-4]
|
||
|
|
if base_module_name not in self.modules:
|
||
|
|
raise Exception('The Module {0} not found'.format(base_module_name))
|
||
|
|
base_module = self.modules[base_module_name]
|
||
|
|
dbg_module = self.modules[module_name]
|
||
|
|
base_module.overwrite(dbg_module)
|
||
|
|
for module_name in dbg_modules:
|
||
|
|
del self.modules[module_name]
|
||
|
|
|
||
|
|
def _clean_component_info(self, clean_dist=True, clean_arch=True):
|
||
|
|
for module in self.modules.values():
|
||
|
|
module.clean_info(clean_dist, clean_arch)
|
||
|
|
|
||
|
|
def _get_versions(self, ctype, dist=None, arch=None):
|
||
|
|
versions = {}
|
||
|
|
modules = self.get_aggregatable_modules()
|
||
|
|
for module_name in self.modules:
|
||
|
|
if module_name not in modules:
|
||
|
|
temp_module = self.modules[module_name].clone(exclude_ctypes=DEFAULT_OVERWRITE_COMPONENTS)
|
||
|
|
modules[module_name] = temp_module
|
||
|
|
for module in modules.values():
|
||
|
|
for component in module.components:
|
||
|
|
if ctype != component.ctype:
|
||
|
|
continue
|
||
|
|
if dist and dist != component.dist:
|
||
|
|
continue
|
||
|
|
if arch and arch != component.arch:
|
||
|
|
continue
|
||
|
|
for package in component.versions:
|
||
|
|
version = component.versions[package]
|
||
|
|
package_versions = versions.get(package, [])
|
||
|
|
if version not in package_versions:
|
||
|
|
package_versions.append(version)
|
||
|
|
versions[package] = package_versions
|
||
|
|
return versions
|
||
|
|
|
||
|
|
def _get_common_versions(self, versions):
|
||
|
|
common_versions = {}
|
||
|
|
for package in versions:
|
||
|
|
package_versions = versions[package]
|
||
|
|
if len(package_versions) == 1:
|
||
|
|
common_versions[package] = package_versions[0]
|
||
|
|
return common_versions
|
||
|
|
|
||
|
|
|
||
|
|
class VersionManagerCommands:
|
||
|
|
def __init__(self):
|
||
|
|
usage = 'version_manager.py <command> [<args>]\n\n'
|
||
|
|
usage = usage + 'The most commonly used commands are:\n'
|
||
|
|
usage = usage + ' freeze Freeze the version files\n'
|
||
|
|
usage = usage + ' generate Generate the version files\n'
|
||
|
|
usage = usage + ' merge Merge the version files'
|
||
|
|
parser = argparse.ArgumentParser(description='Version manager', usage=usage)
|
||
|
|
parser.add_argument('command', help='Subcommand to run')
|
||
|
|
args = parser.parse_args(sys.argv[1:2])
|
||
|
|
if not hasattr(self, args.command):
|
||
|
|
print('Unrecognized command: {0}'.format(args.command))
|
||
|
|
parser.print_help()
|
||
|
|
exit(1)
|
||
|
|
getattr(self, args.command)()
|
||
|
|
|
||
|
|
def freeze(self):
|
||
|
|
parser = argparse.ArgumentParser(description = 'Freeze the version files')
|
||
|
|
parser.add_argument('-t', '--target_path', default='./target', help='target path')
|
||
|
|
parser.add_argument('-s', '--source_path', default='.', help='source path')
|
||
|
|
|
||
|
|
# store_true which implies default=False
|
||
|
|
parser.add_argument('-r', '--rebuild', action='store_true', help='rebuild all versions')
|
||
|
|
parser.add_argument('-d', '--for_all_dist', action='store_true', help='apply the versions for all distributions')
|
||
|
|
parser.add_argument('-a', '--for_all_arch', action='store_true', help='apply the versions for all architectures')
|
||
|
|
parser.add_argument('-c', '--ctypes', default='all', help='component types to freeze')
|
||
|
|
args = parser.parse_args(sys.argv[2:])
|
||
|
|
ctypes = args.ctypes.split(',')
|
||
|
|
if len(ctypes) == 0:
|
||
|
|
ctypes = ['all']
|
||
|
|
build = VersionBuild(target_path=args.target_path, source_path=args.source_path)
|
||
|
|
build.freeze(rebuild=args.rebuild, for_all_dist=args.for_all_dist, for_all_arch=args.for_all_arch, ctypes=ctypes)
|
||
|
|
|
||
|
|
def merge(self):
|
||
|
|
parser = argparse.ArgumentParser(description = 'Merge the version files')
|
||
|
|
parser.add_argument('-t', '--target_path', required=True, help='target path to save the merged version files')
|
||
|
|
parser.add_argument('-m', '--module_path', default=None, help='merge path, use the target path if not specified')
|
||
|
|
parser.add_argument('-b', '--base_path', required=True, help='base path, merge to the module path')
|
||
|
|
parser.add_argument('-e', '--exclude_module_path', default=None, help='exclude module path')
|
||
|
|
args = parser.parse_args(sys.argv[2:])
|
||
|
|
module_path = args.module_path
|
||
|
|
if not module_path:
|
||
|
|
module_path = args.target_path
|
||
|
|
if not os.path.exists(module_path):
|
||
|
|
print('The module path {0} does not exist'.format(module_path))
|
||
|
|
if not os.path.exists(args.target_path):
|
||
|
|
os.makedirs(args.target_path)
|
||
|
|
module = VersionModule()
|
||
|
|
module.load(module_path)
|
||
|
|
base_module = VersionModule()
|
||
|
|
base_module.load(args.base_path)
|
||
|
|
module.overwrite(base_module)
|
||
|
|
if args.exclude_module_path:
|
||
|
|
exclude_module = VersionModule()
|
||
|
|
exclude_module.load(args.exclude_module_path)
|
||
|
|
module.subtract(exclude_module)
|
||
|
|
module.dump(args.target_path)
|
||
|
|
|
||
|
|
def generate(self):
|
||
|
|
parser = argparse.ArgumentParser(description = 'Generate the version files')
|
||
|
|
parser.add_argument('-t', '--target_path', required=True, help='target path to generate the version lock files')
|
||
|
|
group = parser.add_mutually_exclusive_group(required=True)
|
||
|
|
group.add_argument('-n', '--module_name', help="module name, such as docker-lldp, sonic-slave-buster, etc")
|
||
|
|
group.add_argument('-m', '--module_path', help="module apth, such as files/docker/versions/dockers/docker-lldp, files/docker/versions/dockers/sonic-slave-buster, etc")
|
||
|
|
parser.add_argument('-s', '--source_path', default='.', help='source path')
|
||
|
|
parser.add_argument('-d', '--distribution', required=True, help="distribution")
|
||
|
|
parser.add_argument('-a', '--architecture', required=True, help="architecture")
|
||
|
|
parser.add_argument('-p', '--priority', default=999, help="priority of the debian apt preference")
|
||
|
|
|
||
|
|
args = parser.parse_args(sys.argv[2:])
|
||
|
|
module_path = args.module_path
|
||
|
|
if not module_path:
|
||
|
|
module_path = VersionModule.get_module_path_by_name(args.source_path, args.module_name)
|
||
|
|
if not os.path.exists(args.target_path):
|
||
|
|
os.makedirs(args.target_path)
|
||
|
|
module = VersionModule()
|
||
|
|
module.load(module_path, filter_dist=args.distribution, filter_arch=args.architecture)
|
||
|
|
default_module_path = VersionModule.get_module_path_by_name(args.source_path, DEFAULT_MODULE)
|
||
|
|
default_module = VersionModule()
|
||
|
|
default_module.load(default_module_path, filter_dist=args.distribution, filter_arch=args.architecture)
|
||
|
|
config = module.get_config_module(default_module, args.distribution, args.architecture)
|
||
|
|
config.clean_info(force=True)
|
||
|
|
config.dump(args.target_path, config=True, priority=args.priority)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
VersionManagerCommands()
|