3 Ü““i²9ã@s>ddlZddlZddlmZGdd„dƒZGdd„deƒZdS)éN)Úsos_get_command_outputc@s¢eZdZdZdZdZdZdZdZdZ dZ d dd„Zedd„ƒZ edd„ƒZd!dd„Zd d„Zd"dd„Zdd„Zdd„Zdd„Zdd„Zdd„Zdd„Zdd„ZdS)#ÚPackageManagerauEncapsulates a package manager. If you provide a query_command to the constructor it should print each package on the system in the following format:: package name|package.version You may also subclass this class and provide a _generate_pkg_list method to build the list of packages and versions. :cvar query_command: The command to use for querying packages :vartype query_command: ``str`` or ``None`` :cvar verify_command: The command to use for verifying packages :vartype verify_command: ``str`` or ``None`` :cvar verify_filter: Optional filter to use for controlling package verification :vartype verify_filter: ``str or ``None`` :cvar files_command: The command to use for getting file lists for packages :vartype files_command: ``str`` or ``None`` :cvar chroot: Perform a chroot when executing `files_command` :vartype chroot: ``bool`` :cvar remote_exec: If package manager is on a remote system (e.g. for sos collect), use this to execute commands :vartype remote_exec: ``SoSTransport.run_command()`` NcCs d|_g|_||_|r||_dS)N)Ú _packagesÚfilesÚremote_execÚchroot)Úselfrr©r ú/usr/lib/python3.6/__init__.pyÚ__init__8s zPackageManager.__init__cCs|jdkr|jƒ|jS)N)rÚ_generate_pkg_list)rr r r Úpackages@s zPackageManager.packagescCs|jjjƒjddddS)NÚpackageé)Úmaxsplitr)Ú __class__Ú__name__ÚlowerÚsplit)rr r r Úmanager_nameFszPackageManager.manager_nameéFcCsB|jr|j|||||ƒ}nt||||d}|ddkr>|dSdS)a Runs a package manager command, either via sos_get_command_output() if local, or via a SoSTransport's run_command() if this needs to be run remotely, as in the case of remote nodes for use during `sos collect`. :param command: The command to execute :type command: ``str`` :param timeout: Timeout for command to run, in seconds :type timeout: ``int`` :param need_root: Does the command require root privileges? :type need_root: ``bool`` :param env: Environment variables to set :type env: ``dict`` with keys being env vars to define :param use_shell: If running remotely, does the command require obtaining a shell? :type use_shell: ``bool`` :param chroot: If necessary, chroot command execution to here :type chroot: ``None`` or ``str`` :returns: The output of the command :rtype: ``str`` )rÚenvZstatusrÚoutputÚ)rr)rZcommandÚtimeoutZ need_rootrZ use_shellrÚretr r r Úexec_cmdJszPackageManager.exec_cmdcCstj|jjƒ|ƒS)zÔ Get a list of packages that match name. :param name: The name of the package :type name: ``str`` :returns: List of all packages matching `name` :rtype: ``list`` )ÚfnmatchÚfilterr Úkeys)rÚnamer r r Úall_pkgs_by_namess zPackageManager.all_pkgs_by_namercs tj||ƒ‰‡fdd„|jDƒS)aL Get a list of packages that match regex_name. :param regex_name: The regex to use for matching package names against :type regex_name: ``str`` :param flags: Flags for the `re` module when matching `regex_name` :returns: All packages matching `regex_name` :rtype: ``list`` csg|]}ˆj|ƒr|‘qSr )Úmatch)Ú.0Úpkg)Úregr r ú Œsz9PackageManager.all_pkgs_by_name_regex..)ÚreÚcompiler )rZ regex_nameÚflagsr )r%r Úall_pkgs_by_name_regexsz%PackageManager.all_pkgs_by_name_regexcCs$y |j|Stk rdSXdS)zÔ Get a single package that matches name. :param name: The name of the package :type name: ``str`` :returns: The first package that matches `name` :rtype: ``str`` N)r Ú Exception)rr r r r Úpkg_by_nameŽs zPackageManager.pkg_by_namecCst‚dS)a+ Using the output of `query_command`, build the _packages dict. This should be overridden by distinct package managers and be a generator for _generate_pkg_list which will insert the packages into the _packages dict. This method should yield a tuple of name, version, release for each package parsed. If the package manager or distribution does not use a release field, set it to None. :param pkg_list: The output of the result of `query_command` :type pkg_list: ``str`` N)ÚNotImplementedError)rÚpkg_listr r r Ú_parse_pkg_listszPackageManager._parse_pkg_listcCsr|jdkri|_|jrn|j}|j|d|jd}x>|j|ƒD]0}|d|djdƒ|d|jdœ|j|d<q:WdS) amGenerates a dictionary of packages for internal use by the package manager in the format:: {'package_name': {'name': 'package_name', 'version': 'major.minor.version', 'release': 'package release' or None, 'pkg_manager': 'package manager name'}} Nr)rrrrÚ.é)r ÚversionÚreleaseZpkg_manager)rÚ query_commandrrr/rr)rÚcmdr.r$r r r r®s z!PackageManager._generate_pkg_listcCs||jkr|j|SdS)zøReturns the entry in self.packages for pkg if it exists :param pkg: The name of the package :type pkg: ``str`` :returns: Package name and version, if package exists :rtype: ``dict`` if found, else ``None`` N)r )rr$r r r Úpkg_versionÇs zPackageManager.pkg_versioncCs6|jr0|jr0|j}|j|d|jd}|jƒ|_|jS)z• Get a list of files known by the package manager :returns: All files known by the package manager :rtype: ``list`` é´)rr)Ú files_commandrrrÚ splitlines)rr5rr r r Ú all_filesÔs zPackageManager.all_filescCsR|js dSy.|j›d|›}|j|d|jd}|jƒp8dStk rLdSXdS)zúGiven a path, return the package that owns that path. :param path: The filepath to check for package ownership :type path: ``str`` :returns: The package name or 'unknown' :rtype: ``str`` Úunknownú é)rrN)Úquery_path_commandrrr9r+)rÚpathr5r$r r r Úpkg_by_pathás zPackageManager.pkg_by_pathcs„|js dS|j}tdt||ƒƒ}|s(dSd}xH|D]@}x:|D]2‰t‡fdd„|jDƒƒrZq<|rf|d7}|ˆ7}q str Generate a command to verify the list of packages given in ``packages`` using the native package manager's verification tool. The command to be executed is returned as a string that may be passed to a command execution routine (for e.g. ``sos_get_command_output()``. :param packages: a string, or a list of strings giving package names to be verified. :returns: a string containing an executable command that will perform verification of the given packages. :rtype: str or ``NoneType`` Nrc3s|]}|ˆkVqdS)Nr )r#Úf)rr r ú sz6PackageManager.build_verify_command..r<)Úverify_commandr*rÚmapÚanyÚ verify_filter)rr Zby_regexZverify_listZverify_packagesZpackage_listr )rr Úbuild_verify_commandós z#PackageManager.build_verify_command)NN)rFNFN)r)rÚ __module__Ú__qualname__Ú__doc__r4rCrFr8r>rrrÚpropertyr rrr!r*r,r/rr6r:r@rGr r r r rs, ( rcsNeZdZdZd‡fdd„ Zdd„Zdd„Zd d „Zdd„Z‡fd d„Z ‡Z S)ÚMultiPackageManagera± This class is used to leverage multiple individual package managers as a single entity on systems that support multiple concurrent package managers. Policies that use this approach will need to specify a primary package manager, and at least one fallback manager. When queries are sent to this manager, the primary child manager is checked first. If there is a valid, not None, response (e.g. a given package is installed) then that response is used. However, if the response is empty or None, the fallback managers are then queried in the order they were passed to MultiPackageManager during initialization. :param primary: The primary package manager to rely on :type primary: A subclass of `PackageManager` :param fallbacks: A list of package managers to use if the primary does not provide a response :type fallbacks: ``list`` of `PackageManager` subclasses Ncs²tƒjˆˆdt|tƒs*td|j›ƒ‚t|tƒs.z1Must define at least one fallback package manager)ÚsuperrÚ issubclassrr+rÚ isinstanceÚlistÚprimaryÚ fallbacksÚ _managersÚextend)rrRrSrrrM)r)rrr r2s" zMultiPackageManager.__init__cCst‚dS)a+ Using the output of `query_command`, build the _packages dict. This should be overridden by distinct package managers and be a generator for _generate_pkg_list which will insert the packages into the _packages dict. This method should yield a tuple of name, version, release for each package parsed. If the package manager or distribution does not use a release field, set it to None. :param pkg_list: The output of the result of `query_command` :type pkg_list: ``str`` N)r-)rr.r r r r/Rsz#MultiPackageManager._parse_pkg_listcCs,|js&x|jD]}|jj|jƒƒqW|jS)N)rrTrUr:)rrMr r r r:cszMultiPackageManager.all_filescCsd|jdkri|_|jj|jjƒx>|jD]4}|j}x(|jƒD]}||jkr<|||j|<q.pkg_funcr )rrZr[r )rZrr Ú_pm_wrappertszMultiPackageManager._pm_wrappercs8|dkrtƒj|ƒStƒj|ƒ}t|dƒr4|j|ƒS|S)Nrr\r:Ú__call__)rr\r:)rNÚ__getattribute__Úhasattrr\)rÚitemÚattr)rr r r^‹s z$MultiPackageManager.__getattribute__)NN)rrHrIrJrr/r:rr\r^Ú __classcell__r r )rr rLs rL)r'rZ sos.utilitiesrrrLr r r r Ús