Blame progit/hooks/progit_ticket_hook.py

Pierre-Yves Chibon 1d6d27
#-*- coding: utf-8 -*-
Pierre-Yves Chibon 1d6d27
Pierre-Yves Chibon 1d6d27
"""
Pierre-Yves Chibon 1d6d27
 (c) 2014 - Copyright Red Hat Inc
Pierre-Yves Chibon 1d6d27
Pierre-Yves Chibon 1d6d27
 Authors:
Pierre-Yves Chibon 1d6d27
   Pierre-Yves Chibon <pingou@pingoured.fr></pingou@pingoured.fr>
Pierre-Yves Chibon 1d6d27
Pierre-Yves Chibon 1d6d27
"""
Pierre-Yves Chibon 1d6d27
Pierre-Yves Chibon 1d6d27
import os
Pierre-Yves Chibon 1d6d27
import shutil
Pierre-Yves Chibon 1d6d27
Pierre-Yves Chibon 1d6d27
import sqlalchemy as sa
Pierre-Yves Chibon 1d6d27
import pygit2
Pierre-Yves Chibon 1d6d27
import wtforms
Pierre-Yves Chibon 1d6d27
from flask.ext import wtf
Pierre-Yves Chibon 1d6d27
from sqlalchemy.orm import relation
Pierre-Yves Chibon 1d6d27
Pierre-Yves Chibon 1d6d27
from progit.hooks import BaseHook
Pierre-Yves Chibon 1d6d27
from progit.model import BASE, Project
Pierre-Yves Chibon 1d6d27
from progit import SESSION, APP
Pierre-Yves Chibon 1d6d27
Pierre-Yves Chibon 1d6d27
Pierre-Yves Chibon 1d6d27
class ProgitTicketsTable(BASE):
Pierre-Yves Chibon 1d6d27
    """ Stores information about the progit tickets hook deployed on a project.
Pierre-Yves Chibon 1d6d27
Pierre-Yves Chibon 1d6d27
    Table -- hook_progit_tickets
Pierre-Yves Chibon 1d6d27
    """
Pierre-Yves Chibon 1d6d27
Pierre-Yves Chibon 1d6d27
    __tablename__ = 'hook_progit_tickets'
Pierre-Yves Chibon 1d6d27
Pierre-Yves Chibon 1d6d27
    id = sa.Column(sa.Integer, primary_key=True)
Pierre-Yves Chibon 1d6d27
    project_id = sa.Column(
Pierre-Yves Chibon 1d6d27
        sa.Integer,
Pierre-Yves Chibon 1d6d27
        sa.ForeignKey('projects.id', onupdate='CASCADE'),
Pierre-Yves Chibon 1d6d27
        nullable=False,
Pierre-Yves Chibon 1d6d27
        unique=True,
Pierre-Yves Chibon 1d6d27
        index=True)
Pierre-Yves Chibon 1d6d27
Pierre-Yves Chibon 1d6d27
    active = sa.Column(sa.Boolean, nullable=False, default=False)
Pierre-Yves Chibon 1d6d27
Pierre-Yves Chibon 1d6d27
    project = relation(
Pierre-Yves Chibon 7dd914
        'Project', remote_side=[Project.id], backref='progit_hook_tickets')
Pierre-Yves Chibon 1d6d27
Pierre-Yves Chibon 1d6d27
Pierre-Yves Chibon 1d6d27
class ProgitTicketsForm(wtf.Form):
Pierre-Yves Chibon 1d6d27
    ''' Form to configure the progit hook. '''
Pierre-Yves Chibon 1d6d27
    active = wtforms.BooleanField(
Pierre-Yves Chibon 1d6d27
        'Acive',
Pierre-Yves Chibon 1d6d27
        [wtforms.validators.Optional()]
Pierre-Yves Chibon 1d6d27
    )
Pierre-Yves Chibon 1d6d27
Pierre-Yves Chibon 1d6d27
Pierre-Yves Chibon 1d6d27
class ProgitTicketHook(BaseHook):
Pierre-Yves Chibon 1d6d27
    ''' Progit ticket hook. '''
Pierre-Yves Chibon 1d6d27
Pierre-Yves Chibon 1d6d27
    name = 'progit tickets'
Pierre-Yves Chibon 1d6d27
    form = ProgitTicketsForm
Pierre-Yves Chibon 1d6d27
    db_object = ProgitTicketsTable
Pierre-Yves Chibon 1d6d27
    backref = 'progit_tickets_hook'
Pierre-Yves Chibon 1d6d27
    form_fields = ['active']
Pierre-Yves Chibon 1d6d27
Pierre-Yves Chibon 1d6d27
    @classmethod
Pierre-Yves Chibon 1d6d27
    def install(cls, project, dbobj):
Pierre-Yves Chibon 1d6d27
        ''' Method called to install the hook for a project.
Pierre-Yves Chibon 1d6d27
Pierre-Yves Chibon 1d6d27
        :arg project: a ``progit.model.Project`` object to which the hook
Pierre-Yves Chibon 1d6d27
            should be installed
Pierre-Yves Chibon 1d6d27
Pierre-Yves Chibon 1d6d27
        '''
Pierre-Yves Chibon 1d6d27
        repopath = os.path.join(APP.config['TICKETS_FOLDER'], project.path)
Pierre-Yves Chibon 1d6d27
        hook_files = os.path.join(
Pierre-Yves Chibon 1d6d27
            os.path.dirname(os.path.realpath(__file__)), 'files')
Pierre-Yves Chibon 1d6d27
        repo_obj = pygit2.Repository(repopath)
Pierre-Yves Chibon 1d6d27
Pierre-Yves Chibon 1d6d27
        # Install the hook itself
Pierre-Yves Chibon 1d6d27
        shutil.copyfile(
Pierre-Yves Chibon 1d6d27
            os.path.join(hook_files, 'progit_tickets_hook.py'),
Pierre-Yves Chibon 1d6d27
            os.path.join(repopath, 'hooks', 'post-receive.progit')
Pierre-Yves Chibon 1d6d27
        )
Pierre-Yves Chibon 1d6d27
        os.chmod(
Pierre-Yves Chibon 1d6d27
            os.path.join(repopath, 'hooks', 'post-receive.progit_tickets'),
Pierre-Yves Chibon 1d6d27
            0755)
Pierre-Yves Chibon 1d6d27
Pierre-Yves Chibon 1d6d27
    @classmethod
Pierre-Yves Chibon 1d6d27
    def remove(cls, project):
Pierre-Yves Chibon 1d6d27
        ''' Method called to remove the hook of a project.
Pierre-Yves Chibon 1d6d27
Pierre-Yves Chibon 1d6d27
        :arg project: a ``progit.model.Project`` object to which the hook
Pierre-Yves Chibon 1d6d27
            should be installed
Pierre-Yves Chibon 1d6d27
Pierre-Yves Chibon 1d6d27
        '''
Pierre-Yves Chibon 1d6d27
        repopath = os.path.join(APP.config['TICKETS_FOLDER'], project.path)
Pierre-Yves Chibon 1d6d27
        hook_path = os.path.join(
Pierre-Yves Chibon 1d6d27
            repopath, 'hooks', 'post-receive.progit_tickets')
Pierre-Yves Chibon 1d6d27
        if os.path.exists(hook_path):
Pierre-Yves Chibon 1d6d27
            os.unlink(hook_path)