aboutsummaryrefslogtreecommitdiffstats
path: root/scanner/TreeWalker.py
diff options
context:
space:
mode:
authorJason A. Donenfeld <Jason@zx2c4.com>2011-05-05 20:03:47 -0400
committerJason A. Donenfeld <Jason@zx2c4.com>2011-05-05 20:03:47 -0400
commit8d66a56b9169f0ad8fe67e68a030e7c121bdb88f (patch)
tree60ecb6d2832be8105d4a9948b49e69500cd95453 /scanner/TreeWalker.py
parentCompare by name if dates are the same. (diff)
downloadPhotoFloat-8d66a56b9169f0ad8fe67e68a030e7c121bdb88f.tar.xz
PhotoFloat-8d66a56b9169f0ad8fe67e68a030e7c121bdb88f.zip
Add stump of web project.
Diffstat (limited to 'scanner/TreeWalker.py')
-rw-r--r--scanner/TreeWalker.py69
1 files changed, 69 insertions, 0 deletions
diff --git a/scanner/TreeWalker.py b/scanner/TreeWalker.py
new file mode 100644
index 0000000..17b9b54
--- /dev/null
+++ b/scanner/TreeWalker.py
@@ -0,0 +1,69 @@
+import os
+import os.path
+from datetime import datetime
+from PhotoAlbum import Photo, Album
+from CachePath import json_cache, set_cache_path_base, file_mtime
+
+class TreeWalker:
+ def __init__(self, album_path, cache_path):
+ self.album_path = os.path.abspath(album_path)
+ self.cache_path = os.path.abspath(cache_path)
+ set_cache_path_base(self.album_path)
+ self.all_albums = list()
+ self.all_photos = list()
+ self.walk(self.album_path)
+ self.remove_stale()
+ def walk(self, path):
+ print "Walking %s" % path
+ cache = os.path.join(self.cache_path, json_cache(path))
+ cached = False
+ cached_album = None
+ if os.path.exists(cache):
+ print "Has cache %s" % path
+ cached_album = Album.from_cache(cache)
+ if file_mtime(path) <= file_mtime(cache):
+ print "Album is fully cached"
+ cached = True
+ album = cached_album
+ for photo in album.photos:
+ self.all_photos.append(photo)
+ if not cached:
+ album = Album(path)
+ for entry in os.listdir(path):
+ entry = os.path.join(path, entry)
+ if os.path.isdir(entry):
+ album.add_album(self.walk(entry))
+ elif not cached and os.path.isfile(entry):
+ cache_hit = False
+ if cached_album:
+ cached_photo = cached_album.photo_from_path(entry)
+ if cached_photo and file_mtime(entry) <= cached_photo.attributes["DateTimeFile"]:
+ print "Photo cache hit %s" % entry
+ cache_hit = True
+ photo = cached_photo
+ if not cache_hit:
+ print "No cache, scanning %s" % entry
+ photo = Photo(entry, self.cache_path)
+ 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)
+ return album
+ def remove_stale(self):
+ 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:
+ print "Removing stale cache %s" % cache
+ os.unlink(os.path.join(self.cache_path, cache)) \ No newline at end of file