starting refactor and extending app

suchwow/app.py

@@ -3,13 +3,12 @@ import uuid
 import os
 import json
 import requests
-from flask import Flask, g, request, redirect, url_for, abort
+from flask import Flask, request, redirect, url_for, abort
 from flask import jsonify, render_template, flash, session
 from flask import send_from_directory, make_response
 from flask_session import Session
-from playhouse.flask_utils import PaginatedQuery
 from werkzeug.utils import secure_filename
-from suchwow.models import Meme, db
+from suchwow.models import Post, Profile, Comment, Notification, db
 
 
 app = Flask(__name__)
@@ -17,9 +16,9 @@ app.config.from_envvar("FLASK_SECRETS")
 app.secret_key = app.config["SECRET_KEY"]
 Session(app)
 
-OPENID_URL = app.config["OIDC_URL"][0]
-OPENID_CLIENT_ID = app.config["OIDC_CLIENT_ID"][0]
-OPENID_CLIENT_SECRET = app.config["OIDC_CLIENT_SECRET"][0]
+OPENID_URL = app.config["OIDC_URL"]
+OPENID_CLIENT_ID = app.config["OIDC_CLIENT_ID"]
+OPENID_CLIENT_SECRET = app.config["OIDC_CLIENT_SECRET"]
 OPENID_REDIRECT_URI = "http://localhost:5000/auth"
 
 
@@ -27,9 +26,22 @@ def allowed_file(filename):
     return "." in filename and \
            filename.rsplit(".", 1)[1].lower() in app.config["ALLOWED_EXTENSIONS"]
 
+def debug_login():
+    session["auth"] = {
+        "state": "active",
+        "session_state": "debug",
+        "code": "abcdefg",
+        "preferred_username": "debuguser",
+        "debug": True
+    }
+    return
+
 def login_required(f):
     @wraps(f)
     def decorated_function(*args, **kwargs):
+        if app.debug:
+            debug_login()
+            return f(*args, **kwargs)
         if "auth" not in session or not session["auth"]:
             return redirect(url_for("login"))
         return f(*args, **kwargs)
@@ -38,31 +50,31 @@ def login_required(f):
 
 @app.route("/")
 def index():
-    page = request.args.get('page')
+    page = request.args.get("page", "1")
     if page.isdigit() is False:
         flash("Wow, wtf hackerman. Cool it.")
         page = 1
 
-    memes = Meme.select().order_by(Meme.timestamp).paginate(int(page), 10)
-    return render_template("index.html", memes=memes, page=page)
+    posts = Post.select().order_by(Post.timestamp).paginate(int(page), 10)
+    return render_template("index.html", posts=posts, page=page)
 
-@app.route("/meme/<id>")
-def meme(id):
-    if Meme.filter(id=id):
-        meme = Meme.get(Meme.id == id)
-        return render_template("meme.html", meme=meme)
+@app.route("/post/<id>")
+def read_post(id):
+    if Post.filter(id=id):
+        post = Post.get(Post.id == id)
+        return render_template("post.html", post=post)
     else:
         return "no meme there brah"
 
 @app.route("/uploads/<path:filename>")
 def uploaded_file(filename):
-    return send_from_directory(app.config['UPLOAD_FOLDER'], filename)
+    return send_from_directory(app.config["UPLOAD_FOLDER"], filename)
 
-@app.route("/submit", methods=["GET", "POST"])
+@app.route("/post/create", methods=["GET", "POST"])
 @login_required
-def submit():
+def create_post():
     if request.method == "POST":
-        meme_title = request.form.get('title')
+        post_title = request.form.get("title")
         # check if the post request has the file part
         if "file" not in request.files:
             flash("You didn't upload a caliente meme, bro! You're fuckin up!")
@@ -73,21 +85,25 @@ def submit():
         if file.filename == "":
             flash("You didn't upload a caliente meme, bro! You're fuckin up!")
             return redirect(request.url)
-        if meme_title is "":
+        if post_title is "":
             flash("You didn't give your meme a spicy title, bro! You're fuckin up!")
             return redirect(request.url)
         if file and allowed_file(file.filename):
             filename = secure_filename(file.filename)
             save_path = os.path.join(app.config["UPLOAD_FOLDER"], filename)
             file.save(save_path)
