#!/usr/bin/env python import cgi from urllib import quote_plus from string import capwords header = 'Content-Type: text/html\n\n' url = '/cgi-bin/friends3.py' errhtml = ''' Friends CGI Demo

ERROR

%s

''' def showError(error_str): print header + errhtml % (error_str) formhtml = ''' Friends CGI Demo

Friends list for: %s

Your Name:

How many friends do you have? %s

''' fradio = ' %s\n' def showForm(who, howmany): friends = '' for i in [0, 10, 25, 50, 100]: checked = '' if str(i) == howmany: checked = 'CHECKED' friends = friends + fradio % \ (str(i), checked, str(i)) print header + formhtml % (who, url, who, friends) reshtml = ''' Friends CGI Demo

Friends list for: %s

Your name is: %s

You have %s friends.

Click here to edit your data again. ''' def doResults(who, howmany): newurl = url + '?action=reedit&person=%s&howmany=%s' % \ (quote_plus(who), howmany) print header + reshtml % (who, who, howmany, newurl) def process(): error = '' form = cgi.FieldStorage() if form.has_key('person'): who = capwords(form['person'].value) else: who = 'NEW USER' if form.has_key('howmany'): howmany = form['howmany'].value else: if form.has_key('action') and \ form['action'].value == 'edit': error = 'Please select number of friends.' else: howmany = 0 if not error: if form.has_key('action') and \ form['action'].value != 'reedit': doResults(who, howmany) else: showForm(who, howmany) else: showError(error) if __name__ == '__main__': process()