/Demo/cgi/wiki.py

http://unladen-swallow.googlecode.com/ · Python · 123 lines · 105 code · 16 blank · 2 comment · 22 complexity · bc14b50c72536292b5f4068537a7645a MD5 · raw file 

    

  1. """Wiki main program.  Imported and run by cgi3.py."""
    2. 

  2. 

  3. import os, re, cgi, sys, tempfile
    4. 

  4. escape = cgi.escape
    5. 

  5. 

  6. def main():
    7. 

  7.     form = cgi.FieldStorage()
    8. 

  8.     print "Content-type: text/html"
    9. 

  9.     print
    10. 

  10.     cmd = form.getvalue("cmd", "view")
    11. 

  11.     page = form.getvalue("page", "FrontPage")
    12. 

  12.     wiki = WikiPage(page)
    13. 

  13.     method = getattr(wiki, 'cmd_' + cmd, None) or wiki.cmd_view
    14. 

  14.     method(form)
    15. 

  15. 

  16. class WikiPage:
    17. 

  17. 

  18.     homedir = tempfile.gettempdir()
    19. 

  19.     scripturl = os.path.basename(sys.argv[0])
    20. 

  20. 

  21.     def __init__(self, name):
    22. 

  22.         if not self.iswikiword(name):
    23. 

  23.             raise ValueError, "page name is not a wiki word"
    24. 

  24.         self.name = name
    25. 

  25.         self.load()
    26. 

  26. 

  27.     def cmd_view(self, form):
    28. 

  28.         print "<h1>", escape(self.splitwikiword(self.name)), "</h1>"
    29. 

  29.         print "<p>"
    30. 

  30.         for line in self.data.splitlines():
    31. 

  31.             line = line.rstrip()
    32. 

  32.             if not line:
    33. 

  33.                 print "<p>"
    34. 

  34.             else:
    35. 

  35.                 print self.formatline(line)
    36. 

  36.         print "<hr>"
    37. 

  37.         print "<p>", self.mklink("edit", self.name, "Edit this page") + ";"
    38. 

  38.         print self.mklink("view", "FrontPage", "go to front page") + "."
    39. 

  39. 

  40.     def formatline(self, line):
    41. 

  41.         words = []
    42. 

  42.         for word in re.split('(\W+)', line):
    43. 

  43.             if self.iswikiword(word):
    44. 

  44.                 if os.path.isfile(self.mkfile(word)):
    45. 

  45.                     word = self.mklink("view", word, word)
    46. 

  46.                 else:
    47. 

  47.                     word = self.mklink("new", word, word + "*")
    48. 

  48.             else:
    49. 

  49.                 word = escape(word)
    50. 

  50.             words.append(word)
    51. 

  51.         return "".join(words)
    52. 

  52. 

  53.     def cmd_edit(self, form, label="Change"):
    54. 

  54.         print "<h1>", label, self.name, "</h1>"
    55. 

  55.         print '<form method="POST" action="%s">' % self.scripturl
    56. 

  56.         s = '<textarea cols="70" rows="20" name="text">%s</textarea>'
    57. 

  57.         print s % self.data
    58. 

  58.         print '<input type="hidden" name="cmd" value="create">'
    59. 

  59.         print '<input type="hidden" name="page" value="%s">' % self.name
    60. 

  60.         print '<br>'
    61. 

  61.         print '<input type="submit" value="%s Page">' % label
    62. 

  62.         print "</form>"
    63. 

  63. 

  64.     def cmd_create(self, form):
    65. 

  65.         self.data = form.getvalue("text", "").strip()
    66. 

  66.         error = self.store()
    67. 

  67.         if error:
    68. 

  68.             print "<h1>I'm sorry.  That didn't work</h1>"
    69. 

  69.             print "<p>An error occurred while attempting to write the file:"
    70. 

  70.             print "<p>", escape(error)
    71. 

  71.         else:
    72. 

  72.             # Use a redirect directive, to avoid "reload page" problems
    73. 

  73.             print "<head>"
    74. 

  74.             s = '<meta http-equiv="refresh" content="1; URL=%s">'
    75. 

  75.             print s % (self.scripturl + "?cmd=view&page=" + self.name)
    76. 

  76.             print "<head>"
    77. 

  77.             print "<h1>OK</h1>"
    78. 

  78.             print "<p>If nothing happens, please click here:",
    79. 

  79.             print self.mklink("view", self.name, self.name)
    80. 

  80. 

  81.     def cmd_new(self, form):
    82. 

  82.         self.cmd_edit(form, label="Create")
    83. 

  83. 

  84.     def iswikiword(self, word):
    85. 

  85.         return re.match("[A-Z][a-z]+([A-Z][a-z]*)+", word)
    86. 

  86. 

  87.     def splitwikiword(self, word):
    88. 

  88.         chars = []
    89. 

  89.         for c in word:
    90. 

  90.             if chars and c.isupper():
    91. 

  91.                 chars.append(' ')
    92. 

  92.             chars.append(c)
    93. 

  93.         return "".join(chars)
    94. 

  94. 

  95.     def mkfile(self, name=None):
    96. 

  96.         if name is None:
    97. 

  97.             name = self.name
    98. 

  98.         return os.path.join(self.homedir, name + ".txt")
    99. 

  99. 

  100.     def mklink(self, cmd, page, text):
    101. 

  101.         link = self.scripturl + "?cmd=" + cmd + "&page=" + page
    102. 

  102.         return '<a href="%s">%s</a>' % (link, text)
    103. 

  103. 

  104.     def load(self):
    105. 

  105.         try:
    106. 

  106.             f = open(self.mkfile())
    107. 

  107.             data = f.read().strip()
    108. 

  108.             f.close()
    109. 

  109.         except IOError:
    110. 

  110.             data = ""
    111. 

  111.         self.data = data
    112. 

  112. 

  113.     def store(self):
    114. 

  114.         data = self.data
    115. 

  115.         try:
    116. 

  116.             f = open(self.mkfile(), "w")
    117. 

  117.             f.write(data)
    118. 

  118.             if data and not data.endswith('\n'):
    119. 

  119.                 f.write('\n')
    120. 

  120.             f.close()
    121. 

  121.             return ""
    122. 

  122.         except IOError, err:
    123. 

  123.             return "IOError: %s" % str(err)
    124.