mirror of
				https://github.com/Telecominfraproject/ols-nos.git
				synced 2025-10-31 18:17:52 +00:00 
			
		
		
		
	 55a707586b
			
		
	
	55a707586b
	
	
	
		
			
			* Support SONiC reproduceable build for deb/py2/py3/web * Remove j2 files * Fix bug * Fix some issues 1. Change some code format issues 2. Fix curl calling wget command, pip2 calling pip3 issue 3. Fix wget/curl downloading multiple urls issue * Fix some code format issue * Fix bug * Fix bug * Fix command path hard code in build info scripts issue * Add debian package sonic-build-tools * Fix auto debian package removed issue * Change build debian package name, and change the folder * Collect the pre-versions and post-versions * Change to use debian:buster * Remove apt-mark and improve code * Remove set_build_hooks * Change docker trusted gpg files * Fix docker build COPY directory name issue * Move the trusted gpg files into the sonic-build-hooks package
		
			
				
	
	
		
			667 lines
		
	
	
		
			28 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
	
	
	
			
		
		
	
	
			667 lines
		
	
	
		
			28 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
	
	
	
| #!/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()
 |