"""
Per-directory gallery state
"""
import filesystem, image, html
from utils import lazy_load, lazy_load_iter
import math
class Folder (filesystem.Directory) :
"""
A Folder is a filesystem Directory that contains any number of other Folders and Images.
"""
def __init__ (self, *args, **kwargs) :
super(Folder, self).__init__(*args, **kwargs)
# info
self.title = self.name.title()
self.description = None
@lazy_load
def preview_dir (self) :
"""
Load and return the Directory for previews
"""
return self.subdir(self.config.preview_dir, create=True)
@lazy_load
def thumb_dir (self) :
"""
Load and return the Directory for thumbs
"""
return self.subdir(self.config.thumb_dir, create=True)
@lazy_load_iter
def subnodes (self) :
"""
Load and return an ordered list of child-Nodes
"""
return super(Folder, self).subnodes(skip_dotfiles=True, sort=True)
@lazy_load_iter
def subfolders (self) :
"""
Load and return an ordered list of sub-Folders
"""
# filter out valid subfolders
for node in self.subnodes :
# skip non-dirs
if not node.is_dir() :
continue
# skip the previews/thumbs dirs
if node.name in (self.config.preview_dir, self.config.thumb_dir) :
continue
# ok
yield Folder(node)
@lazy_load_iter
def images (self) :
"""
Load and return an ordered/linked list of sub-Images
"""
prev = None
for node in self.subnodes :
# skip non-relevant ones
if not node.is_file() or not self.config.is_image(filesystem.File(node)) :
continue
# create new
img = image.Image(node)
# link up
if prev :
img.prev = prev
prev.next = img
# yield the linked-up prev
yield prev
# continue
prev = img
# and the last one
if prev :
yield prev
@property
def page_count (self) :
"""
Returns the number of pages needed to show this folder's images
"""
return int(math.ceil(len(self.images) / float(self.config.images_per_page)))
def images_for_page (self, page) :
"""
Returns the list of Images to be displayed for the given page, if any
"""
# offset to first image
offset = page * self.config.images_per_page
# slice
return self.images[offset : offset + self.config.images_per_page]
def html_file (self, page=0) :
"""
Returns the File representing the .html for the given page
"""
if page :
return self.subfile("index_%d.html" % page)
else :
return self.subfile("index.html")
def index_images (self, for_update=True) :
"""
Return a series of Images inside of this folder.
If `for_update` is given, only images that are stale will be returned.
"""
for image in self.images :
if for_update and not image.stale() :
# skip
continue
yield image