From b415e4855970131a77112940646a95641d3bd27b Mon Sep 17 00:00:00 2001
From: Thomas Wiest <twiest@redhat.com>
Date: Wed, 25 Jan 2017 14:07:22 -0500
Subject: Added repoquery to lib_utils.

---
 roles/lib_utils/src/class/repoquery.py | 156 +++++++++++++++++++++++++++++++++
 1 file changed, 156 insertions(+)
 create mode 100644 roles/lib_utils/src/class/repoquery.py

(limited to 'roles/lib_utils/src/class/repoquery.py')

diff --git a/roles/lib_utils/src/class/repoquery.py b/roles/lib_utils/src/class/repoquery.py
new file mode 100644
index 000000000..2447719e2
--- /dev/null
+++ b/roles/lib_utils/src/class/repoquery.py
@@ -0,0 +1,156 @@
+# pylint: skip-file
+# flake8: noqa
+
+
+class Repoquery(RepoqueryCLI):
+    ''' Class to wrap the repoquery
+    '''
+    # pylint: disable=too-many-arguments
+    def __init__(self, name, query_type, show_duplicates,
+                 match_version, verbose):
+        ''' Constructor for YumList '''
+        super(Repoquery, self).__init__(None)
+        self.name = name
+        self.query_type = query_type
+        self.show_duplicates = show_duplicates
+        self.match_version = match_version
+        self.verbose = verbose
+
+        if self.match_version:
+            self.show_duplicates = True
+
+        self.query_format = "%{version}|%{release}|%{arch}|%{repo}|%{version}-%{release}"
+
+    def build_cmd(self):
+        ''' build the repoquery cmd options '''
+
+        repo_cmd = []
+
+        repo_cmd.append("--pkgnarrow=" + self.query_type)
+        repo_cmd.append("--queryformat=" + self.query_format)
+
+        if self.show_duplicates:
+            repo_cmd.append('--show-duplicates')
+
+        repo_cmd.append(self.name)
+
+        return repo_cmd
+
+    @staticmethod
+    def process_versions(query_output):
+        ''' format the package data into something that can be presented '''
+
+        version_dict = defaultdict(dict)
+
+        for version in query_output.split('\n'):
+            pkg_info = version.split("|")
+
+            pkg_version = {}
+            pkg_version['version'] = pkg_info[0]
+            pkg_version['release'] = pkg_info[1]
+            pkg_version['arch'] = pkg_info[2]
+            pkg_version['repo'] = pkg_info[3]
+            pkg_version['version_release'] = pkg_info[4]
+
+            version_dict[pkg_info[4]] = pkg_version
+
+        return version_dict
+
+    def format_versions(self, formatted_versions):
+        ''' Gather and present the versions of each package '''
+
+        versions_dict = {}
+        versions_dict['available_versions_full'] = formatted_versions.keys()
+
+        # set the match version, if called
+        if self.match_version:
+            versions_dict['matched_versions_full'] = []
+            versions_dict['requested_match_version'] = self.match_version
+            versions_dict['matched_versions'] = []
+
+        # get the "full version (version - release)
+        versions_dict['available_versions_full'].sort(key=LooseVersion)
+        versions_dict['latest_full'] = versions_dict['available_versions_full'][-1]
+
+        # get the "short version (version)
+        versions_dict['available_versions'] = []
+        for version in versions_dict['available_versions_full']:
+            versions_dict['available_versions'].append(formatted_versions[version]['version'])
+
+            if self.match_version:
+                if version.startswith(self.match_version):
+                    versions_dict['matched_versions_full'].append(version)
+                    versions_dict['matched_versions'].append(formatted_versions[version]['version'])
+
+        versions_dict['available_versions'].sort(key=LooseVersion)
+        versions_dict['latest'] = versions_dict['available_versions'][-1]
+
+        # finish up the matched version
+        if self.match_version:
+            if versions_dict['matched_versions_full']:
+                versions_dict['matched_version_found'] = True
+                versions_dict['matched_versions'].sort(key=LooseVersion)
+                versions_dict['matched_version_latest'] = versions_dict['matched_versions'][-1]
+                versions_dict['matched_version_full_latest'] = versions_dict['matched_versions_full'][-1]
+            else:
+                versions_dict['matched_version_found'] = False
+                versions_dict['matched_versions'] = []
+                versions_dict['matched_version_latest'] = ""
+                versions_dict['matched_version_full_latest'] = ""
+
+        return versions_dict
+
+    def repoquery(self):
+        '''perform a repoquery '''
+
+        repoquery_cmd = self.build_cmd()
+
+        rval = self._repoquery_cmd(repoquery_cmd, True, 'raw')
+
+        # check to see if there are actual results
+        if rval['results']:
+            processed_versions = Repoquery.process_versions(rval['results'].strip())
+            formatted_versions = self.format_versions(processed_versions)
+
+            rval['package_found'] = True
+            rval['versions'] = formatted_versions
+            rval['package_name'] = self.name
+
+            if self.verbose:
+                rval['raw_versions'] = processed_versions
+            else:
+                del rval['results']
+
+        # No packages found
+        else:
+            rval['package_found'] = False
+
+        return rval
+
+    @staticmethod
+    def run_ansible(params, check_mode):
+        '''run the ansible idempotent code'''
+
+        repoquery = Repoquery(
+            params['name'],
+            params['query_type'],
+            params['show_duplicates'],
+            params['match_version'],
+            params['verbose'],
+        )
+
+        state = params['state']
+
+        if state == 'list':
+            results = repoquery.repoquery()
+
+            if results['returncode'] != 0:
+                return {'failed': True,
+                        'msg': results}
+
+            return {'changed': False, 'results': results, 'state': 'list', 'check_mode': check_mode}
+
+        return {'failed': True,
+                'changed': False,
+                'msg': 'Unknown state passed. %s' % state,
+                'state': 'unknown'}
-- 
cgit v1.2.3