contentdb/app/views/packages.py

88 lines
2.8 KiB
Python
Raw Normal View History

2018-03-19 19:08:41 +01:00
from flask import *
from flask_user import *
from flask.ext import menu
from app import app
from app.models import *
2018-03-20 03:58:31 +01:00
from flask_wtf import FlaskForm
from wtforms import *
2018-03-20 02:19:19 +01:00
# TODO: the following could be made into one route, except I'm not sure how
# to do the menu
2018-03-19 19:08:41 +01:00
@app.route('/mods/')
2018-03-20 02:19:19 +01:00
@menu.register_menu(app, '.mods', 'Mods', order=10)
2018-03-19 19:08:41 +01:00
def mods_page():
2018-03-20 01:44:47 +01:00
packages = Package.query.filter_by(type=PackageType.MOD).all()
2018-03-19 19:08:41 +01:00
return render_template('packages.html', title="Mods", packages=packages)
2018-03-20 01:44:47 +01:00
@app.route('/games/')
2018-03-20 02:19:19 +01:00
@menu.register_menu(app, '.games', 'Games', order=11)
2018-03-20 01:44:47 +01:00
def games_page():
packages = Package.query.filter_by(type=PackageType.GAME).all()
return render_template('packages.html', title="Games", packages=packages)
@app.route('/texturepacks/')
2018-03-20 02:19:19 +01:00
@menu.register_menu(app, '.txp', 'Texture Packs', order=12)
2018-03-20 01:44:47 +01:00
def txp_page():
packages = Package.query.filter_by(type=PackageType.TXP).all()
return render_template('packages.html', title="Texture Packs", packages=packages)
2018-03-20 02:19:19 +01:00
2018-03-20 03:58:31 +01:00
def getPageByInfo(type, author, name):
user = User.query.filter_by(username=author).first()
if user is None:
abort(404)
package = Package.query.filter_by(name=name, author_id=user.id,
2018-03-20 05:11:50 +01:00
type=PackageType[type.upper()]).first()
2018-03-19 19:08:41 +01:00
if package is None:
abort(404)
2018-03-20 03:58:31 +01:00
return package
2018-03-20 19:20:30 +01:00
def getReleases(package):
if package.checkPerm(current_user, Permission.MAKE_RELEASE):
return package.releases
else:
return [rel for rel in package.releases if rel.approved]
2018-03-20 03:58:31 +01:00
@app.route("/<type>s/<author>/<name>/")
def package_page(type, author, name):
package = getPageByInfo(type, author, name)
2018-03-20 19:20:30 +01:00
releases = getReleases(package)
return render_template('package_details.html', package=package, releases=releases)
2018-03-20 03:58:31 +01:00
class PackageForm(FlaskForm):
name = StringField("Name")
title = StringField("Title")
shortDesc = StringField("Short Description")
desc = StringField("Long Description")
type = SelectField("Type", choices=PackageType.choices(), coerce=PackageType.coerce)
repo = StringField("Repo URL")
website = StringField("Website URL")
issueTracker = StringField("Issue Tracker URL")
forums = StringField("Forum Topic ID")
submit = SubmitField('Save')
@app.route("/<type>s/<author>/<name>/edit/", methods=['GET', 'POST'])
@login_required
def edit_package_page(type, author, name):
package = getPageByInfo(type, author, name)
if not package.checkPerm(current_user, Permission.EDIT_PACKAGE):
return redirect(package.getDetailsURL())
# Initial form class from post data and default data
form = PackageForm(formdata=request.form, obj=package)
if request.method == "POST" and form.validate():
# Successfully submitted!
form.populate_obj(package) # copy to row
db.session.commit() # save
return redirect(package.getDetailsURL()) # redirect
return render_template('package_edit.html', package=package, form=form)