Blame progit/app.py

Pierre-Yves Chibon 2088eb
#-*- coding: utf-8 -*-
Pierre-Yves Chibon 2088eb
Pierre-Yves Chibon 2088eb
"""
Pierre-Yves Chibon 2088eb
 (c) 2014 - Copyright Red Hat Inc
Pierre-Yves Chibon 2088eb
Pierre-Yves Chibon 2088eb
 Authors:
Pierre-Yves Chibon 2088eb
   Pierre-Yves Chibon <pingou@pingoured.fr></pingou@pingoured.fr>
Pierre-Yves Chibon 2088eb
Pierre-Yves Chibon 2088eb
"""
Pierre-Yves Chibon 2088eb
Pierre-Yves Chibon 2088eb
import flask
Pierre-Yves Chibon 2088eb
import os
Pierre-Yves Chibon c55274
import tempfile
Pierre-Yves Chibon 2088eb
from math import ceil
Pierre-Yves Chibon 2088eb
Pierre-Yves Chibon 2088eb
import pygit2
Pierre-Yves Chibon 2088eb
from sqlalchemy.exc import SQLAlchemyError
Pierre-Yves Chibon 2088eb
from pygments import highlight
Pierre-Yves Chibon 2088eb
from pygments.lexers import guess_lexer
Pierre-Yves Chibon 2088eb
from pygments.lexers.text import DiffLexer
Pierre-Yves Chibon 2088eb
from pygments.formatters import HtmlFormatter
Pierre-Yves Chibon 2088eb
Pierre-Yves Chibon 2088eb
Pierre-Yves Chibon 8a621f
import progit.exceptions
Pierre-Yves Chibon 8a621f
import progit.lib
Pierre-Yves Chibon 8a621f
import progit.forms
Pierre-Yves Chibon 47950c
from progit import APP, SESSION, LOG, __get_file_in_tree, cla_required
Pierre-Yves Chibon 2088eb
Pierre-Yves Chibon 2088eb
Pierre-Yves Chibon 2088eb
### Application
Pierre-Yves Chibon 2088eb
Pierre-Yves Chibon 2088eb
Pierre-Yves Chibon 47950c
@APP.route('/')
Pierre-Yves Chibon 47950c
def index():
Pierre-Yves Chibon 47950c
    """ Front page of the application.
Pierre-Yves Chibon 2088eb
    """
Pierre-Yves Chibon 47950c
    page = flask.request.args.get('page', 1)
Pierre-Yves Chibon 2088eb
    try:
Pierre-Yves Chibon 47950c
        page = int(page)
Pierre-Yves Chibon 2088eb
    except ValueError:
Pierre-Yves Chibon 2088eb
        page = 1
Pierre-Yves Chibon 2088eb
Pierre-Yves Chibon 2088eb
    limit = APP.config['ITEM_PER_PAGE']
Pierre-Yves Chibon 2088eb
    start = limit * (page - 1)
Pierre-Yves Chibon 2088eb
Pierre-Yves Chibon 47950c
    repos = progit.lib.list_projects(
Pierre-Yves Chibon 47950c
        SESSION,
Pierre-Yves Chibon 47950c
        fork=False,
Pierre-Yves Chibon 47950c
        start=start,
Pierre-Yves Chibon 47950c
        limit=limit)
Pierre-Yves Chibon 47950c
    num_repos = progit.lib.list_projects(
Pierre-Yves Chibon 47950c
        SESSION,
Pierre-Yves Chibon 47950c
        fork=False,
Pierre-Yves Chibon 47950c
        count=True)
Pierre-Yves Chibon 47950c
Pierre-Yves Chibon 47950c
    total_page = int(ceil(num_repos / float(limit)))
Pierre-Yves Chibon bd2213
Pierre-Yves Chibon 2088eb
    return flask.render_template(
Pierre-Yves Chibon 47950c
        'index.html',
Pierre-Yves Chibon 47950c
        repos=repos,
Pierre-Yves Chibon 2088eb
        total_page=total_page,
Pierre-Yves Chibon 47950c
        page=page,
Pierre-Yves Chibon 2088eb
    )
Pierre-Yves Chibon 2088eb
Pierre-Yves Chibon 2088eb
Pierre-Yves Chibon 47950c
@APP.route('/users/')
Pierre-Yves Chibon 47950c
def view_users():
Pierre-Yves Chibon 47950c
    """ Present the list of users.
Pierre-Yves Chibon 2088eb
    """
