120 lines
4.2 KiB
Python
120 lines
4.2 KiB
Python
import functools
|
|
import logging
|
|
|
|
from flask import Blueprint, render_template
|
|
|
|
from pillar.api.utils import jsonify
|
|
from pillar.web.system_util import pillar_api
|
|
import pillarsdk
|
|
|
|
blueprint = Blueprint('attract', __name__)
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
@blueprint.route('/')
|
|
def index():
|
|
return render_template('attract/index.html')
|
|
|
|
|
|
def error_project_not_setup_for_attract():
|
|
return render_template('attract/errors/project_not_setup.html')
|
|
|
|
|
|
def attract_project_view(extra_project_projections=None, extension_props=False):
|
|
"""Decorator, replaces the first parameter project_url with the actual project.
|
|
|
|
Assumes the first parameter to the decorated function is 'project_url'. It then
|
|
looks up that project, checks that it's set up for Attract, and passes it to the
|
|
decorated function.
|
|
|
|
If not set up for attract, uses error_project_not_setup_for_attract() to render
|
|
the response.
|
|
|
|
:param extra_project_projections: extra projections to use on top of the ones already
|
|
used by this decorator.
|
|
:type extra_project_projections: dict
|
|
:param extension_props: whether extension properties should be included. Includes them
|
|
in the projections, and verifies that they are there.
|
|
:type extension_props: bool
|
|
"""
|
|
|
|
from .node_types.task import node_type_task
|
|
from . import EXTENSION_NAME
|
|
|
|
if callable(extra_project_projections):
|
|
raise TypeError('Use with @attract_project_view() <-- note the parentheses')
|
|
|
|
projections = {
|
|
'_id': 1,
|
|
'name': 1,
|
|
'node_types': 1,
|
|
# We don't need this here, but this way the wrapped function has access
|
|
# to the orignal URL passed to it.
|
|
'url': 1,
|
|
}
|
|
if extra_project_projections:
|
|
projections.update(extra_project_projections)
|
|
if extension_props:
|
|
projections['extension_props.%s' % EXTENSION_NAME] = 1
|
|
|
|
def decorator(wrapped):
|
|
@functools.wraps(wrapped)
|
|
def wrapper(project_url, *args, **kwargs):
|
|
api = pillar_api()
|
|
|
|
project = pillarsdk.Project.find_by_url(
|
|
project_url,
|
|
{'projection': projections},
|
|
api=api)
|
|
|
|
node_type_name = node_type_task['name']
|
|
node_type = project.get_node_type(node_type_name)
|
|
if not node_type:
|
|
log.warning('createProject url=%s does not have node type %r',
|
|
project_url, node_type_name)
|
|
return error_project_not_setup_for_attract()
|
|
|
|
if extension_props:
|
|
try:
|
|
pprops = project.extension_props.attract
|
|
except AttributeError:
|
|
log.warning("attract_project_view(%s): Project url=%r doesn't have any"
|
|
" extension properties.", wrapped, project_url)
|
|
if log.isEnabledFor(logging.DEBUG):
|
|
import pprint
|
|
log.debug('Project: %s', pprint.pformat(project.to_dict()))
|
|
return error_project_not_setup_for_attract()
|
|
|
|
if pprops is None:
|
|
log.warning("attract_project_view(%s): Project url=%r doesn't have Attract"
|
|
" extension properties.", wrapped, project_url)
|
|
return error_project_not_setup_for_attract()
|
|
|
|
return wrapped(project, pprops, *args, **kwargs)
|
|
|
|
return wrapped(project, *args, **kwargs)
|
|
|
|
return wrapper
|
|
|
|
return decorator
|
|
|
|
|
|
@blueprint.route('/<project_url>/subversion/kick')
|
|
@attract_project_view(extension_props=True)
|
|
def subversion_kick(project, attract_props):
|
|
from . import subversion
|
|
|
|
svn_server_url = attract_props.svn_url # 'svn://localhost/agent327'
|
|
log.info('Re-examining SVN server %s', svn_server_url)
|
|
client = subversion.obtain(svn_server_url)
|
|
|
|
# TODO: last_seen_revision should be stored, probably at the project level.
|
|
last_seen_revision = 0
|
|
observer = subversion.CommitLogObserver(client, last_seen_revision=last_seen_revision)
|
|
observer.fetch_and_observe()
|
|
|
|
return jsonify({
|
|
'previous_last_seen_revision': last_seen_revision,
|
|
'last_seen_revision': observer.last_seen_revision,
|
|
})
|