Merge branch 'profile-pics'

Tweeder.py

@@ -1,5 +1,5 @@
-from flask import Flask, render_template, request, redirect, url_for, session
-from backend import accounts, timeline
+from flask import Flask, render_template, request, redirect, url_for, session, make_response, abort
+from backend import accounts, timeline, files
 
 app = Flask(__name__)
 app.secret_key = "eVZ4EmVK70iETb03KqDAXV5sBHb3T73t"
@@ -127,16 +127,26 @@ def user_settings():
         else:
             return redirect(url_for('login'))
     elif request.method == "POST":
+        print(request.files)
+        print(request.form)
         if 'username' not in session.keys():
             return redirect(url_for('login'))
-        profile = {
+        updated_profile = {
             'bio': request.form['bio'],
             'gender': request.form['gender'],
             'location': request.form['location']
         }
+        if 'profile_pic' in request.files.keys():
+            if request.files['profile_pic'].filename == '':
+                if accounts.account_details(session['username'].lower())['profile']['profile_pic']:
+                    profile_pic = accounts.account_details(session['username'].lower())['profile']['profile_pic']
+                    updated_profile['profile_pic'] = profile_pic
+            else:
+                profile_pic = files.upload_file(request.files['profile_pic'])
+                updated_profile['profile_pic'] = profile_pic
         accounts.set_theme(session['username'].lower(), request.form['theme'])
         username = session['username']
-        accounts.update_profile(username, profile)
+        accounts.update_profile(username, updated_profile)
         return redirect(request.referrer)
 
 
@@ -216,5 +226,15 @@ def unlike_post(post_id):
     elif request.method == "GET":
         pass
 
+
+@app.route("/files/<oid>", methods=['GET'])
+def get_file(oid):
+    fl = files.get_file(oid)
+    if not fl: return abort(404)
+    response = make_response(fl.read())
+    response.mimetype = fl.content_type
+    return response
+
+
 if __name__ == '__main__':
     app.run(host="127.0.0.1", debug=True)

backend/files.py

@@ -0,0 +1,25 @@
+from pymongo import MongoClient
+from bson.objectid import ObjectId
+from gridfs.errors import NoFile
+from werkzeug.utils import secure_filename
+import gridfs
+
+client = MongoClient()
+db = client.tweeder
+files_db = gridfs.GridFS(client.tweeder_files)
+accounts_db = db.accounts
+
+
+def get_file(oid):
+    try:
+        return files_db.get(ObjectId(oid))
+    except NoFile:
+        return False
+
+
+def upload_file(file_to_upload):
+    if file_to_upload.filename == '':
+        return False
+    filename = secure_filename(file_to_upload.filename)
+    obj = files_db.put(file_to_upload, content_type=file_to_upload.content_type, filename=filename)
+    return obj

templates/profile.html

@@ -3,6 +3,9 @@
 {% block content %}
     <div class="container">
         <div class="jumbotron">
+            {% if user.profile.profile_pic %}
+                <img src="/files/{{ user.profile.profile_pic }}" class="rounded float-left" style="margin-right: 24px; max-height: 100px;"/>
+            {% endif %}
             <h1>{% if user.verified %} <i class="fas fa-shield-check" alt="Verified"></i> {% endif %}{{ user.displayname }} <small>on Tweeder</small></h1>
             <h2>
                 {% if user.profile.bio %}
@@ -11,6 +14,7 @@
                 Tweeder user since 1984
                 {% endif %}
             </h2>
+            <div style="clear:both"></div>
         </div>
 
         <div class="row">

templates/settings.html

@@ -3,8 +3,10 @@
 {% block content %}
     <div class="container">
         <h1>Settings for {{ logged_in }}</h1><hr />
-        <form method="POST" action="/settings" class="col-lg-6">
+        <form method="POST" action="/settings" class="col-lg-6" enctype="multipart/form-data">
             <div class="form-group row">
+                <h2>Profile Picture</h2>
+                    <input type="file" name="profile_pic" size="40">
                 <h2>Profile Fields</h2>
 
                 <div class="input-group mb-3">