diff --git a/instance/Picture_Puzzle_web.db b/instance/Picture_Puzzle_web.db new file mode 100644 index 0000000..0ac512f Binary files /dev/null and b/instance/Picture_Puzzle_web.db differ diff --git a/main.py b/main.py index 2884185..5b7fa3b 100644 --- a/main.py +++ b/main.py @@ -1,21 +1,34 @@ from flask import Flask, render_template, redirect, request, session, url_for from flask_sqlalchemy import SQLAlchemy +from flask_admin import Admin +from flask_admin.contrib.sqla import ModelView app = Flask(__name__) +app.secret_key = 'bebra' app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///Picture_Puzzle_web.db' db = SQLAlchemy(app) + class User(db.Model): id = db.Column(db.Integer, primary_key=True) username = db.Column(db.String(100), unique=True, nullable=False) + password = db.Column(db.String(100), nullable=False) email = db.Column(db.String(100), nullable=False) class Post(db.Model): id = db.Column(db.Integer, primary_key=True) + date_created = db.Column(db.String(100), unique=True, nullable=False) alias = db.Column(db.String(100), unique=True, nullable=False) title = db.Column(db.String(100), nullable=False) image = db.Column(db.String(100), nullable=False) - -db.create_all() + +def create_tables(): + with app.app_context(): + db.create_all() + +def register_admin_views(): + admin = Admin(app, name='Admin Panel', template_mode='bootstrap3') + admin.add_view(ModelView(User, db.session)) + admin.add_view(ModelView(Post, db.session)) users = { 'thegamer': 'gamerpro1', @@ -83,5 +96,7 @@ def register(): return redirect(url_for("login")) return render_template("auth/register.html") -if __name__ == "__main__": +if __name__ == '__main__': + create_tables() + register_admin_views() app.run(debug=True) \ No newline at end of file