mirror of
https://github.com/flynx/ImageGrid.git
synced 2025-10-29 18:30:09 +00:00
156 lines
3.5 KiB
Python
Executable File
156 lines
3.5 KiB
Python
Executable File
#=======================================================================
|
|
|
|
__version__ = '''0.0.01'''
|
|
__sub_version__ = '''20130528154208'''
|
|
__copyright__ = '''(c) Alex A. Naanou 2011'''
|
|
|
|
|
|
#-----------------------------------------------------------------------
|
|
|
|
import os
|
|
import shutil
|
|
import json
|
|
import urllib2
|
|
|
|
from pli.logictypes import OR
|
|
|
|
from pprint import pprint
|
|
|
|
|
|
|
|
#-----------------------------------------------------------------------
|
|
|
|
|
|
|
|
|
|
#-----------------------------------------------------------------------
|
|
#----------------------------------------------------------build_dirs---
|
|
##!!! make this figure out the path by itself...
|
|
def build_dirs(data, path, rewrite=None):
|
|
'''
|
|
rewrite can be:
|
|
'force' - allways rewrite a destination file.
|
|
'older' - rewrite only older files.
|
|
None - never rewrite.
|
|
'''
|
|
version = data.get('version', 'gen1')
|
|
|
|
print 'FORMAT:', version
|
|
|
|
# gen3
|
|
if version == '2.0':
|
|
img = data['image_file']
|
|
# absolute path...
|
|
if img.startswith('file:///'):
|
|
images = json.load(urllib2.urlopen(data['image_file']))
|
|
|
|
# relative...
|
|
else:
|
|
images = json.load(open(os.path.join(path, data['image_file']), 'r'))
|
|
|
|
def get_image(gid, _):
|
|
return images[gid]
|
|
|
|
# gen1
|
|
else:
|
|
def get_image(gid, ribbon):
|
|
return ribbon[gid]
|
|
|
|
ribbons = data['ribbons']
|
|
|
|
depth = len(ribbons)-1
|
|
fav_path = os.path.join(path, *(['fav'] * depth))
|
|
|
|
if not os.path.exists(fav_path):
|
|
os.makedirs(fav_path)
|
|
|
|
err_urls = []
|
|
|
|
for i, ribbon in enumerate(ribbons):
|
|
level_path = os.path.join(path, *(['fav'] * (depth - i)))
|
|
|
|
for guid in ribbon:
|
|
image = get_image(guid, ribbon)
|
|
try:
|
|
p = image['path']
|
|
# absolute path...
|
|
if img.startswith('file:///'):
|
|
## # XXX for some magical reason this works and url2pathname does not...
|
|
## ##!!! this will also break on utf paths...
|
|
## img = urllib2.urlopen(image['path'])
|
|
## p = img.fp.name
|
|
## img.close()
|
|
p = urllib2.decode(p)
|
|
p = p.split('file:///')[-1]
|
|
|
|
# relative path...
|
|
else:
|
|
p = os.path.join(path, p)
|
|
|
|
# XXX do we need to indicate overwriting in the status?
|
|
if rewrite == 'force' or not os.path.exists(os.path.join(level_path, os.path.split(p)[-1])):
|
|
shutil.copy(p, level_path)
|
|
yield 'written', level_path, image
|
|
elif rewrite == 'older':
|
|
raise NotImplementedError
|
|
##!!! check dates...
|
|
##!!!
|
|
yield 'written', level_path, image
|
|
else:
|
|
yield 'skipped', level_path, image
|
|
|
|
except urllib2.URLError:
|
|
yield 'err', level_path, image
|
|
|
|
|
|
|
|
#-----------------------------------------------------------------------
|
|
if __name__ == '__main__':
|
|
from optparse import OptionParser
|
|
|
|
parser = OptionParser(
|
|
usage='%prog [options] DATAJSON TARGETDIR')
|
|
|
|
##!!! need to define the path so that it shoes up in -h
|
|
|
|
options, args = parser.parse_args()
|
|
|
|
# settings...
|
|
##!!! make this configurable...
|
|
|
|
if len(args) != 2:
|
|
parser.print_usage()
|
|
|
|
else:
|
|
data, path = args[0], args[1]
|
|
|
|
path = path.replace('\\', '/')
|
|
data = json.load(file(data, 'r'))
|
|
err_urls = []
|
|
cur_p = None
|
|
c = s = e = 0
|
|
|
|
for status, p, img in build_dirs(data, path):
|
|
if cur_p != p:
|
|
cur_p = p
|
|
|
|
if status == 'written':
|
|
c += 1
|
|
elif status == 'skipped':
|
|
s += 1
|
|
elif status == 'err':
|
|
e += 1
|
|
err_urls += [img['path']]
|
|
print 'Copied: %s, Skipped: %s, Err: %s\r' % (c, s, e),
|
|
print
|
|
|
|
if len(err_urls) != 0:
|
|
print 'Could not open %s files:' % len(err_urls)
|
|
pprint(err_urls)
|
|
|
|
|
|
|
|
|
|
#=======================================================================
|
|
# vim:set ts=4 sw=4 nowrap :
|