mirror of
https://github.com/flynx/ImageGrid.git
synced 2025-10-28 18:00:09 +00:00
72 lines
1.5 KiB
Python
72 lines
1.5 KiB
Python
|
#=======================================================================
|
||
|
|
|
||
|
|
__version__ = '''0.0.01'''
|
||
|
|
__sub_version__ = '''20111103010916'''
|
||
|
|
__copyright__ = '''(c) Alex A. Naanou 2011'''
|
||
|
|
|
||
|
|
|
||
|
|
#-----------------------------------------------------------------------
|
||
|
|
|
||
|
|
import os
|
||
|
|
import json
|
||
|
|
|
||
|
|
from pli.logictypes import OR
|
||
|
|
|
||
|
|
|
||
|
|
#-----------------------------------------------------------------------
|
||
|
|
|
||
|
|
CONFIG_NAME = 'config.json'
|
||
|
|
|
||
|
|
config = json.load(open(CONFIG_NAME))
|
||
|
|
|
||
|
|
ITEM_EXTENSIONS = (
|
||
|
|
# RAW formats...
|
||
|
|
'NEF', 'nef',
|
||
|
|
# JPEGs...
|
||
|
|
'JPG', 'JPEG', 'jpg', 'jpeg',
|
||
|
|
# Editid images...
|
||
|
|
'PSD', 'psd',
|
||
|
|
'TIFF', 'tiff', 'TIF', 'tif',
|
||
|
|
# metadata sidecar files...
|
||
|
|
'XMP', 'xmp',
|
||
|
|
)
|
||
|
|
|
||
|
|
SUBTREE_CLASSES = {
|
||
|
|
'preview': 'preview',
|
||
|
|
'preview (RAW)': 'RAW preview',
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
#-----------------------------------------------------------------------
|
||
|
|
|
||
|
|
def list_files(root, sub_trees=SUBTREE_CLASSES, ext=OR(*ITEM_EXTENSIONS)):
|
||
|
|
'''
|
||
|
|
'''
|
||
|
|
for path, dirs, files in os.walk(root):
|
||
|
|
# clasify by subtree...
|
||
|
|
p = os.path.split(path)
|
||
|
|
subtree_type = None
|
||
|
|
for t in sub_trees:
|
||
|
|
if t in p:
|
||
|
|
subtree_type = sub_trees[t]
|
||
|
|
break
|
||
|
|
# process files...
|
||
|
|
for f in files:
|
||
|
|
# filter by ext...
|
||
|
|
if f.split('.')[-1] == ext:
|
||
|
|
yield subtree_type, path, f
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
#-----------------------------------------------------------------------
|
||
|
|
if __name__ == '__main__':
|
||
|
|
lst = list(list_files(config['ARCHIVE_ROOT']))
|
||
|
|
|
||
|
|
print len(lst)
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
#=======================================================================
|
||
|
|
# vim:set ts=4 sw=4 nowrap :
|