#!/usr/bin/python

import sys
import os

os.environ['DJANGO_SETTINGS_MODULE'] = 'settings'

sys.path.append("../sommitrealweird")

from django.conf import settings
from photo.models import Album, Photo

def usage():
    print "You must specify a photo album to update."

# check if we have an argument for the album name
if len(sys.argv) < 2:
    usage()
    sys.exit(2)

alb_slug = sys.argv[1]

# check if the album exists...
if not Album.objects.filter(slug__exact=alb_slug):
    # there's no album, can't update.
    print "Can't find album '%s'" %(alb_slug)

album = Album.objects.get(slug__exact=alb_slug)

# loop through photos, carefully
root, dirs, files = os.walk("../media/photos/%s" %(alb_slug)).next()
for file in files:
    if not Photo.objects.filter(image__exact="photos/%s/%s" %(alb_slug, file)):
        if file[-4:] == ".jpg":
            photo = Photo()
            photo.album = album
            photo.image = "photos/%s/%s" %(alb_slug, file)
            photo.save()