diff --git a/app/models.py b/app/models.py index cfe19315..f71725d8 100644 --- a/app/models.py +++ b/app/models.py @@ -58,6 +58,12 @@ class UserRoles(db.Model): user_id = db.Column(db.Integer(), db.ForeignKey('user.id', ondelete='CASCADE')) role_id = db.Column(db.Integer(), db.ForeignKey('role.id', ondelete='CASCADE')) +class Permission(enum.Enum): + EDIT_PACKAGE = "EDIT_PACKAGE" + APPROVE = "APPROVE" + DELETE_PACKAGE = "DELETE_PACKAGE" + CHANGE_AUTHOR = "CHANGE_AUTHOR" + class PackageType(enum.Enum): MOD = "Mod" GAME = "Game" @@ -82,6 +88,23 @@ class PackageType(enum.Enum): else: return None + @classmethod + def choices(cls): + return [(choice, choice.value) for choice in cls] + + @classmethod + def coerce(cls, item): + """item will be both type(enum) AND type(unicode). + """ + if item == 'PackageType.MOD' or item == PackageType.MOD: + return PackageType.MOD + elif item == 'PackageType.GAME' or item == PackageType.GAME: + return PackageType.GAME + elif item == 'PackageType.TXP' or item == PackageType.TXP: + return PackageType.TXP + else: + print("Can't coerce", item, type(item)) + class Package(db.Model): id = db.Column(db.Integer, primary_key=True) @@ -99,6 +122,19 @@ class Package(db.Model): issueTracker = db.Column(db.String(200), nullable=True) forums = db.Column(db.String(200), nullable=False) + def getDetailsURL(self): + return url_for("package_page", + type=self.type.getTitle().lower(), + author=self.author.username, name=self.name) + + def getEditURL(self): + return url_for("edit_package_page", + type=self.type.getTitle().lower(), + author=self.author.username, name=self.name) + + def checkPerm(self, user, perm): + return user == self.author + # Setup Flask-User db_adapter = SQLAlchemyAdapter(db, User) # Register the User model user_manager = UserManager(db_adapter, app) # Initialize Flask-User diff --git a/app/templates/package_details.html b/app/templates/package_details.html index b75be243..457673aa 100644 --- a/app/templates/package_details.html +++ b/app/templates/package_details.html @@ -20,6 +20,7 @@
+{% endblock %} diff --git a/app/views/packages.py b/app/views/packages.py index 5f3f0fe1..e525caac 100644 --- a/app/views/packages.py +++ b/app/views/packages.py @@ -4,6 +4,9 @@ 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 @@ -27,9 +30,7 @@ def txp_page(): return render_template('packages.html', title="Texture Packs", packages=packages) - -@app.route("/