#!/usr/bin/env python
'''
$Id: advcgi.py,v 1.1 2000/12/31 01:32:45 wesc Exp $
advcgi.py -- tests CGI file uploads and multi-valued CGI variables
Advanced CGI demo
created by wesc on 00/07/25
'''
from cgi import FieldStorage
from os import environ
from StringIO import StringIO
from urllib import quote, unquote
from string import capwords, strip, split, join
#from sys import stderr
#s = stderr.write
class AdvCGI:
header = 'Content-Type: text/html\n\n'
url = '/py/advcgi.py'
formhtml = '''
Advanced CGI Demo
Advanced CGI Demo Form
'''
langSet = ('Python', 'PERL', 'Java', 'C++', 'PHP', 'C', 'JavaScript')
langItem = ' %s\n'
# reads cookies from client, creates cookies, who, langs
def getCPPCookies(self):
if environ.has_key('HTTP_COOKIE'):
#s('reading cookies from server...\n')
for eachCookie in map(strip, split(environ['HTTP_COOKIE'], ';')):
if len(eachCookie) > 6 and eachCookie[:3] == 'CPP':
tag = eachCookie[3:7]
try:
self.cookies[tag] = eval(unquote(eachCookie[8:]))
except (NameError, SyntaxError):
self.cookies[tag] = unquote(eachCookie[8:])
else:
#s('no cookies on server...\n')
self.cookies['info'] = self.cookies['user'] = ''
#s('cookies: %s\n' % str(self.cookies))
if self.cookies['info'] != '':
self.who, langStr, self.fn = split(self.cookies['info'], ':')
self.langs = split(langStr, ',')
else:
self.who = self.fn = ''
self.langs = ['Python']
def showForm(self):
self.getCPPCookies() # get cookies
# put together lang checkboxes
langStr = ''
for eachLang in AdvCGI.langSet:
if eachLang in self.langs:
langStr = langStr + AdvCGI.langItem % (eachLang, ' CHECKED', eachLang)
else:
langStr = langStr + AdvCGI.langItem % (eachLang, '', eachLang)
# see if user cookie set up yet
if not self.cookies.has_key('user') or self.cookies['user'] == '':
cookieStatus = '(cookie has not been set yet)'
userCook = ''
else:
userCook = cookieStatus = self.cookies['user']
# output results
#s('filename: ' + self.fn + '\n')
print AdvCGI.header + AdvCGI.formhtml % (AdvCGI.url, cookieStatus, userCook, self.who, langStr, self.fn)
#print AdvCGI.header + AdvCGI.formhtml % (AdvCGI.url, cookieStatus, userCook, self.who, langStr)
errhtml = '''
Advanced CGI Demo
ERROR
%s
'''
def showError(self):
print AdvCGI.header + AdvCGI.errhtml % (self.error)
reshtml = '''
Advanced CGI Demo
Your Uploaded Data
Your cookie value is: %s
Your name is: %s
You can program in the following languages:
Your uploaded file...
Name: %s
Contents:
%s
Click here to return to form.
'''
def setCPPCookies(self):
for eachCookie in self.cookies.keys():
#s('setting %s cookie...\n' % eachCookie)
print 'Set-Cookie: CPP%s=%s; path=/' % (eachCookie, quote(self.cookies[eachCookie]))
def doResults(self):
MAXBYTES = 1024
langlist = ''
for eachLang in self.langs:
langlist = langlist + '%s
' % eachLang
filedata = ''
while len(filedata) < MAXBYTES:
data = self.fp.readline()
if data == '': break
filedata = filedata + data
else:
filedata = filedata + '... (file truncated due to size)'
self.fp.close()
if filedata == '':
filedata = '(file upload error or file not given)'
filename = self.fn
# see if user cookie set up yet
if not self.cookies.has_key('user') or self.cookies['user'] == '':
cookieStatus = '(cookie has not been set yet)'
userCook = ''
else:
userCook = cookieStatus = self.cookies['user']
# set cookies
self.cookies['info'] = join([self.who, join(self.langs, ','), filename], ':')
self.setCPPCookies()
# output page
print AdvCGI.header + AdvCGI.reshtml % (cookieStatus, self.who, langlist, filename, filedata, AdvCGI.url)
def __init__(self):
self.cookies = {}
def go(self):
self.error = ''
form = FieldStorage()
if form.keys() == []:
#s('calling showForm()\n')
self.showForm()
return
if form.has_key('person'):
self.who = capwords(strip(form['person'].value))
if self.who == '':
self.error = 'Your name is required. (blank)'
else:
self.error = 'Your name is required. (missing)'
if form.has_key('cookie'):
self.cookies['user'] = unquote(strip(form['cookie'].value))
else:
self.cookies['user'] = ''
self.langs = []
if form.has_key('lang'):
langdata = form['lang']
if type(langdata) == type([]):
for eachLang in langdata:
self.langs.append(eachLang.value)
else:
self.langs.append(langdata.value)
else:
self.error = 'At least one language required.'
if form.has_key('upfile'):
upfile = form["upfile"]
self.fn = upfile.filename or ''
#s('filename is %s??\n' % self.fn)
if upfile.file:
self.fp = upfile.file
else:
self.fp = StringIO('(no data)')
else:
self.fp = StringIO('(no file)')
self.fn = ''
if not self.error:
#s('calling doResults()\n')
self.doResults()
else:
#s('calling showError()\n')
self.showError()
if __name__ == '__main__':
page = AdvCGI()
page.go()