Pierre-Yves Chibon 47950c
    page = flask.request.args.get('page', 1)
Pierre-Yves Chibon 47950c
    try:
Pierre-Yves Chibon 47950c
        page = int(page)
Pierre-Yves Chibon 47950c
    except ValueError:
Pierre-Yves Chibon 47950c
        page = 1
Mathieu Bridon 4c73b0
Pierre-Yves Chibon 47950c
    ## TODO: retrieve this from the DB
Pierre-Yves Chibon 47950c
    users = ['pingou']
Mathieu Bridon 4c73b0
Pierre-Yves Chibon 47950c
    limit = APP.config['ITEM_PER_PAGE']
Pierre-Yves Chibon 47950c
    start = limit * (page - 1)
Pierre-Yves Chibon 47950c
    end = limit * page
Pierre-Yves Chibon 47950c
    users_length = len(users)
Pierre-Yves Chibon 47950c
    users = users[start:end]
Pierre-Yves Chibon 2088eb
Pierre-Yves Chibon 47950c
    total_page = int(ceil(users_length / float(limit)))
Pierre-Yves Chibon 2088eb
Pierre-Yves Chibon 2088eb
    return flask.render_template(
Pierre-Yves Chibon 47950c
        'user_list.html',
Pierre-Yves Chibon 47950c
        users=users,
Pierre-Yves Chibon 47950c
        total_page=total_page,
Pierre-Yves Chibon 47950c
        page=page,
Pierre-Yves Chibon 2088eb
    )
Pierre-Yves Chibon 2088eb
Pierre-Yves Chibon 84f83a
Pierre-Yves Chibon 47950c
@APP.route('/user/<username>')</username>
Pierre-Yves Chibon 47950c
def view_user(username):
Pierre-Yves Chibon 47950c
    """ Front page of a specific user.
Pierre-Yves Chibon 2088eb
    """
Pierre-Yves Chibon 2088eb
Pierre-Yves Chibon 47950c
    repopage = flask.request.args.get('repopage', 1)
Pierre-Yves Chibon 2088eb
    try:
Pierre-Yves Chibon 47950c
        repopage = int(repopage)
Pierre-Yves Chibon 2088eb
    except ValueError:
Pierre-Yves Chibon 47950c
        repopage = 1
Mathieu Bridon 6dd1b4
Pierre-Yves Chibon 47950c
    forkpage = flask.request.args.get('forkpage', 1)
Pierre-Yves Chibon 47950c
    try:
Pierre-Yves Chibon 47950c
        forkpage = int(forkpage)
Pierre-Yves Chibon 47950c
    except ValueError:
Pierre-Yves Chibon 47950c
        forkpage = 1
Pierre-Yves Chibon 12e862
Pierre-Yves Chibon 47950c
    limit = APP.config['ITEM_PER_PAGE']
Pierre-Yves Chibon 47950c
    repo_start = limit * (repopage - 1)
Pierre-Yves Chibon 47950c
    fork_start = limit * (forkpage - 1)
Pierre-Yves Chibon 47950c
Pierre-Yves Chibon 47950c
    repos = progit.lib.list_projects(
Pierre-Yves Chibon 47950c
        SESSION,
Pierre-Yves Chibon 47950c
        username=flask.g.fas_user.username,
Pierre-Yves Chibon 47950c
        fork=False,
Pierre-Yves Chibon 47950c
        start=repo_start,
Pierre-Yves Chibon 47950c
        limit=limit)
Pierre-Yves Chibon 47950c
    repos_length = progit.lib.list_projects(
Pierre-Yves Chibon 47950c
        SESSION,
Pierre-Yves Chibon 47950c
        username=flask.g.fas_user.username,
Pierre-Yves Chibon 47950c
        fork=False,
Pierre-Yves Chibon 47950c
        count=True)
Pierre-Yves Chibon 47950c
Pierre-Yves Chibon 47950c
    forks = progit.lib.list_projects(
Pierre-Yves Chibon 47950c
        SESSION,
Pierre-Yves Chibon 47950c
        username=flask.g.fas_user.username,
Pierre-Yves Chibon 47950c
        fork=True,
Pierre-Yves Chibon 47950c
        start=fork_start,
Pierre-Yves Chibon 47950c
        limit=limit)
