Now with less embarrassment!

flickrscripts.py

@@ -0,0 +1,11 @@
+import flickrapi
+
+
+def setup_flickr(self, api_key, api_secret):
+    self.flickr = flickrapi.FlickrAPI(api_key, api_secret)
+
+    token, frob = self.flickr.get_token_part_one(perms='write')
+    if not token:
+        raw_input("Press ENTER after you authorized this program")
+
+    self.flickr.get_token_part_two((token, frob))

timewarp.py

@@ -1,78 +1,94 @@
-from __future__ import print_function
-
 import argparse
 import json
+import logging
 import sys
+from datetime import datetime, timedelta
 from os.path import expanduser
 
-import flickrapi
+from flickrscripts import setup_flickr
 
 
 class TimeWarper(object):
+    flickr = None
+    _base_query_args = dict(user_id='me', page=1)
 
-    def __init__(self, sysargs=sys.argv[:]):
+    def __init__(self, api_key, api_secret):
-        self.sysargs = sysargs
+        setup_flickr(self, api_key, api_secret)
-        parser = argparse.ArgumentParser()
+        self.log = logging.getLogger(self.__class__.__name__)
-        parser.add_argument('min_upload_date')
+        self.log.level = logging.INFO
-        parser.add_argument('max_upload_date')
-        parser.add_argument('set_to_date')
 
-        parser.add_argument(
+    def run(self, set_to_date, **query_args):
-            '-c', '--rcfile', default=expanduser('~/.frupplerc'),
+        self.log.info(
-            type=argparse.FileType('r')
+            'Setting upload dates for photos matching query %r', query_args
         )
+        query = self._base_query_args.copy()
+        query.update(query_args)
 
-        self.args = parser.parse_args(self.sysargs[1:])
+        for i, page in self._get_pages(query):
-        self._rc_conf = json.load(self.args.rcfile)
+            self.log.info('Processing page %s, attributes=%r',
-
+                         i, page.attrib)
-        self.flickr = flickrapi.FlickrAPI(
+            self._set_upload_date_for_photos(i, page, set_to_date)
-            self._rc_conf['APIKEY'], self._rc_conf['APISECRET']
-        )
-
-        token, frob = self.flickr.get_token_part_one(perms='write')
-        if not token:
-            raw_input("Press ENTER after you authorized this program")
-
-        self.flickr.get_token_part_two((token, frob))
-
-    def set_upload_date_for_photos(self, pagenum, photo_nodes, to_date):
-        for i, photo in enumerate(photo_nodes):
-            setdate_response = self.flickr.photos_setDates(
-                photo_id=photo.attrib['id'],
-                date_posted=to_date
-            )
-            print('page {} photo {}: rsp={}, attribs={}'.format(
-                  pagenum, i, setdate_response, setdate_response.attrib))
-
-    def run(self):
-        query_args = dict(
-            user_id='me',
-            min_upload_date=self.args.min_upload_date,
-            max_upload_date=self.args.max_upload_date,
-            page=1
-        )
-
-        print('Getting first page')
-        firstpage = self.flickr.photos_search(**query_args)
-        self.set_upload_date_for_photos(
-            1,
-            firstpage.getchildren()[0].getchildren(),
-            self.args.set_to_date
-        )
-
-        npages = int(firstpage.getchildren()[0].attrib['pages'])
-
-        for pagenum in range(1, npages + 1):
-            print('Getting another page')
-            page = self.flickr.photos_search(**query_args)
-            self.set_upload_date_for_photos(
-                pagenum,
-                page.getchildren()[0].getchildren(),
-                self.args.set_to_date
-            )
 
         return 0
 
+    def _set_upload_date_for_photos(self, pagenum, page, set_to_date):
+        for i, photo in enumerate(page.getchildren()[0].getchildren()):
+            setdate_response = self.flickr.photos_setDates(
+                photo_id=photo.attrib['id'],
+                date_posted=set_to_date
+            )
+            self.log.info('page %s photo %s: response attribs=%s',
+                  pagenum, i, setdate_response.attrib)
+
+    def _get_pages(self, query):
+        self.log.info('Getting pages for query %r', query)
+        i = 1
+        while True:
+            page = self.flickr.photos_search(**query)
+            if len(page.getchildren()[0].getchildren()):
+                yield i, page
+                i += 1
+            else:
+                raise StopIteration
+
+
+def main(sysargs=sys.argv[:]):
+    parser = argparse.ArgumentParser(
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter)
+    parser.add_argument('set_to_date')
+    parser.add_argument('-q', '--search-query-param', default=[],
+        action='append', help='Add a key=value param to the search query list')
+
+    parser.add_argument(
+        '-c', '--rcfile', default=expanduser('~/.flickrscripts.json'),
+        help='JSON file containing `APIKEY` and `APISECRET` keys',
+        type=argparse.FileType('r')
+    )
+
+    args = parser.parse_args(sysargs[1:])
+    rc_conf = json.load(args.rcfile)
+
+    logging.basicConfig(level=logging.INFO, stream=sys.stdout)
+    log = logging.getLogger('main')
+    log.level = logging.INFO
+    log.info('Preparing query args')
+
+    query_args = {}
+    for key_val in args.search_query_param:
+        key, value = key_val.split('=', 1)
+        query_args[key] = value
+
+    if len(query_args.keys()) == 0:
+        query_args = dict(
+            min_upload_date=(datetime.now() + timedelta(days=-1)),
+            max_upload_date=(datetime.now() + timedelta(days=1))
+        )
+        log.warn('No query arguments provided!  ' +
+                 'Defaulting to %r', query_args)
+
+    warper = TimeWarper(rc_conf['APIKEY'], rc_conf['APISECRET'])
+    return warper.run(args.set_to_date, **query_args)
+
 
 if __name__ == '__main__':
-    sys.exit(TimeWarper().run())
+    sys.exit(main())