pagure-new/pagure/hooks/pagure_ticket_hook.py

117 lines
3.3 KiB
Python

# -*- coding: utf-8 -*-
"""
(c) 2014-2016 - Copyright Red Hat Inc
Authors:
Pierre-Yves Chibon <pingou@pingoured.fr>
"""
import os
import flask
import flask_wtf as wtf
import sqlalchemy as sa
import wtforms
from sqlalchemy.orm import relation
from sqlalchemy.orm import backref
from pagure.hooks import BaseHook
from pagure.lib.model import BASE, Project
from pagure import APP
class PagureTicketsTable(BASE):
""" Stores information about the pagure tickets hook deployed on a project.
Table -- hook_pagure_tickets
"""
__tablename__ = 'hook_pagure_tickets'
id = sa.Column(sa.Integer, primary_key=True)
project_id = sa.Column(
sa.Integer,
sa.ForeignKey(
'projects.id', onupdate='CASCADE', ondelete='CASCADE'),
nullable=False,
unique=True,
index=True)
active = sa.Column(sa.Boolean, nullable=False, default=False)
project = relation(
'Project', remote_side=[Project.id],
backref=backref(
'pagure_hook_tickets', cascade="delete, delete-orphan",
single_parent=True, uselist=False)
)
class PagureTicketsForm(wtf.Form):
''' Form to configure the pagure hook. '''
active = wtforms.BooleanField(
'Active',
[wtforms.validators.Optional()]
)
class PagureTicketHook(BaseHook):
''' Pagure ticket hook. '''
name = 'Pagure tickets'
description = 'Pagure specific hook to update tickets stored in the database '\
'based on the information pushed in the tickets git repository.'
form = PagureTicketsForm
db_object = PagureTicketsTable
backref = 'pagure_hook_tickets'
form_fields = ['active']
@classmethod
def set_up(cls, project):
''' Install the generic post-receive hook that allow us to call
multiple post-receive hooks as set per plugin.
'''
repopath = os.path.join(APP.config['TICKETS_FOLDER'], project.path)
if not os.path.exists(repopath):
flask.abort(404, 'No git repo found')
hook_files = os.path.join(
os.path.dirname(os.path.realpath(__file__)), 'files')
# Make sure the hooks folder exists
hookfolder = os.path.join(repopath, 'hooks')
if not os.path.exists(hookfolder):
os.makedirs(hookfolder)
# Install the main post-receive file
postreceive = os.path.join(hookfolder, 'post-receive')
hook_file = os.path.join(hook_files, 'post-receive')
if not os.path.exists(postreceive):
os.symlink(hook_file, postreceive)
@classmethod
def install(cls, project, dbobj):
''' Method called to install the hook for a project.
:arg project: a ``pagure.model.Project`` object to which the hook
should be installed
'''
repopaths = [os.path.join(APP.config['TICKETS_FOLDER'], project.path)]
cls.base_install(repopaths, dbobj, 'pagure-ticket',
'pagure_hook_tickets.py')
@classmethod
def remove(cls, project):
''' Method called to remove the hook of a project.
:arg project: a ``pagure.model.Project`` object to which the hook
should be installed
'''
repopaths = [os.path.join(APP.config['TICKETS_FOLDER'], project.path)]
cls.base_remove(repopaths, 'pagure-ticket')