|
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)
|