Add RPM plugin to client

This plugin implements commands for listing, showing, creating and
updating RPM metadata in PDC. When retrieving information about
particular RPM (or when updating), the database identifier is used. This
could be replaced by NEVRA in the future, but it would require
dependency on a library that would parse it.

JIRA: PDC-1023

pdc_client/plugins/rpm.py

@@ -0,0 +1,121 @@
+# -*- coding: utf-8 -*-
+#
+# Copyright (c) 2015 Red Hat
+# Licensed under The MIT License (MIT)
+# http://opensource.org/licenses/MIT
+#
+import sys
+import json
+
+from pdc_client.plugins import PDCClientPlugin, get_paged, add_parser_arguments, extract_arguments
+
+
+class RPMPlugin(PDCClientPlugin):
+    def register(self):
+        subcmd = self.add_command('rpm-list', help='list all rpms')
+        filters = ('name version release arch compose conflicts obsoletes provides '
+                   'suggests recommends requires'.split())
+        for arg in filters:
+            subcmd.add_argument('--' + arg, dest='filter_' + arg)
+        subcmd.set_defaults(func=self.rpm_list)
+
+        subcmd = self.add_command('rpm-info', help='display details of an RPM')
+        subcmd.add_argument('rpmid', metavar='ID')
+        subcmd.set_defaults(func=self.rpm_info)
+
+        subcmd = self.add_admin_command('rpm-create',
+                                        help='create new RPM')
+        self.add_rpm_arguments(subcmd)
+        subcmd.set_defaults(func=self.rpm_create)
+
+        subcmd = self.add_admin_command('rpm-update',
+                                        help='update existing RPM')
+        subcmd.add_argument('rpmid', metavar='ID')
+        self.add_rpm_arguments(subcmd)
+        subcmd.set_defaults(func=self.rpm_update)
+
+    def add_rpm_arguments(self, parser):
+        add_parser_arguments(parser, {
+            'arch': {},
+            'epoch': {'type': int},
+            'filename': {},
+            'name': {},
+            'release': {},
+            'srpm_name': {},
+            'srpm_nevra': {},
+            'version': {},
+            'linked_releases': {'nargs': '*', 'metavar': 'RELEASE_ID'}})
+        add_parser_arguments(parser, {
+            'dependencies__requires': {'nargs': '*', 'metavar': 'DEPENDENCY', 'arg': 'requires'},
+            'dependencies__provides': {'nargs': '*', 'metavar': 'DEPENDENCY', 'arg': 'provides'},
+            'dependencies__suggests': {'nargs': '*', 'metavar': 'DEPENDENCY', 'arg': 'suggests'},
+            'dependencies__obsoletes': {'nargs': '*', 'metavar': 'DEPENDENCY', 'arg': 'obsoletes'},
+            'dependencies__recommends': {'nargs': '*', 'metavar': 'DEPENDENCY', 'arg': 'recommends'},
+            'dependencies__conflicts': {'nargs': '*', 'metavar': 'DEPENDENCY', 'arg': 'conflicts'}},
+            group='Dependencies (optional)')
+
+    def rpm_list(self, args):
+        filters = extract_arguments(args, prefix='filter_')
+        if not filters:
+            sys.stderr.write('At least some filter must be used.\n')
+            sys.exit(1)
+        rpms = get_paged(self.client.rpms._, **filters)
+        if args.json:
+            print json.dumps(list(rpms))
+            return
+
+        for rpm in rpms:
+            print '{id:10} {name:45} {epoch}:{version}-{release}.{arch}'.format(**rpm)
+
+    def rpm_info(self, args, rpm_id=None):
+        response = self.client.rpms[rpm_id or args.rpmid]._()
+        if args.json:
+            print json.dumps(response)
+            return
+        fmt = '{:20} {}'
+        print fmt.format('ID', response['id'])
+        print fmt.format('Name', response['name'])
+        print fmt.format('Epoch', response['epoch'])
+        print fmt.format('Version', response['version'])
+        print fmt.format('Release', response['release'])
+        print fmt.format('Arch', response['arch'])
+        print fmt.format('SRPM Name', response['srpm_name'])
+        print fmt.format('SRPM NEVRA', response['srpm_nevra'] or '')
+        print fmt.format('Filename', response['filename'])
+
+        if response['linked_composes']:
+            print '\nIncluded in composes:'
+            for compose in sorted(response['linked_composes']):
+                print compose
+
+        if response['linked_releases']:
+            print '\nLinked to releases:'
+            for release in sorted(response['linked_releases']):
+                print release
+
+        for type in ('recommends', 'suggests', 'obsoletes', 'provides', 'conflicts', 'requires'):
+            if response['dependencies'][type]:
+                print '\n{}:'.format(type.capitalize())
+                for dep in response['dependencies'][type]:
+                    print dep
+
+    def rpm_create(self, args):
+        data = {}
+        for key, value in args.__dict__.iteritems():
+            if key.startswith('data_') and value is not None:
+                data[key[5:]] = value if value != '' else None
+        self.logger.debug('Creating rpm with data %r', data)
+        response = self.client.rpms._(data)
+        self.rpm_info(args, response['id'])
+
+    def rpm_update(self, args):
+        data = extract_arguments(args)
+        if data:
+            self.logger.debug('Updating rpm %s with data %r', args.rpmid, data)
+            self.client.rpms[args.rpmid]._ += data
+        else:
+            self.logger.debug('Empty data, skipping request')
+        self.rpm_info(args)
+
+
+PLUGIN_CLASSES = [RPMPlugin]