[fad9e0]: Allura / allura / tasks / repo_tasks.py History

Child: [bea16d] (diff)

Download this file

repo_tasks.py    105 lines (95 with data), 3.7 kB 

  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
import shutil

import logging

import traceback



from pylons import c, g



from allura.lib.decorators import task

from allura.lib.repository import RepositoryApp



@task

def init(**kwargs):

    from allura import model as M

    c.app.repo.init()

    M.Notification.post_user(

        c.user, c.app.repo, 'created',

        text='Repository %s/%s created' % (

            c.project.shortname, c.app.config.options.mount_point))



@task

def clone(cloned_from_path, cloned_from_name, cloned_from_url):

    from allura import model as M

    try:

        c.app.repo.init_as_clone(

            cloned_from_path,

            cloned_from_name,

            cloned_from_url)

        M.Notification.post_user(

            c.user, c.app.repo, 'created',

            text='Repository %s/%s created' % (

                c.project.shortname, c.app.config.options.mount_point))

    except Exception, e:

        g.post_event('repo_clone_task_failed', cloned_from_url, cloned_from_path, traceback.format_exc())



@task

def reclone(*args, **kwargs):

    from allura import model as M

    from ming.orm import ThreadLocalORMSession

    repo = c.app.repo

    if repo is not None:

        shutil.rmtree(repo.full_fs_path, ignore_errors=True)

        repo.delete()

    ThreadLocalORMSession.flush_all()

    M.MergeRequest.query.remove(dict(

            app_config_id=c.app.config._id))

    clone(*args, **kwargs)



@task

def refresh(**kwargs):

    from allura import model as M

    log = logging.getLogger(__name__)

    #don't create multiple refresh tasks

    q = {

        'task_name': 'allura.tasks.repo_tasks.refresh',

        'state': {'$in': ['busy', 'ready']},

        'context.app_config_id': c.app.config._id,

        'context.project_id': c.project._id,

    }

    refresh_tasks_count = M.MonQTask.query.find(q).count()

    if refresh_tasks_count <= 1: #only this task

        c.app.repo.refresh()

        #checking if we have new commits arrived

        #during refresh and re-queue task if so

        new_commit_ids = c.app.repo.unknown_commit_ids()

        if len(new_commit_ids) > 0:

            refresh.post()

            log.info('New refresh task is queued due to new commit(s).')

    else:

        log.info('Refresh task for %s:%s skipped due to backlog', c.project.shortname, c.app.config.options.mount_point)



@task

def uninstall(**kwargs):

    from allura import model as M

    repo = c.app.repo

    if repo is not None:

        shutil.rmtree(repo.full_fs_path, ignore_errors=True)

        repo.delete()

    M.MergeRequest.query.remove(dict(

            app_config_id=c.app.config._id))

    super(RepositoryApp, c.app).uninstall(c.project)

    from ming.orm import ThreadLocalORMSession

    ThreadLocalORMSession.flush_all()



@task

def nop():

    log = logging.getLogger(__name__)

    log.info('nop')



@task

def reclone_repo(*args, **kwargs):

    from allura import model as M

    try:

        nbhd = M.Neighborhood.query.get(url_prefix='/%s/' % kwargs['prefix'])

        c.project = M.Project.query.get(shortname=kwargs['shortname'], neighborhood_id=nbhd._id)

        c.app = c.project.app_instance(kwargs['mount_point'])

        source_url = c.app.config.options.get('init_from_url')

        source_path = c.app.config.options.get('init_from_path')

        c.app.repo.init_as_clone(source_path, None, source_url)

        M.Notification.post_user(

            c.user, c.app.repo, 'created',

            text='Repository %s/%s created' % (

                c.project.shortname, c.app.config.options.mount_point))

    except Exception, e:

        source_url = source_path or source_url

        g.post_event('repo_clone_task_failed', source_url, traceback.format_exc())