Home | History | Annotate | Download | only in cgi 
      1 """Wiki main program.  Imported and run by cgi3.py."""
      2 
      3 import os, re, cgi, sys, tempfile
      4 escape = cgi.escape
      5 
      6 def main():
      7     form = cgi.FieldStorage()
      8     print "Content-type: text/html"
      9     print
     10     cmd = form.getvalue("cmd", "view")
     11     page = form.getvalue("page", "FrontPage")
     12     wiki = WikiPage(page)
     13     method = getattr(wiki, 'cmd_' + cmd, None) or wiki.cmd_view
     14     method(form)
     15 
     16 class WikiPage:
     17 
     18     homedir = tempfile.gettempdir()
     19     scripturl = os.path.basename(sys.argv[0])
     20 
     21     def __init__(self, name):
     22         if not self.iswikiword(name):
     23             raise ValueError, "page name is not a wiki word"
     24         self.name = name
     25         self.load()
     26 
     27     def cmd_view(self, form):
     28         print "<h1>", escape(self.splitwikiword(self.name)), "</h1>"
     29         print "<p>"
     30         for line in self.data.splitlines():
     31             line = line.rstrip()
     32             if not line:
     33                 print "<p>"
     34             else:
     35                 print self.formatline(line)
     36         print "<hr>"
     37         print "<p>", self.mklink("edit", self.name, "Edit this page") + ";"
     38         print self.mklink("view", "FrontPage", "go to front page") + "."
     39 
     40     def formatline(self, line):
     41         words = []
     42         for word in re.split('(\W+)', line):
     43             if self.iswikiword(word):
     44                 if os.path.isfile(self.mkfile(word)):
     45                     word = self.mklink("view", word, word)
     46                 else:
     47                     word = self.mklink("new", word, word + "*")
     48             else:
     49                 word = escape(word)
     50             words.append(word)
     51         return "".join(words)
     52 
     53     def cmd_edit(self, form, label="Change"):
     54         print "<h1>", label, self.name, "</h1>"
     55         print '<form method="POST" action="%s">' % self.scripturl
     56         s = '<textarea cols="70" rows="20" name="text">%s</textarea>'
     57         print s % self.data
     58         print '<input type="hidden" name="cmd" value="create">'
     59         print '<input type="hidden" name="page" value="%s">' % self.name
     60         print '<br>'
     61         print '<input type="submit" value="%s Page">' % label
     62         print "</form>"
     63 
     64     def cmd_create(self, form):
     65         self.data = form.getvalue("text", "").strip()
     66         error = self.store()
     67         if error:
     68             print "<h1>I'm sorry.  That didn't work</h1>"
     69             print "<p>An error occurred while attempting to write the file:"
     70             print "<p>", escape(error)
     71         else:
     72             # Use a redirect directive, to avoid "reload page" problems
     73             print "<head>"
     74             s = '<meta http-equiv="refresh" content="1; URL=%s">'
     75             print s % (self.scripturl + "?cmd=view&page=" + self.name)
     76             print "<head>"
     77             print "<h1>OK</h1>"
     78             print "<p>If nothing happens, please click here:",
     79             print self.mklink("view", self.name, self.name)
     80 
     81     def cmd_new(self, form):
     82         self.cmd_edit(form, label="Create")
     83 
     84     def iswikiword(self, word):
     85         return re.match("[A-Z][a-z]+([A-Z][a-z]*)+", word)
     86 
     87     def splitwikiword(self, word):
     88         chars = []
     89         for c in word:
     90             if chars and c.isupper():
     91                 chars.append(' ')
     92             chars.append(c)
     93         return "".join(chars)
     94 
     95     def mkfile(self, name=None):
     96         if name is None:
     97             name = self.name
     98         return os.path.join(self.homedir, name + ".txt")
     99 
    100     def mklink(self, cmd, page, text):
    101         link = self.scripturl + "?cmd=" + cmd + "&page=" + page
    102         return '<a href="%s">%s</a>' % (link, text)
    103 
    104     def load(self):
    105         try:
    106             f = open(self.mkfile())
    107             data = f.read().strip()
    108             f.close()
    109         except IOError:
    110             data = ""
    111         self.data = data
    112 
    113     def store(self):
    114         data = self.data
    115         try:
    116             f = open(self.mkfile(), "w")
    117             f.write(data)
    118             if data and not data.endswith('\n'):
    119                 f.write('\n')
    120             f.close()
    121             return ""
    122         except IOError, err:
    123             return "IOError: %s" % str(err)
    124