pagure/files/load_from_disk.py

#!/usr/bin/env python

import argparse
import requests
import os

from sqlalchemy.exc import SQLAlchemyError

if 'PAGURE_CONFIG' not in os.environ \
        and os.path.exists('/etc/pagure/pagure.cfg'):
    print 'Using configuration file `/etc/pagure/pagure.cfg`'
    os.environ['PAGURE_CONFIG'] = '/etc/pagure/pagure.cfg'

import pagure
import pagure.lib
import pagure.lib.model


def get_poc_of_pkgs(debug=False):
    """ Retrieve a dictionary giving the point of contact of each package
    in pkgdb.
    """
    if debug:
        print 'Querying pkgdb'
    PKGDB_URL = 'https://admin.stg.fedoraproject.org/pkgdb/api/'
    req = requests.get(PKGDB_URL + 'bugzilla').text
    if debug:
        print 'Pkgdb data retrieved, getting POC'
    pkgs = {}
    for line in req.split('\n'):
        line = line.strip()
        if not line or line.startswith('#'):
            continue
        line = line.split('|')
        if len(line) < 4:
            continue
        pkgs[line[1]] = line[3]

    return pkgs


def main(folder, debug=False):
    """
    Logic:
    - Query the list of maintainer/PoC from pkgdb
    - Browse the directory
    - For each git in the directory, create the project with the correct POC
    """
    pocs = get_poc_of_pkgs(debug=debug)

    if debug:
        print 'Adding the user to the DB'
    for user in sorted(set(pocs.values())):
        if debug:
            print user
        try:
            pagure.lib.set_up_user(
                session=pagure.SESSION,
                username=user,
                fullname=user,
                default_email='%s@fedoraproject.org' % user,
                keydir=pagure.APP.config.get('GITOLITE_KEYDIR', None),
            )
            pagure.SESSION.commit()
        except SQLAlchemyError as err:
            pagure.SESSION.rollback()
            print 'ERROR with user %s' % user
            print err

    for project in sorted(os.listdir(folder)):
        if debug:
            print project

        if not project.endswith('.git'):
            if debug:
                print '  -skip: not a git repository'
            continue

        if project.split('.git')[0] not in pocs:
            if debug:
                print '  -skip: no pocs'
            continue

        try:
            name = project.split('.git')[0]
            orig_name = name
            name = 'rpms/%s' % name
            if name in pagure.APP.config['BLACKLISTED_PROJECTS']:
                raise pagure.exceptions.RepoExistsException(
                    'No project "%s" are allowed to be created due to potential '
                    'conflicts in URLs with pagure itself' % name
                )

            user_obj = pagure.lib.__get_user(pagure.SESSION, pocs[orig_name])
            allowed_prefix = pagure.APP.config[
                'ALLOWED_PREFIX'] + [grp for grp in user_obj.groups]

            first_part, _, second_part = name.partition('/')
            if second_part and first_part not in allowed_prefix:
                raise pagure.exceptions.PagureException(
                    'The prefix of your project must be in the list of allowed '
                    'prefixes set by the admins of this pagure instance, or the name '
                    'of a group of which you are a member.'
                )

            gitfolder = pagure.APP.config['GIT_FOLDER']
            docfolder = pagure.APP.config['DOCS_FOLDER']
            ticketfolder = pagure.APP.config['TICKETS_FOLDER']
            requestfolder = pagure.APP.config['REQUESTS_FOLDER']

            gitrepo = os.path.join(gitfolder, '%s.git' % name)

            project = pagure.lib.model.Project(
                name=name,
                description=None,
                url=None,
                avatar_email=None,
                user_id=user_obj.id,
                parent_id=None,
                hook_token=pagure.lib.login.id_generator(40)
            )
            pagure.SESSION.add(project)
            # Make sure we won't have SQLAlchemy error before we create the repo
            pagure.SESSION.flush()

            http_clone_file = os.path.join(gitrepo, 'git-daemon-export-ok')
            if not os.path.exists(http_clone_file):
                with open(http_clone_file, 'w') as stream:
                    pass

            docrepo = os.path.join(docfolder, project.path)
            if os.path.exists(docrepo):
                shutil.rmtree(gitrepo)
                raise pagure.exceptions.RepoExistsException(
                    'The docs repo "%s" already exists' % project.path
                )
            pygit2.init_repository(docrepo, bare=True)

            ticketrepo = os.path.join(ticketfolder, project.path)
            if os.path.exists(ticketrepo):
                shutil.rmtree(gitrepo)
                shutil.rmtree(docrepo)
                raise pagure.exceptions.RepoExistsException(
                    'The tickets repo "%s" already exists' % project.path
                )
            pygit2.init_repository(
                ticketrepo, bare=True,
                mode=pygit2.C.GIT_REPOSITORY_INIT_SHARED_GROUP)

            requestrepo = os.path.join(requestfolder, project.path)
            if os.path.exists(requestrepo):
                shutil.rmtree(gitrepo)
                shutil.rmtree(docrepo)
                shutil.rmtree(ticketrepo)
                raise pagure.exceptions.RepoExistsException(
                    'The requests repo "%s" already exists' % project.path
                )
            pygit2.init_repository(
                requestrepo, bare=True,
                mode=pygit2.C.GIT_REPOSITORY_INIT_SHARED_GROUP)

            pagure.SESSION.commit()
        except pagure.exceptions.PagureException as err:
            print 'ERROR with project %s' % project
            print err
        except SQLAlchemyError as err:  # pragma: no cover
            pagure.SESSION.rollback()
            print 'ERROR (DB) with project %s' % project
            print err


if __name__ == '__main__':
    parser = argparse.ArgumentParser(
        description='Script creating projects on pagure based on the git '
        'repos present in the specified folder and the pkgdb information.'
    )
    parser.add_argument(
        'folder',
        help='Folder containing all the git repos of the projects to create')
    parser.add_argument(
        '--debug', dest='debug', action='store_true', default=False,
        help='Print the debugging output')

    args = parser.parse_args()

    main(args.folder, debug=args.debug)