|
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 |
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 |
2088eb |
from progit import APP, SESSION, LOG
|
|
Pierre-Yves Chibon |
2088eb |
|
|
Pierre-Yves Chibon |
2088eb |
|
|
Pierre-Yves Chibon |
2088eb |
### Application
|
|
Pierre-Yves Chibon |
2088eb |
@APP.route('/')
|
|
Pierre-Yves Chibon |
2088eb |
def index():
|
|
Pierre-Yves Chibon |
2088eb |
""" Front page of the application.
|
|
Pierre-Yves Chibon |
2088eb |
"""
|
|
Pierre-Yves Chibon |
2088eb |
page = flask.request.args.get('page', 1)
|
|
Pierre-Yves Chibon |
2088eb |
|
|
Pierre-Yves Chibon |
2088eb |
start = APP.config['ITEM_PER_PAGE'] * (page - 1)
|
|
Pierre-Yves Chibon |
2088eb |
end = APP.config['ITEM_PER_PAGE'] * page
|
|
Pierre-Yves Chibon |
2088eb |
repos = sorted(os.listdir(APP.config['GIT_FOLDER']))[start:end]
|
|
Pierre-Yves Chibon |
2088eb |
repos_obj = [
|
|
Pierre-Yves Chibon |
2088eb |
pygit2.Repository(os.path.join(APP.config['GIT_FOLDER'], repo))
|
|
Pierre-Yves Chibon |
2088eb |
for repo in repos]
|
|
Pierre-Yves Chibon |
2088eb |
|
|
Pierre-Yves Chibon |
2088eb |
return flask.render_template(
|
|
Pierre-Yves Chibon |
2088eb |
'index.html',
|
|
Pierre-Yves Chibon |
2088eb |
repos=repos,
|
|
Pierre-Yves Chibon |
2088eb |
repos_obj=repos_obj,
|
|
Pierre-Yves Chibon |
2088eb |
)
|
|
Pierre-Yves Chibon |
2088eb |
|
|
Pierre-Yves Chibon |
2088eb |
|
|
Pierre-Yves Chibon |
2088eb |
@APP.route('/<repo>')</repo>
|
|
Pierre-Yves Chibon |
2088eb |
def view_repo(repo):
|
|
Pierre-Yves Chibon |
2088eb |
""" Front page of a specific repo.
|
|
Pierre-Yves Chibon |
2088eb |
"""
|
|
Pierre-Yves Chibon |
2088eb |
if repo not in os.listdir(APP.config['GIT_FOLDER']):
|
|
Pierre-Yves Chibon |
2088eb |
flask.abort(404)
|
|
Pierre-Yves Chibon |
2088eb |
repo_obj = pygit2.Repository(
|
|
Pierre-Yves Chibon |
2088eb |
os.path.join(APP.config['GIT_FOLDER'], repo))
|
|
Pierre-Yves Chibon |
2088eb |
|
|
Pierre-Yves Chibon |
2088eb |
cnt = 0
|
|
Pierre-Yves Chibon |
2088eb |
last_commits = []
|
|
Pierre-Yves Chibon |
2088eb |
for commit in repo_obj.walk(repo_obj.head.target, pygit2.GIT_SORT_TIME):
|
|
Pierre-Yves Chibon |
2088eb |
last_commits.append(commit)
|
|
Pierre-Yves Chibon |
2088eb |
cnt += 1
|
|
Pierre-Yves Chibon |
2088eb |
if cnt == 10:
|
|
Pierre-Yves Chibon |
2088eb |
break
|
|
Pierre-Yves Chibon |
2088eb |
|
|
Pierre-Yves Chibon |
2088eb |
return flask.render_template(
|
|
Pierre-Yves Chibon |
2088eb |
'repo_info.html',
|
|
Pierre-Yves Chibon |
2088eb |
repo=repo,
|
|
Pierre-Yves Chibon |
2088eb |
branches=repo_obj.listall_branches(),
|
|
Pierre-Yves Chibon |
2088eb |
branchname='master',
|
|
Pierre-Yves Chibon |
2088eb |
last_commits=last_commits,
|
|
Pierre-Yves Chibon |
2088eb |
tree=sorted(last_commits[0].tree, key=lambda x: x.filemode),
|
|
Pierre-Yves Chibon |
2088eb |
)
|
|
Pierre-Yves Chibon |
2088eb |
|
|
Pierre-Yves Chibon |
2088eb |
|
|
Pierre-Yves Chibon |
2088eb |
@APP.route('/<repo>/branch/<branchname>')</branchname></repo>
|
|
Pierre-Yves Chibon |
2088eb |
def view_repo_branch(repo, branchname):
|
|
Pierre-Yves Chibon |
2088eb |
""" Displays the information about a specific branch.
|
|
Pierre-Yves Chibon |
2088eb |
"""
|
|
Pierre-Yves Chibon |
2088eb |
if repo not in os.listdir(APP.config['GIT_FOLDER']):
|
|
Pierre-Yves Chibon |
2088eb |
flask.abort(404)
|
|
Pierre-Yves Chibon |
2088eb |
repo_obj = pygit2.Repository(
|
|
Pierre-Yves Chibon |
2088eb |
os.path.join(APP.config['GIT_FOLDER'], repo))
|
|
Pierre-Yves Chibon |
2088eb |
|
|
Pierre-Yves Chibon |
2088eb |
if not branchname in repo_obj.listall_branches():
|
|
Pierre-Yves Chibon |
2088eb |
flask.abort(404)
|
|
Pierre-Yves Chibon |
2088eb |
|
|
Pierre-Yves Chibon |
2088eb |
branch = repo_obj.lookup_branch(branchname)
|
|
Pierre-Yves Chibon |
2088eb |
|
|
Pierre-Yves Chibon |
2088eb |
cnt = 0
|
|
Pierre-Yves Chibon |
2088eb |
last_commits = []
|
|
Pierre-Yves Chibon |
2088eb |
for commit in repo_obj.walk(branch.get_object().hex, pygit2.GIT_SORT_TIME):
|
|
Pierre-Yves Chibon |
2088eb |
last_commits.append(commit)
|
|
Pierre-Yves Chibon |
2088eb |
cnt += 1
|
|
Pierre-Yves Chibon |
2088eb |
if cnt == 10:
|
|
Pierre-Yves Chibon |
2088eb |
break
|
|
Pierre-Yves Chibon |
2088eb |
|
|
Pierre-Yves Chibon |
2088eb |
return flask.render_template(
|
|
Pierre-Yves Chibon |
2088eb |
'repo_info.html',
|
|
Pierre-Yves Chibon |
2088eb |
repo=repo,
|
|
Pierre-Yves Chibon |
2088eb |
branches=repo_obj.listall_branches(),
|
|
Pierre-Yves Chibon |
2088eb |
branchname=branchname,
|
|
Pierre-Yves Chibon |
2088eb |
last_commits=last_commits,
|
|
Pierre-Yves Chibon |
2088eb |
tree=sorted(last_commits[0].tree, key=lambda x: x.filemode),
|
|
Pierre-Yves Chibon |
2088eb |
)
|
|
Pierre-Yves Chibon |
2088eb |
|
|
Pierre-Yves Chibon |
2088eb |
@APP.route('/<repo>/log')</repo>
|
|
Pierre-Yves Chibon |
2088eb |
def view_log(repo):
|
|
Pierre-Yves Chibon |
2088eb |
""" Displays the logs of the specified repo.
|
|
Pierre-Yves Chibon |
2088eb |
"""
|
|
Pierre-Yves Chibon |
2088eb |
if repo not in os.listdir(APP.config['GIT_FOLDER']):
|
|
Pierre-Yves Chibon |
2088eb |
flask.abort(404)
|
|
Pierre-Yves Chibon |
2088eb |
repo_obj = pygit2.Repository(
|
|
Pierre-Yves Chibon |
2088eb |
os.path.join(APP.config['GIT_FOLDER'], repo))
|
|
Pierre-Yves Chibon |
2088eb |
|
|
Pierre-Yves Chibon |
2088eb |
try:
|
|
Pierre-Yves Chibon |
2088eb |
page = int(flask.request.args.get('page', 1))
|
|
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 |
end = limit * page
|
|
Pierre-Yves Chibon |
2088eb |
|
|
Pierre-Yves Chibon |
2088eb |
n_commits = 0
|
|
Pierre-Yves Chibon |
2088eb |
last_commits = []
|
|
Pierre-Yves Chibon |
2088eb |
for commit in repo_obj.walk(
|
|
Pierre-Yves Chibon |
2088eb |
repo_obj.head.target.hex, pygit2.GIT_SORT_TIME):
|
|
Pierre-Yves Chibon |
2088eb |
if n_commits >= start and n_commits <= end:
|
|
Pierre-Yves Chibon |
2088eb |
last_commits.append(commit)
|
|
Pierre-Yves Chibon |
2088eb |
n_commits += 1
|
|
Pierre-Yves Chibon |
2088eb |
|
|
Pierre-Yves Chibon |
2088eb |
total_page = int(ceil(n_commits / float(limit)))
|
|
Pierre-Yves Chibon |
2088eb |
|
|
Pierre-Yves Chibon |
2088eb |
return flask.render_template(
|
|
Pierre-Yves Chibon |
2088eb |
'repo_info.html',
|
|
Pierre-Yves Chibon |
2088eb |
origin='view_log',
|
|
Pierre-Yves Chibon |
2088eb |
repo=repo,
|
|
Pierre-Yves Chibon |
2088eb |
branches=repo_obj.listall_branches(),
|
|
Pierre-Yves Chibon |
2088eb |
branchname=branchname,
|
|
Pierre-Yves Chibon |
2088eb |
last_commits=last_commits,
|
|
Pierre-Yves Chibon |
2088eb |
page=page,
|
|
Pierre-Yves Chibon |
2088eb |
total_page=total_page,
|
|
Pierre-Yves Chibon |
2088eb |
)
|
|
Pierre-Yves Chibon |
2088eb |
|
|
Pierre-Yves Chibon |
2088eb |
|
|
Pierre-Yves Chibon |
2088eb |
@APP.route('/<repo>/blob/<identifier>/<path:filename>')</path:filename></identifier></repo>
|
|
Pierre-Yves Chibon |
2088eb |
@APP.route('/<repo>/blob/<identifier>/<path:filename>')</path:filename></identifier></repo>
|
|
Pierre-Yves Chibon |
2088eb |
def view_file(repo, identifier, filename):
|
|
Pierre-Yves Chibon |
2088eb |
""" Displays the content of a file or a tree for the specified repo.
|
|
Pierre-Yves Chibon |
2088eb |
"""
|
|
Pierre-Yves Chibon |
2088eb |
|
|
Pierre-Yves Chibon |
2088eb |
if repo not in os.listdir(APP.config['GIT_FOLDER']):
|
|
Pierre-Yves Chibon |
2088eb |
flask.abort(404, 'Git not found')
|
|
Pierre-Yves Chibon |
2088eb |
repo_obj = pygit2.Repository(
|
|
Pierre-Yves Chibon |
2088eb |
os.path.join(APP.config['GIT_FOLDER'], repo))
|
|
Pierre-Yves Chibon |
2088eb |
|
|
Pierre-Yves Chibon |
2088eb |
if identifier in repo_obj.listall_branches():
|
|
Pierre-Yves Chibon |
2088eb |
branchname = identifier
|
|
Pierre-Yves Chibon |
2088eb |
branch = repo_obj.lookup_branch(identifier)
|
|
Pierre-Yves Chibon |
2088eb |
commit = branch.get_object()
|
|
Pierre-Yves Chibon |
2088eb |
else:
|
|
Pierre-Yves Chibon |
2088eb |
branchname = 'master'
|
|
Pierre-Yves Chibon |
2088eb |
try:
|
|
Pierre-Yves Chibon |
2088eb |
commit = repo_obj.get(commitid)
|
|
Pierre-Yves Chibon |
2088eb |
except ValueError:
|
|
Pierre-Yves Chibon |
2088eb |
# If it's not a commit id then it's part of the filename
|
|
Pierre-Yves Chibon |
2088eb |
commit = repo_obj[repo_obj.head.target]
|
|
Pierre-Yves Chibon |
2088eb |
|
|
Pierre-Yves Chibon |
2088eb |
def __get_file_in_tree(tree, filepath):
|
|
Pierre-Yves Chibon |
2088eb |
''' Retrieve the entry corresponding to the provided filename in a
|
|
Pierre-Yves Chibon |
2088eb |
given tree.
|
|
Pierre-Yves Chibon |
2088eb |
'''
|
|
Pierre-Yves Chibon |
2088eb |
filename = filepath[0]
|
|
Pierre-Yves Chibon |
2088eb |
if isinstance(tree, pygit2.Blob):
|
|
Pierre-Yves Chibon |
2088eb |
return
|
|
Pierre-Yves Chibon |
2088eb |
for el in tree:
|
|
Pierre-Yves Chibon |
2088eb |
if el.name == filename:
|
|
Pierre-Yves Chibon |
2088eb |
if len(filepath) == 1:
|
|
Pierre-Yves Chibon |
2088eb |
return repo_obj[el.oid]
|
|
Pierre-Yves Chibon |
2088eb |
else:
|
|
Pierre-Yves Chibon |
2088eb |
return __get_file_in_tree(repo_obj[el.oid], filepath[1:])
|
|
Pierre-Yves Chibon |
2088eb |
|
|
Pierre-Yves Chibon |
2088eb |
content = __get_file_in_tree(commit.tree, filename.split('/'))
|
|
Pierre-Yves Chibon |
2088eb |
if not content:
|
|
Pierre-Yves Chibon |
2088eb |
flask.abort(404, 'File not found')
|
|
Pierre-Yves Chibon |
2088eb |
|
|
Pierre-Yves Chibon |
2088eb |
content = repo_obj[content.oid]
|
|
Pierre-Yves Chibon |
2088eb |
if isinstance(content, pygit2.Blob):
|
|
Pierre-Yves Chibon |
2088eb |
content = highlight(
|
|
Pierre-Yves Chibon |
2088eb |
content.data,
|
|
Pierre-Yves Chibon |
2088eb |
guess_lexer(content.data),
|
|
Pierre-Yves Chibon |
2088eb |
HtmlFormatter(
|
|
Pierre-Yves Chibon |
2088eb |
noclasses=True,
|
|
Pierre-Yves Chibon |
2088eb |
style="tango",)
|
|
Pierre-Yves Chibon |
2088eb |
)
|
|
Pierre-Yves Chibon |
2088eb |
output_type = 'file'
|
|
Pierre-Yves Chibon |
2088eb |
else:
|
|
Pierre-Yves Chibon |
2088eb |
content=sorted(content, key=lambda x: x.filemode)
|
|
Pierre-Yves Chibon |
2088eb |
output_type = 'tree'
|
|
Pierre-Yves Chibon |
2088eb |
|
|
Pierre-Yves Chibon |
2088eb |
return flask.render_template(
|
|
Pierre-Yves Chibon |
2088eb |
'file.html',
|
|
Pierre-Yves Chibon |
2088eb |
repo=repo,
|
|
Pierre-Yves Chibon |
2088eb |
branchname=branchname,
|
|
Pierre-Yves Chibon |
2088eb |
filename=filename,
|
|
Pierre-Yves Chibon |
2088eb |
content=content,
|
|
Pierre-Yves Chibon |
2088eb |
output_type=output_type,
|
|
Pierre-Yves Chibon |
2088eb |
)
|
|
Pierre-Yves Chibon |
2088eb |
|
|
Pierre-Yves Chibon |
2088eb |
@APP.route('/<repo>/<commitid>')</commitid></repo>
|
|
Pierre-Yves Chibon |
2088eb |
def view_commit(repo, commitid):
|
|
Pierre-Yves Chibon |
2088eb |
""" Render a commit in a repo
|
|
Pierre-Yves Chibon |
2088eb |
"""
|
|
Pierre-Yves Chibon |
2088eb |
if repo not in os.listdir(APP.config['GIT_FOLDER']):
|
|
Pierre-Yves Chibon |
2088eb |
flask.abort(404)
|
|
Pierre-Yves Chibon |
2088eb |
repo_obj = pygit2.Repository(
|
|
Pierre-Yves Chibon |
2088eb |
os.path.join(APP.config['GIT_FOLDER'], repo))
|
|
Pierre-Yves Chibon |
2088eb |
|
|
Pierre-Yves Chibon |
2088eb |
try:
|
|
Pierre-Yves Chibon |
2088eb |
commit = repo_obj.get(commitid)
|
|
Pierre-Yves Chibon |
2088eb |
except ValueError:
|
|
Pierre-Yves Chibon |
2088eb |
flask.abort(404)
|
|
Pierre-Yves Chibon |
2088eb |
parent = repo_obj.revparse_single('%s^' % commitid)
|
|
Pierre-Yves Chibon |
2088eb |
diff = repo_obj.diff(parent, commit)
|
|
Pierre-Yves Chibon |
2088eb |
html_diff = highlight(
|
|
Pierre-Yves Chibon |
2088eb |
diff.patch,
|
|
Pierre-Yves Chibon |
2088eb |
DiffLexer(),
|
|
Pierre-Yves Chibon |
2088eb |
HtmlFormatter(
|
|
Pierre-Yves Chibon |
2088eb |
noclasses=True,
|
|
Pierre-Yves Chibon |
2088eb |
style="tango",)
|
|
Pierre-Yves Chibon |
2088eb |
)
|
|
Pierre-Yves Chibon |
2088eb |
|
|
Pierre-Yves Chibon |
2088eb |
return flask.render_template(
|
|
Pierre-Yves Chibon |
2088eb |
'commit.html',
|
|
Pierre-Yves Chibon |
2088eb |
repo=repo,
|
|
Pierre-Yves Chibon |
2088eb |
commitid=commitid,
|
|
Pierre-Yves Chibon |
2088eb |
commit=commit,
|
|
Pierre-Yves Chibon |
2088eb |
diff=diff,
|
|
Pierre-Yves Chibon |
2088eb |
html_diff=html_diff,
|
|
Pierre-Yves Chibon |
2088eb |
)
|
|
Pierre-Yves Chibon |
12e862 |
|
|
Pierre-Yves Chibon |
12e862 |
|
|
Pierre-Yves Chibon |
12e862 |
@APP.route('/<repo>/tree/')</repo>
|
|
Pierre-Yves Chibon |
12e862 |
@APP.route('/<repo>/tree/<identifier>')</identifier></repo>
|
|
Pierre-Yves Chibon |
12e862 |
def view_tree(repo, identifier=None):
|
|
Pierre-Yves Chibon |
12e862 |
""" Render the tree of the repo
|
|
Pierre-Yves Chibon |
12e862 |
"""
|
|
Pierre-Yves Chibon |
12e862 |
if repo not in os.listdir(APP.config['GIT_FOLDER']):
|
|
Pierre-Yves Chibon |
12e862 |
flask.abort(404)
|
|
Pierre-Yves Chibon |
12e862 |
repo_obj = pygit2.Repository(
|
|
Pierre-Yves Chibon |
12e862 |
os.path.join(APP.config['GIT_FOLDER'], repo))
|
|
Pierre-Yves Chibon |
12e862 |
|
|
Pierre-Yves Chibon |
12e862 |
if identifier in repo_obj.listall_branches():
|
|
Pierre-Yves Chibon |
12e862 |
branchname = identifier
|
|
Pierre-Yves Chibon |
12e862 |
branch = repo_obj.lookup_branch(identifier)
|
|
Pierre-Yves Chibon |
12e862 |
commit = branch.get_object()
|
|
Pierre-Yves Chibon |
12e862 |
else:
|
|
Pierre-Yves Chibon |
12e862 |
try:
|
|
Pierre-Yves Chibon |
12e862 |
commit = repo_obj.get(identifier)
|
|
Pierre-Yves Chibon |
12e862 |
branchname = identifier
|
|
Pierre-Yves Chibon |
12e862 |
except (ValueError, TypeError):
|
|
Pierre-Yves Chibon |
12e862 |
# If it's not a commit id then it's part of the filename
|
|
Pierre-Yves Chibon |
12e862 |
commit = repo_obj[repo_obj.head.target]
|
|
Pierre-Yves Chibon |
12e862 |
branchname = repo_obj.head.target
|
|
Pierre-Yves Chibon |
12e862 |
|
|
Pierre-Yves Chibon |
12e862 |
content=sorted(commit.tree, key=lambda x: x.filemode)
|
|
Pierre-Yves Chibon |
12e862 |
output_type = 'tree'
|
|
Pierre-Yves Chibon |
12e862 |
|
|
Pierre-Yves Chibon |
12e862 |
return flask.render_template(
|
|
Pierre-Yves Chibon |
12e862 |
'file.html',
|
|
Pierre-Yves Chibon |
12e862 |
repo=repo,
|
|
Pierre-Yves Chibon |
12e862 |
branchname=branchname,
|
|
Pierre-Yves Chibon |
12e862 |
filename='',
|
|
Pierre-Yves Chibon |
12e862 |
content=content,
|
|
Pierre-Yves Chibon |
12e862 |
output_type=output_type,
|
|
Pierre-Yves Chibon |
12e862 |
)
|