pv2/pv2/scripts/import_pkg.py

#!/usr/bin/python3
# This is called to do imports, whether from an RPM or a git repo (e.g. CentOS
# stream gitlab)

import argparse
import pv2.importer as importutil

parser = argparse.ArgumentParser(description="Importer Utility")
subparser = parser.add_subparsers(dest='cmd')
subparser.required = True

rpm_parser = subparser.add_parser('rpm')
git_parser = subparser.add_parser('git')

rpm_parser.add_argument('--gituser', type=str, required=False, default='git')
rpm_parser.add_argument('--giturl', type=str, required=True)
rpm_parser.add_argument('--branch', type=str, required=True)
rpm_parser.add_argument('--srpm', type=str, required=True)
rpm_parser.add_argument('--release', type=str, required=False, default='')
rpm_parser.add_argument('--gitorg', type=str, required=False, default='rpms')
rpm_parser.add_argument('--distprefix', type=str, required=False, default='el')
rpm_parser.add_argument('--dest-lookaside', type=str, required=False, default='/var/www/html/sources')
rpm_parser.add_argument('--no-verify-signature', action='store_true')
rpm_parser.add_argument('--skip-lookaside-upload',
                        action='store_true',
                        help='Set this flag to skip uploading to /var/www/html/sources esque lookaside')
rpm_parser.add_argument('--upload-to-s3', action='store_true')
rpm_parser.add_argument('--aws-access-key-id', type=str, required=False, default='')
rpm_parser.add_argument('--aws-access-key', type=str, required=False, default='')
rpm_parser.add_argument('--aws-bucket', type=str, required=False, default='')

git_parser.add_argument('--name', type=str, required=True)
git_parser.add_argument('--source-gituser', type=str, required=False, default='git')
git_parser.add_argument('--source-giturl', type=str, required=True)
git_parser.add_argument('--source-gitorg', type=str, required=True)
git_parser.add_argument('--source-branch', type=str, required=True)
git_parser.add_argument('--dest-gituser', type=str, required=False, default='git')
git_parser.add_argument('--dest-giturl', type=str, required=True)
git_parser.add_argument('--dest-gitorg', type=str, required=False, default='rpms')
git_parser.add_argument('--dest-branch', type=str, required=False, default='')
git_parser.add_argument('--release', type=str, required=False, default='')
git_parser.add_argument('--distprefix', type=str, required=False, default='el')
git_parser.add_argument('--dest-lookaside', type=str, required=False, default='/var/www/html/sources')
git_parser.add_argument('--upstream-lookaside',
                        choices=('rocky8', 'rocky', 'centos', 'stream', 'fedora'),
                        required=True)
git_parser.add_argument('--alternate-spec-name',
                        type=str, required=False,
                        default='',
                        help='ex: if kernel-rt, use kernel. only use if built-in finder is failing')
git_parser.add_argument('--skip-lookaside-upload',
                        action='store_true',
                        help='Set this flag to skip uploading to /var/www/html/sources esque lookaside')
git_parser.add_argument('--upload-to-s3', action='store_true')
git_parser.add_argument('--aws-access-key-id', type=str, required=False, default='')
git_parser.add_argument('--aws-access-key', type=str, required=False, default='')
git_parser.add_argument('--aws-bucket', type=str, required=False, default='')

results = parser.parse_args()
command = parser.parse_args().cmd

def main():
    """
    Run the main program. Callable via poetry or __main__
    """
    if command == 'rpm':
        classy = importutil.SrpmImport(
                git_url_path=results.giturl,
                srpm_path=results.srpm,
                release=results.release,
                branch=results.branch,
                distprefix=results.distprefix,
                git_user=results.gituser,
                org=results.gitorg,
                dest_lookaside=results.dest_lookaside,
                verify_signature=results.no_verify_signature,
                aws_access_key_id=results.aws_access_key_id,
                aws_access_key=results.aws_access_key,
                aws_bucket=results.aws_bucket,
        )
        classy.pkg_import(skip_lookaside=results.skip_lookaside_upload,
                          s3_upload=results.upload_to_s3)
    elif command == 'git':
        classy = importutil.GitImport(
                package=results.name,
                source_git_user=results.source_gituser,
                source_git_url_path=results.source_giturl,
                source_git_org_path=results.source_gitorg,
                dest_git_user=results.dest_gituser,
                dest_git_url_path=results.dest_giturl,
                dest_org=results.dest_gitorg,
                release=results.release,
                source_branch=results.source_branch,
                dest_branch=results.dest_branch,
                upstream_lookaside=results.upstream_lookaside,
                distprefix=results.distprefix,
                alternate_spec_name=results.alternate_spec_name,
                dest_lookaside=results.dest_lookaside,
                aws_access_key_id=results.aws_access_key_id,
                aws_access_key=results.aws_access_key,
                aws_bucket=results.aws_bucket,
        )
        classy.pkg_import(skip_lookaside=results.skip_lookaside_upload,
                          s3_upload=results.upload_to_s3)
    else:
        print('Unknown command')

if __name__ == '__main__':
    main()