-            meme = Meme(
-                title=meme_title,
+            # gen wallet
+            post = Post(
+                title=post_title,
+                subtitle=request.form.get("subtitle", ""),
                 submitter=session["auth"]["preferred_username"],
                 image_name=filename,
+                account_index=0,
+                address_index=0
             )
-            meme.save()
-            return redirect(url_for("meme", id=meme.id))
-    return render_template("submit.html")
+            post.save()
+            return redirect(url_for("read_post", id=post.id))
+    return render_template("create_post.html")
 
 @app.route("/login")
 def login():
@@ -99,7 +115,11 @@ def login():
           f"response_type=code&" \
           f"state={state}"
 
-    return redirect(url)
+    if app.debug:
+        debug_login()
+        return redirect(url_for("index"))
+    else:
+        return redirect(url)
 
 
 @app.route("/auth/")
@@ -169,9 +189,9 @@ def logout():
 def not_found(error):
     return make_response(jsonify({"error": "Page not found"}), 404)
 
-@app.cli.command('dbinit')
+@app.cli.command("dbinit")
 def dbinit():
-    db.create_tables([Meme])
+    db.create_tables([Post, Profile, Comment, Notification])
 
 if __name__ == "__main__":
     app.run()

suchwow/config.example.py

@@ -5,5 +5,5 @@ OIDC_REDIRECT_URL = 'http://localhost:5000/auth'
 SECRET = 'yyyyyyyyyyyyy',
 SESSION_TYPE = 'filesystem'
 UPLOAD_FOLDER = '/path/to/the/uploads'
-ALLOWED_EXTENSIONS = set(['png', 'jpg', 'jpeg', 'gif'])
-MAX_CONTENT_LENGTH = 16 * 1024 * 1024
+ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif'}
+MAX_CONTENT_LENGTH = 16 * 1024 * 1024

suchwow/models.py

@@ -4,11 +4,45 @@ from datetime import datetime
 
 db = SqliteDatabase('data/sqlite.db')
 
-class Meme(Model):
+class Post(Model):
     id = AutoField()
     title = CharField()
+    subtitle = CharField()
+    # submitter = ForeignKeyField(Profile, field=Profile.username)
     submitter = CharField()
     image_name = CharField()
+    readonly = BooleanField(default=False)
+    hidden = BooleanField(default=False)
+    account_index = IntegerField()
+    address_index = IntegerField()
+    timestamp = DateTimeField(default=datetime.now)
+
+    class Meta:
+        database = db
+
+class Profile(Model):
+    id = AutoField()
+    username = CharField()
+    address = CharField()
+    notifications = IntegerField()
+
+    class Meta:
+        database = db
+
+class Comment(Model):
+    id = AutoField()
+    comment = TextField()
+    commenter = ForeignKeyField(Profile, field=Profile.username)
+    post = ForeignKeyField(Post, field=id)
+    timestamp = DateTimeField(default=datetime.now)
+
+    class Meta:
+        database = db
+
+class Notification(Model):
+    type = CharField()
+    message = TextField()
+    username = ForeignKeyField(Profile, field=Profile.username)
     timestamp = DateTimeField(default=datetime.now)
 
     class Meta:

suchwow/templates/create_post.html

@@ -5,6 +5,7 @@
 <h1>Upload new File</h1>
 <form method=post enctype=multipart/form-data>
   <input type="text" name="title"><br>
+  <input type="text" name="subtitle"><br>
   <input type=file name=file>
   <input type=submit value=Submit>
 </form>

suchwow/templates/index.html

@@ -2,7 +2,7 @@
 
  {% block content %}
 
- <a href="{{ url_for('submit') }}">Submit A Meme</a><br>
+ <a href="{{ url_for('create_post') }}">Submit A Meme</a><br>
  <a href="{{ url_for('debug') }}">Debug</a>
 
  {% block header %}
@@ -10,10 +10,11 @@
  {% endblock %}
 
 <ul>
- {% for meme in memes %}
-   <li><a href="{{ url_for('meme', id=meme.id) }}">{{ meme.title }}</a></li>
+ {% for post in posts %}
+   <li><a href="{{ url_for('read_post', id=post.id) }}">{{ post.title }}</a></li>
  {% endfor %}
 </ul>
 
 {{ page }}
- {% endblock %}
+
+{% endblock %}

suchwow/templates/meme.html

@@ -1,14 +0,0 @@
-{% extends 'base.html' %}
-
-{% block header %}
-  <h1>{% block title %}View Meme{% endblock %}</h1>
-{% endblock %}
-
-{% block content %}
-<p>Submitter: {{ session['auth']['preferred_username'] }}</p>
-<p>ID: {{ meme.id }}</p>
-<p>Title: {{ meme.title }}</p>
-<p>Submitted: {{ meme.timestamp }}</p>
-<p>Image Name: {{ meme.image_name }}</p>
-<img src="{{ url_for('uploaded_file', filename=meme.image_name) }}" width=400/>
-{% endblock %}

suchwow/templates/post.html

@@ -0,0 +1,19 @@
+{% extends 'base.html' %}
+
+{% block header %}
+  <h1>{% block title %}View Post{% endblock %}</h1>
+{% endblock %}
+
+{% block content %}
+    {% if post.hidden %}
+You cannot see this post
+    {% else %}
+<p>Submitter: {{ post.submitter }}</p>
+<p>ID: {{ post.id }}</p>
+<p>Title: {{ post.title }}</p>
+<p>Subtitle: {{ post.subtitle }}</p>
+<p>Submitted: {{ post.timestamp }}</p>
+<p>Image Name: {{ post.image_name }}</p>
+<img src="{{ url_for('uploaded_file', filename=post.image_name) }}" width=500/>
+    {% endif %}
+{% endblock %}