| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950 |
- import flickrapi, json, os
- import settings_local as settings
- ROOT = os.path.dirname(os.path.realpath(__file__))
- # Instantiate flickr client
- flickr = flickrapi.FlickrAPI(settings.FLICKR_KEY, settings.FLICKR_SECRET)
- (token, frob) = flickr.get_token_part_one(perms='write')
- if not token: raw_input("Press ENTER after you authorized this program")
- flickr.get_token_part_two((token, frob))
- # Create data dir if don't exists
- d = os.path.join(ROOT, 'data')
- if not os.path.exists(d):
- os.makedirs(d)
- # Populate a list of dicts with photosets id and titles
- photosets = [
- {
- 'id': ps.attrib['id'],
- 'title': ps[0].text
- } for ps in flickr.photosets_getList()[0]
- ]
- for i in range(len(photosets)):
- photoset = photosets[i]
- f = os.path.join(d, '%s.json' % (photoset['id']))
- if not os.path.isfile(f):
- photos = flickr.photosets_getPhotos(photoset_id=photoset['id'])[0]
- photos = [
- {
- 'id': photo.attrib['id'],
- 'title': photo.attrib['title'],
- } for photo in photos
- ]
-
- for j in range(len(photos)):
- sizes = flickr.photos_getSizes(photo_id=photos[j]['id'])[0]
- photos[j]['sizes'] = {
- size.attrib['label'].lower(): {
- 'width': size.attrib['width'],
- 'height': size.attrib['height'],
- 'source': size.attrib['source'],
- } for size in sizes
- }
- photoset['photos'] = photos
- with open(f, 'w') as jsonfile:
- json.dump(photoset, jsonfile, indent=4)
|
