python-packagecloud/promotepackages at master · denisbr/python-packagecloud · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
#!/usr/bin/env python
"""Promote Packages"""

import argparse
import configparser

from Packagecloud import get_all_packages
from Packagecloud import promote_package


def configure():
    """Configure"""
    parser = argparse.ArgumentParser()
    parser.add_argument('--config', required=True,
                        help='Config file')
    parser.add_argument('--srcrepo', required=True,
                        help='Source Repository')
    parser.add_argument('--destrepo', required=True,
                        help='Destination Repository')
    parser.add_argument('--interactive', action='store_true',
                        help='Interactive Mode')
    parser.add_argument('--distro', required=False,
                        help='Filter selection by distro')
    parser.add_argument('--version', required=False,
                        help='Filter selection by package version')
    parser.add_argument('--name', required=False,
                        help='Filter selection by package name')
    parser.add_argument('--type', required=False,
                        help='Filter selection by package type')
    parser.add_argument('--match', required=False,
                        help='Filter selection by filename substring match')
    parser.add_argument('--debug', action='store_true',
                        help='Print debug output')
    args = parser.parse_args()

    config = configparser.SafeConfigParser()
    config.read(args.config)

    token = config.get('packagecloud', 'token')
    http_scheme = config.get('packagecloud', 'http_scheme')
    api_domain = config.get('packagecloud', 'api_domain')
    api_version = config.get('packagecloud', 'api_version')

    conf = {
        'domain_base': '{}://{}:@{}'.format(
            http_scheme, token, api_domain),
        'url_base': '{}://{}:@{}/api/{}'.format(
            http_scheme, token, api_domain, api_version),
        'repouser': config.get('packagecloud', 'repo_user'),
        'srcrepo': args.srcrepo,
        'destrepo': args.destrepo,
        'debug': args.debug,
        'interactive': args.interactive,
        'distro': args.distro,
        'version': args.version,
        'name': args.name,
        'type': args.type,
        'match': args.match,
        'args': args
    }

    return conf


def main():
    """Main"""
    conf = configure()

    procount = 0
    skpcount = 0
    errcount = 0
    packages = get_all_packages(conf['repouser'], conf['srcrepo'], conf)
    for package in packages:
        if conf['distro'] and not package['distro_version'] == conf['distro']:
            continue
        elif conf['version'] and not package['version'] == conf['version']:
            continue
        elif conf['name'] and not package['name'] == conf['name']:
            continue
        elif conf['type'] and not conf['type'] == package['type']:
            continue
        elif conf['match'] and not conf['match'] in package['filename']:
            continue

        print('Going to promote {:<45} from {:>15}/{:<15} to {:>15}/{:<15}'.
              format(package['filename'], conf['repouser'], conf['srcrepo'],
                     conf['repouser'], conf['destrepo']), end='')

        if conf['interactive']:
            promote = input("Promote this package (y/N)? ").lower() == 'y'
        else:
            promote = True

        if promote:
            promoteresult = promote_package(package, conf['repouser'],
                                            conf['destrepo'], conf)
            if promoteresult:
                print('[OK]')
#                 print('{} promoted to {}/{} successfully'.format(
#                     package['filename'], conf['repouser'], conf['destrepo']))
                procount += 1
            else:
                print('[ERROR]')
                errcount += 1
        else:
            print('Skipped {}'.format(package['filename']))
            skpcount += 1

    print('-----------------------------------------')
    print('Total packages processed: ', (procount + skpcount + errcount))
    print('Promoted: ', procount)
    print('Skipped:  ', skpcount)
    print('Errors:   ', errcount)

main()