From 8bd9d1519eee62660c395333a764a5d7443b85e2 Mon Sep 17 00:00:00 2001
From: "Jason A. Donenfeld" <Jason@zx2c4.com>
Date: Fri, 6 May 2011 22:48:09 -0400
Subject: LIl things.

---
 scanner/PhotoAlbum.py | 20 ++++++++++++++++++--
 1 file changed, 18 insertions(+), 2 deletions(-)

(limited to 'scanner/PhotoAlbum.py')

diff --git a/scanner/PhotoAlbum.py b/scanner/PhotoAlbum.py
index 18cd425..50216c9 100644
--- a/scanner/PhotoAlbum.py
+++ b/scanner/PhotoAlbum.py
@@ -52,6 +52,17 @@ class Album(object):
 		if not self._albums_sorted:
 			self._albums.sort()
 			self._albums_sorted = True
+	@property
+	def empty(self):
+		if len(self._photos) != 0:
+			return False
+		if len(self._albums) == 0:
+			return True
+		for album in self._albums:
+			if not album.empty:
+				return False
+		return True
+		
 	def cache(self, base_dir):
 		self._sort()
 		fp = open(os.path.join(base_dir, self.cache_path), 'w')
@@ -75,10 +86,15 @@ class Album(object):
 		return album
 	def to_dict(self, cripple=True):
 		self._sort()
+		subalbums = []
 		if cripple:
-			subalbums = [ { "path": trim_base_custom(sub.path, self._path), "date": sub.date } for sub in self._albums ]
+			for sub in self._albums:
+				if not sub.empty:
+					subalbums.append({ "path": trim_base_custom(sub.path, self._path), "date": sub.date })
 		else:
-			subalbums = self._albums
+			for sub in self._albums:
+				if not sub.empty:
+					subalbums.append(sub)
 		return { "path": self.path, "date": self.date, "albums": subalbums, "photos": self._photos }
 	def photo_from_path(self, path):
 		for photo in self._photos:
-- 
cgit v1.2.3-59-g8ed1b