def facebook_mode(self):
# Connect to Facebook
logging.debug("list of tags: %s" % self.photo_tags)
self.facebookphotos = FacebookPhotos()
# if len(self.photo_tags) > 0:
# photo_filenames = self.flickr.get_tagged(self.photo_tags, self.photo_count, download_dir=self.download_dir)
# else:
# photo_filenames = self.flickr.get_recent(self.photo_count,download_dir=self.download_dir)
photo_filenames = self.facebookphotos.get_recent(self.photo_count,download_dir=self.download_dir)
return photo_filenames
class AirFrame(object):
def __init__(self):
pass
def _parse_csv_list(self, s):
try:
assert isinstance(s,str)
s = s.strip()
value_list = s.split(',')
except:
raise argparse.ArgumentTypeError("Could not parse tag list")
return value_list
def get_options(self, argv):
"""
Parse the command-line options and set the following object properties:
I really need to convert this to use docopt!
:param argv: usually just sys.argv[1:]
:returns: Nothing
:ivar debug: Enable logging debug statements
:ivar verbose: Enable verbose logging
"""
p = argparse.ArgumentParser(
description = "Push pictures from Flickr, Facebook or local files to a Toshiba FlashAir automatically",
epilog = "AirFrame version %s (Copyright 2014 Virantha Ekanayake)" % __version__,
)
p.add_argument('-b', '--facebook', action='store_true',
default=False, help='Upload pictures from Flickr to Flashair')
p.add_argument('-r', '--flickr', action='store_true',
default=True, help='Upload pictures from Facebook to Flashair (default)')
p.add_argument('-l', '--local-dir', type=str,
help='Upload all .jpg files from this directory instead of Flickr or Facebook')
p.add_argument('-s', '--resize', type=str,
help='Resize all images to fit in this box (e.g. 1024x768) before uploading them to Flashair')
p.add_argument('-d', '--debug', action='store_true',
default=False, dest='debug', help='Turn on debugging')
p.add_argument('-v', '--verbose', action='store_true',
default=False, dest='verbose', help='Turn on verbose mode')
p.add_argument('-f', '--force', action='store_true',
default=False, help='Force upload of all pictures to Flashair (instead of only new pictures)')
p.add_argument('-n', '--number', type=int,
default=100, dest='number', help='Max number of photos to sync')
p.add_argument('-t', '--tags', type=self._parse_csv_list,
default=[], dest='tags', help='List of Flickr tags to match')
p.add_argument('flashair_ip', type=str,
help='The ip/hostname of your FlashAir card')
args = p.parse_args(argv)
self.debug = args.debug
self.verbose = args.verbose
self.photo_count = args.number
self.photo_tags = args.tags
self.force_upload = args.force
self.flashair_ip = args.flashair_ip
self.local_dir = args.local_dir
self.resize = args.resize
self.facebook = args.facebook
self.flickr = args.flickr
if self.debug:
logging.basicConfig(level=logging.DEBUG, format='%(message)s')
if self.verbose:
logging.basicConfig(level=logging.INFO, format='%(message)s')
def flickr_mode(self):
# Connect to Flickr
logging.debug("list of tags: %s" % self.photo_tags)
self.flickr = Flickr()
if len(self.photo_tags) > 0:
photo_filenames = self.flickr.get_tagged(self.photo_tags, self.photo_count, download_dir=self.download_dir)
else:
photo_filenames = self.flickr.get_recent(self.photo_count,download_dir=self.download_dir)
return photo_filenames
def facebook_mode(self):
# Connect to Facebook
logging.debug("list of tags: %s" % self.photo_tags)
self.facebookphotos = FacebookPhotos()
# if len(self.photo_tags) > 0:
# photo_filenames = self.flickr.get_tagged(self.photo_tags, self.photo_count, download_dir=self.download_dir)
# else:
# photo_filenames = self.flickr.get_recent(self.photo_count,download_dir=self.download_dir)
photo_filenames = self.facebookphotos.get_recent(self.photo_count,download_dir=self.download_dir)
return photo_filenames
def local_dir_mode(self):
# Copy all the files in the named directory to the cache (download_dir).
# completely replaces both the local cached files and the ones on the
# wifi sd card
pattern = '/*.[jJ][pP][gG]'
match = self.local_dir + pattern
if os.path.isdir(self.download_dir):
logging.debug("removing existing cache dir: %s", self.download_dir)
shutil.rmtree(self.download_dir)
os.mkdir(self.download_dir)
logging.debug("caching files from: %s" % match)
photo_filenames = glob.glob(match)
for filename in photo_filenames:
logging.debug("copy %s to %s" % (filename, self.download_dir))
shutil.copy(filename, self.download_dir)
return photo_filenames
def resize_pictures(self, photo_filenames):
print 'Resizing images...'
size = self.resize.split('x')
size = [int(x) for x in tuple(size)]
for infile in photo_filenames:
im = Image.open(infile)
if im.size[0] > size[0] or im.size[1] > size[1]:
im.thumbnail(size, Image.ANTIALIAS)
im.save(infile)
im.close()
def go(self, argv):
self.download_dir = ".airframe"
self.get_options(argv)
if self.local_dir:
photo_filenames = self.local_dir_mode()
elif self.facebook:
photo_filenames = self.facebook_mode()
else:
photo_filenames = self.flickr_mode()
if self.resize:
self.resize_pictures(photo_filenames)
self.flashair = FlashAir(self.flashair_ip)
self.flashair.sync_files_on_card_to_list(photo_filenames, self.force_upload)