#!/usr/bin/env python

# -*- encoding:utf8 -*-

top = "."

feed_title = "Gauret : photos récentes"

start_url = "http://aurelien.bompard.org/photos"

limit = 100

feed_author = "Aurélien Bompard"

import sys, os, datetime, cgi, ConfigParser

if len(sys.argv) > 1:

    top = sys.argv[1]

use_exif = True

try:

    import PIL.Image, PIL.ExifTags

except ImportError:

    use_exif = False

    print "Warning: PIL is not installed, no EXIF info will be used"

files = []

for dirpath, dirnames, filenames in os.walk(top):

    # exclude directories starting with "_"

    for i, d in enumerate(dirnames):

        if d.startswith("_"):

            del dirnames[i]

    for name in filenames:

        # only include jpg images and videos

        if name.lower().endswith(".jpg") \

                or name.lower().endswith(".flv") \

                or name.lower().endswith(".avi"):

            files.append(os.path.join(dirpath, name))

metadata = {"mtimes": {}, "sizes": {}}

for f in files:

    stat = os.stat(f)

    metadata["mtimes"][f] = stat.st_mtime

    metadata["sizes"][f] = stat.st_size

def get_mtime(f):

    return metadata["mtimes"][f]

files.sort( key=get_mtime, reverse=True )

files = files[:limit]

output = """<?xml version="1.0" encoding="utf-8" standalone="yes"?>

<feed xmlns="http://www.w3.org/2005/Atom"

      xmlns:dc="http://purl.org/dc/elements/1.1/" >

	<title>%(title)s</title>

	<link rel="self" href="%(url)s/_atom.xml" />

	<link rel="alternate" type="text/html" href="%(url)s"/>

	<id>%(url)s</id>

    <updated>%(time)sZ</updated>

    <generator uri="http://aurelien.bompard.org/site/Mon-afficheur-de-photos-par-le-web">Photos-Index</generator>

	<author>

		<name>%(author)s</name>

		<uri>%(url)s</uri>

	</author>

""" % { "title": feed_title,

        "url": start_url,

        "time": datetime.datetime.now().isoformat(),

        "author": cgi.escape(feed_author, quote=True),

      }

for f in files:

    relpath = f.replace(top,"")

    url = start_url+relpath

    url_sb = start_url+os.path.dirname(relpath)+"/?name="+os.path.basename(f)

    title = os.path.basename(f)

    author = None

    dir_name = os.path.basename(os.path.dirname(f))

    thumb = os.path.join( os.path.dirname(url), "_thumbs", os.path.basename(f)[:-4]+".jpg" )

    # Read _infos.ini file

    infos_file = os.path.join(os.path.dirname(f), "_infos.ini")

    if os.path.exists(infos_file):

        infos = ConfigParser.ConfigParser()

        infos.read(infos_file)

        if infos.has_option(os.path.basename(f), "title"):

            title = infos.get(os.path.basename(f), "title").strip('"')

        if infos.has_option(os.path.basename(f), "author"):

            author = infos.get(os.path.basename(f), "author").strip('"')

        if infos.has_option("general", "title"):

            dir_name = infos.get("general", "title").strip('"')

    # Author

    if author is None and dir_name.startswith("Photos de "):

        author = dir_name[10:]

    if author is None and dir_name.startswith("Photos d'"):

        author = dir_name[9:]

    if author is None:

        author = feed_author

    # Is it protected ? Only embed the thumbnail if it's not

    protected = False

    subdirs = os.path.dirname(relpath[1:]).split("/")

    for i in range(len(subdirs)):

        subdir = os.path.join(top,"/".join(subdirs[:i+1]))

        if os.path.exists(os.path.join(subdir, ".htaccess")):

            protected = True

    # Summary

    if protected:

        content = '<a href="%s">%s</a>' % (url_sb, title)

    else: # it's not protected, embed the thumbnail

        content = '<a href="%s"><img style="border:none" src="%s" /></a>' % (url_sb, thumb)

    content += " (dans %s)" % os.path.dirname(relpath)

    # MIME type

    if f.endswith(".jpg"):

        ftype = "image/jpeg"

    elif f.endswith(".flv"):

        ftype = "video/x-flv"

    elif f.endswith(".avi"):

        ftype = "video/x-msvideo"

    # Make XML

    output += """

	<entry>

		<title>%(title)s</title>

		<link rel="alternate" type="text/html" href="%(url_sb)s"/>

		<id>%(url)s</id>

		<updated>%(mtime)sZ</updated>

		<content type="html">%(content)s</content>

		<link rel="enclosure" type="%(type)s" length="%(size)s" href="%(url)s"/>

""" % {"title": cgi.escape(title, quote=True),

       "url_sb": url_sb,

       "url": url,

       "mtime": datetime.datetime.fromtimestamp(metadata["mtimes"][f]).isoformat(),

       "content": cgi.escape(content, quote=True),

       "size": metadata["sizes"][f],

       "type": ftype,

      }

    output += """		<author>

			<name>"""+cgi.escape(author, quote=True)+"""</name>

		</author>

"""

    if use_exif:

        try:

            exif = PIL.Image.open(f)._getexif()

        except (IOError, KeyError):

            exif = None

        if exif is not None:

            try:

                exif_dt = exif[0x0132] # DateTime

                exif_dt = datetime.datetime(int(exif_dt[0:4]),int(exif_dt[5:7]),int(exif_dt[8:10]),

                                            int(exif_dt[11:13]),int(exif_dt[14:16]),int(exif_dt[17:19]))

                output += """		<dc:date.Taken>"""+exif_dt.isoformat()+"""Z</dc:date.Taken>"""

            except KeyError, e:

                print "Warning: No EXIF DateTime info on", title

            except ValueError, e:

                print "Warning: EXIF data is wrong:", e, "on", exif_dt, "("+title+")"

        else:

            print "Warning: Can't extract EXIF data from", f

    output += """

	</entry>

"""

output += "</feed>\n"

out = open(os.path.join(top,"_atom.xml"), "w")

out.write(output)

out.close()