diff --git a/drama/routes/posts.py b/drama/routes/posts.py index ba37c665b..fcb891a25 100644 --- a/drama/routes/posts.py +++ b/drama/routes/posts.py @@ -1087,9 +1087,10 @@ def delete_post_pid(pid, v): file.write(chunk) image = upload_from_file("resizing", "resizing", (100, 100)) - u.profileurl = image - u.resized = True - g.db.add(u) + if image != None: + u.profileurl = image + u.resized = True + g.db.add(u) print(f"2 {u.profileurl}") return "", 204 diff --git a/drama/routes/users.py b/drama/routes/users.py index 14f0def8c..62fa8f9d6 100644 --- a/drama/routes/users.py +++ b/drama/routes/users.py @@ -41,9 +41,10 @@ def leaderboard(v): file.write(chunk) image = upload_from_file("resizing", "resizing", (100, 100)) - u.profileurl = image - u.resized = True - g.db.add(u) + if image != None: + u.profileurl = image + u.resized = True + g.db.add(u) print(f"2 {u.profileurl}") return render_template("leaderboard.html", v=v, users1=users1, users2=users2)