from flask import * from flask_user import * from flask.ext import menu from app import app from app.models import * from flask_wtf import FlaskForm from wtforms import * # TODO: the following could be made into one route, except I'm not sure how # to do the menu @app.route('/mods/') @menu.register_menu(app, '.mods', 'Mods', order=10) def mods_page(): packages = Package.query.filter_by(type=PackageType.MOD).all() return render_template('packages.html', title="Mods", packages=packages) @app.route('/games/') @menu.register_menu(app, '.games', 'Games', order=11) def games_page(): packages = Package.query.filter_by(type=PackageType.GAME).all() return render_template('packages.html', title="Games", packages=packages) @app.route('/texturepacks/') @menu.register_menu(app, '.txp', 'Texture Packs', order=12) def txp_page(): packages = Package.query.filter_by(type=PackageType.TXP).all() return render_template('packages.html', title="Texture Packs", packages=packages) 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, type=PackageType.fromName(type)).first() if package is None: abort(404) return package @app.route("/s///") def package_page(type, author, name): package = getPageByInfo(type, author, name) return render_template('package_details.html', package=package) 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("/s///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)