118 lines
3.3 KiB
Python
118 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')
|