diff options
Diffstat (limited to 'scanner/TreeWalker.py')
| -rw-r--r-- | scanner/TreeWalker.py | 69 |
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 |