Pierre-Yves Chibon 47950c
    forks_length = progit.lib.list_projects(
Pierre-Yves Chibon 47950c
        SESSION,
Pierre-Yves Chibon 47950c
        username=flask.g.fas_user.username,
Pierre-Yves Chibon 47950c
        fork=True,
Pierre-Yves Chibon 47950c
        count=True)
Pierre-Yves Chibon 47950c
Pierre-Yves Chibon 47950c
    total_page_repos = int(ceil(repos_length / float(limit)))
Pierre-Yves Chibon 47950c
    total_page_forks = int(ceil(forks_length / float(limit)))
Pierre-Yves Chibon 47950c
Pierre-Yves Chibon 47950c
    repos_obj = [
Pierre-Yves Chibon 47950c
        pygit2.Repository(
Pierre-Yves Chibon 47950c
            os.path.join(APP.config['GIT_FOLDER'], repo.path))
Pierre-Yves Chibon 47950c
        for repo in repos]
Pierre-Yves Chibon 47950c
Pierre-Yves Chibon 47950c
    forks_obj = [
Pierre-Yves Chibon 47950c
        pygit2.Repository(
Pierre-Yves Chibon 47950c
            os.path.join(APP.config['FORK_FOLDER'], repo.path))
Pierre-Yves Chibon 47950c
        for repo in forks]
Pierre-Yves Chibon 12e862
Pierre-Yves Chibon 12e862
    return flask.render_template(
Pierre-Yves Chibon 47950c
        'user_info.html',
Pierre-Yves Chibon ac8023
        username=username,
Pierre-Yves Chibon 47950c
        repos=repos,
Pierre-Yves Chibon 47950c
        repos_obj=repos_obj,
Pierre-Yves Chibon 47950c
        total_page_repos=total_page_repos,
Pierre-Yves Chibon 47950c
        forks=forks,
Pierre-Yves Chibon 47950c
        forks_obj=forks_obj,
Pierre-Yves Chibon 47950c
        total_page_forks=total_page_forks,
Pierre-Yves Chibon 47950c
        repopage=repopage,
Pierre-Yves Chibon 47950c
        forkpage=forkpage,
Pierre-Yves Chibon 12e862
    )
Pierre-Yves Chibon eb2817
Pierre-Yves Chibon eb2817
Pierre-Yves Chibon a204a8
Pierre-Yves Chibon 47950c
@APP.route('/new/', methods=('GET', 'POST'))
Pierre-Yves Chibon 47950c
@cla_required
Pierre-Yves Chibon 47950c
def new_project():
Pierre-Yves Chibon 47950c
    """ Form to create a new project.
Pierre-Yves Chibon a204a8
    """
Pierre-Yves Chibon 47950c
    form = progit.forms.ProjectForm()
Pierre-Yves Chibon a204a8
    if form.validate_on_submit():
Pierre-Yves Chibon 47950c
        name = form.name.data
Pierre-Yves Chibon 47950c
        description = form.description.data
Pierre-Yves Chibon a204a8
Pierre-Yves Chibon a204a8
        try:
Pierre-Yves Chibon 47950c
            message = progit.lib.new_project(
Pierre-Yves Chibon a204a8
                SESSION,
Pierre-Yves Chibon 47950c
                name=name,
Pierre-Yves Chibon 47950c
                description=description,
Pierre-Yves Chibon a204a8
                user=flask.g.fas_user.username,
Pierre-Yves Chibon 47950c
                folder=APP.config['GIT_FOLDER'],
Pierre-Yves Chibon a204a8
            )
Pierre-Yves Chibon a204a8
            SESSION.commit()
Pierre-Yves Chibon a204a8
            flask.flash(message)
Pierre-Yves Chibon 47950c
            return flask.redirect(flask.url_for('view_repo', repo=name))
Pierre-Yves Chibon 542ff6
        except progit.exceptions.ProgitException, err:
Pierre-Yves Chibon 542ff6
            flask.flash(str(err), 'error')
Pierre-Yves Chibon 542ff6
        except SQLAlchemyError, err:  # pragma: no cover
Pierre-Yves Chibon 542ff6
            SESSION.rollback()
Pierre-Yves Chibon 542ff6
            flask.flash(str(err), 'error')
Pierre-Yves Chibon 542ff6
Pierre-Yves Chibon 542ff6
    return flask.render_template(
Pierre-Yves Chibon 47950c
        'new_project.html',
Pierre-Yves Chibon 542ff6
        form=form,
Pierre-Yves Chibon c30d42
    )
Pierre-Yves Chibon c30d42