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/TreeWalker.py | 33 ++++++++++++++++-----------------
 1 file changed, 16 insertions(+), 17 deletions(-)

(limited to 'scanner/TreeWalker.py')

diff --git a/scanner/TreeWalker.py b/scanner/TreeWalker.py
index 96ca5e5..872b830 100644
--- a/scanner/TreeWalker.py
+++ b/scanner/TreeWalker.py
@@ -49,9 +49,12 @@ class TreeWalker:
 				if photo.is_valid:
 					self.all_photos.append(photo)
 					album.add_photo(photo)
-		print "Writing cache of %s" % album.cache_path
-		album.cache(self.cache_path)
-		self.all_albums.append(album)
+		if not album.empty:
+			print "Writing cache of %s" % album.cache_path
+			album.cache(self.cache_path)
+			self.all_albums.append(album)
+		else:
+			print "Not writing cache of %s because it's empty" % album.cache_path
 		return album
 	def big_lists(self):
 		photo_list = []
@@ -69,19 +72,15 @@ class TreeWalker:
 		fp.close()
 		
 	def remove_stale(self):
-		print "Removing stale cache entries."
+		print "Building list of all cache entries."
+		all_cache_entries = { "all_photos.json": True, "latest_photos.json": True }
+		for album in self.all_albums:
+			all_cache_entries[album.cache_path] = True
+		for photo in self.all_photos:
+			for entry in photo.image_caches:
+				all_cache_entries[entry] = True
+		print "Searching stale cache entries."
 		for cache in os.listdir(self.cache_path):
-			match = False
-			for album in self.all_albums:
-				if cache == album.cache_path:
-					match = True
-					break
-			if match:
-				continue
-			for photo in self.all_photos:
-				if cache in photo.image_caches:
-					match = True
-					break
-			if not match:
+			if cache not in all_cache_entries:
 				print "Removing stale cache %s" % cache
-				os.unlink(os.path.join(self.cache_path, cache))
\ No newline at end of file
+				os.unlink(os.path.join(self.cache_path, cache))
-- 
cgit v1.2.3-59-g8ed1b