-
Notifications
You must be signed in to change notification settings - Fork 4
/
server.py
106 lines (84 loc) · 2.72 KB
/
server.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
import logging
from flask import Flask, g, abort, render_template
from werkzeug.utils import secure_filename
from mypi import db as model
LOG = logging.getLogger(__name__)
app = Flask(__name__)
@app.before_request
def before_request():
g.db = model.Session()
@app.teardown_request
def teardown_request(exception):
g.db.close()
@app.route("/", methods=['GET'])
@app.route("/package/")
def index():
packages = model.Package.all(g.db)
return render_template("package_list.html", packages=packages)
@app.route("/", methods=['POST'])
def post():
from flask import request
frm = request.form
#TODO: raise error if :action not in frm
action_name = "_do_%s" % frm[":action"]
action = globals().get(action_name, None)
if action:
return action(frm)
return abort(501, description="Action %s is not yet implemented" % frm[":action"])
@app.route("/<name>/")
@app.route("/package/<name>/")
def package(name):
"""
Display the Package details
"""
proj = model.Package.get(g.db, name)
if not proj:
return abort(404, description = "No such package")
return render_template("package.html", package=proj)
@app.route("/download/<package>/<filename>/")
def download(package, filename):
file = model.File.find_by_filename(g.db, package, filename)
if not file:
return abort(404, description="File not found")
from werkzeug.wrappers import Response
resp = Response(file.data)
resp.headers['Content-Type'] = 'application/x-gzip'
resp.headers['Content-Disposition'] = 'attachement; filename="%s"' % filename
return resp
@app.route("/simple/")
def simple():
"""
List all available packages
"""
packages = model.Package.all(g.db)
return render_template("simple/packages.html", packages=packages)
@app.route("/simple/<package>")
@app.route("/simple/<package>/")
def simple_package(package):
"""
List all available package releases
"""
package = model.Package.get(g.db, package)
if not package:
abort(404, description = "No such package")
return render_template("simple/releases.html", package=package)
def _do_file_upload(data):
from flask import request
file = request.files['content']
filename = secure_filename(file.filename)
model.File.upload(g.db, data, filename, file.stream)
try:
g.db.commit()
except model.IntegrityError, exc:
LOG.error(exc)
abort(409, description="This file exists already!")
return "OK"
def _do_submit(data):
try:
rel = model.Release.register(g.db, data)
g.db.commit()
return "added release %s" % rel
except ValueError, ex:
return abort(409, description=str(ex))
if __name__ == "__main__":
app.run()