#!/usr/bin/python

"Main entry point; listing of currently uploaded pictures"

import cgi
import sqlite3

form = cgi.FieldStorage()
if form.has_key("init"):
    import os
    try: os.remove("demo.db")
    except OSError: pass
    conn = sqlite3.connect("demo.db")
    cursor = conn.cursor()
    cursor.execute("create table album "
                   " (id integer, title text, "
                   " description text, picture binary, "
                   " mimetype text)")
    conn.close()
    
conn = sqlite3.connect("demo.db")
cursor = conn.cursor()
sql = "select id, title from album order by id"
cursor.execute(sql)
lines = []
for row in cursor.fetchall():
    id, title = row
    lines.append("<li><a href='showpic.py?id=%s'>%s</a></li>" % (id, cgi.escape(title)))
conn.close()

print "Content-Type: text/html"
print "Pragma: no-cache" 
print "Cache-Control: no-cache"
print
print "<html><head><title>Picture album</title></head>"
print "<body>"
print "<h1>Picture album</h1>"
if lines:
    print "<ul>"
    print "\n".join(lines)
    print "</ul>"
else:
    print "No pictures yet."
print "<p>You can <a href='upload.py'>upload a new picture</a>.</p>"
print "</body></html>"
