Create album directories from Flickr album JSON. Copy over JPGs to relevant directories

create_album.py

import json
import shutil
import string
import re
import os
import glob

with open("albums.json", "r") as read_file:
    data = json.load(read_file)

albums = data["albums"]

for album in albums:
    title = album["title"]

    title_stripped = re.sub(r'[^\w\s]','',title) 
    title_stripped2 = re.sub(r'[\s]','_',title_stripped) 
    album_dir = "albums/{0}".format(title_stripped2)
    if not os.path.isdir(album_dir):
        os.makedirs(album_dir)

    print title_stripped2

    for photo_id in album["photos"]:
        print photo_id
        with open("photo_{0}.json".format(photo_id), "r") as photo_file:
            photo_json = json.load(photo_file)

        shutil.copy("photo_{0}.json".format(photo_id), "{0}/".format(album_dir))
        listing = glob.glob('*/*_{0}_o.jpg'.format(photo_id))
        for filename in listing:
           shutil.copy(filename, "{0}/".format(album_dir))