From 1d3dc51bcaee8532aa51a26d6ba22d908aa32b98 Mon Sep 17 00:00:00 2001
From: Noah Petherbridge <root@kirsle.net>
Date: Tue, 19 Apr 2016 22:31:55 +0000
Subject: [PATCH] Bugfixes with blog and photo albums

---
 rophako/model/blog.py             | 6 ++++--
 rophako/modules/photo/__init__.py | 3 +++
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/rophako/model/blog.py b/rophako/model/blog.py
index b69456f..f02338e 100644
--- a/rophako/model/blog.py
+++ b/rophako/model/blog.py
@@ -45,12 +45,14 @@ def get_index():
     db = JsonDB.get("blog/index")
 
     # Hide any private posts if we aren't logged in.
+    new_db = dict()
     if not g.info["session"]["login"]:
         for post_id, data in db.items():
             if data["privacy"] == "private":
-                del db[post_id]
+                continue
+            new_db[post_id] = db[post_id]
 
-    return db
+    return new_db
 
 
 def rebuild_index():
diff --git a/rophako/modules/photo/__init__.py b/rophako/modules/photo/__init__.py
index a387fb0..eb97d51 100644
--- a/rophako/modules/photo/__init__.py
+++ b/rophako/modules/photo/__init__.py
@@ -43,6 +43,9 @@ def album_index(name):
 
     g.info["album"]      = name
     g.info["album_info"] = Photo.get_album(name)
+    if not g.info["album_info"]:
+        flash("That photo album wasn't found!")
+        return redirect(url_for(".albums"))
     g.info["markdown"]   = render_markdown(g.info["album_info"]["description"])
     g.info["photos"]     = photos