#!/usr/bin/python
"An HTML form to upload a picture, and process the uploaded object too"
import cgi
import sqlite3
import os.path
def guess_mimetype(ext):
if ext in ('.jpg','.jpeg'): return 'image/jpeg'
if ext == '.png': return 'image/png'
if ext == '.bmp': return 'image/bmp'
return 'application/octet-stream'
form = cgi.FieldStorage()
if form.has_key("upload"):
# the user has uploaded a picture
title = form['title'].value
description = form['description'].value
picture = form['picture'].value
ext = os.path.splitext(form['picture'].filename)[1]
mimetype = guess_mimetype(ext)
conn = sqlite3.connect("demo.db")
cursor = conn.cursor()
# fake autoincrement - unsafe!
cursor.execute("select max(id) from album")
picid = int(cursor.fetchone()[0] or 0)+1
sql = ("insert into album "
"(id, title, description, picture, mimetype) values "
"(?,?,?,?,?)")
cursor.execute(sql, (
picid, title, description, sqlite3.Binary(picture), mimetype))
conn.commit()
conn.close()
print "Content-Type: text/html"
print
print "Upload OK"
print ""
print "Upload OK
"
print "Title: %s
" % cgi.escape(title)
print "Description: %s
" % cgi.escape(description)
print "Size: %s
" % len(picture)
print "Back to album
"
print ""
else:
# show form to upload pictures
print "Content-Type: text/html"
print
print "Upload new picture"
print ""
print "Upload new picture
"
print ""
print "Back to album
"
print ""