phd.py

   1 import os, re, time, urllib
   2 from Cheetah.Template import Template
   3
   4
   5 url_re = r"(((https?|ftp|gopher|telnet)://|(mailto|file|news|about|ed2k|irc|sip|magnet):)[^' \t<>\"]+|(www|web|w3)[A-Za-z0-9_-]*\.[A-Za-z0-9._-]+\.[^' \t<>\"]+)[A-Za-z0-9/]"
   6
   7 def _url2href(match):
   8    url = match.group(0)
   9    return '<a href="%s">%s</a>' % (url, url)
  10
  11
  12 full_dirs = len(os.getcwd().split('/')) + 1
  13
  14 class phd(Template):
  15    def __init__(self, *args, **kw):
  16       if not hasattr(self, "_fileBaseName"):
  17          self._fileDirName, self._fileBaseName = os.path.split(os.path.abspath(self._CHEETAH_src))
  18       Template.__init__(self, *args, **kw)
  19       directories = self._fileDirName.split('/')[full_dirs:] # remove directories up to "./files"
  20       dirs_to_root = len(directories)
  21       if dirs_to_root:
  22          root = "../"*dirs_to_root
  23       else:
  24          root = ''
  25       self.root = root
  26       path = '/'.join(directories) + '/' + \
  27          self._fileBaseName.replace(".tmpl", ".html")
  28       if path[0] <> '/': path = '/' + path
  29       self.path = path
  30
  31    def copyright(self, start_year):
  32       this_year = time.localtime()[0]
  33       if start_year >= this_year:
  34          return this_year
  35       if start_year == this_year - 1:
  36          return "%s, %s" % (start_year, this_year)
  37       return "%s-%s" % (start_year, this_year)
  38
  39
  40    def body(self):
  41       if hasattr(self, "body_html"):
  42          return self.body_html()
  43       if hasattr(self, "body_text"):
  44          return self.text2html()
  45       if hasattr(self, "body_rst"):
  46          return self.rst2html()
  47       if hasattr(self, "body_mkd"):
  48          return self.mkd2html()
  49
  50    def text2html(self):
  51       body = re.sub(url_re, _url2href, self.body_text())
  52
  53       paragraphs = body.split("\n\n")
  54
  55       new_paras = []
  56       for p in paragraphs:
  57          parts = p.split("\n   ")
  58          parts[0] = parts[0].strip()
  59          new_paras.append('\n</p>\n<p>\n'.join(parts))
  60
  61       if self.Title:
  62          title = "<h1>%s</h1>\n\n" % self.Title
  63       else:
  64          title = ''
  65
  66       body = '\n</p>\n\n<p class="head">\n'.join(new_paras)
  67       return "%s<p>%s</p>" % (title, body)
  68
  69    def rst2html(self):
  70       from docutils.core import publish_parts
  71
  72       parts = publish_parts(self.body_rst(), writer_name="html")
  73
  74       title = parts["title"] or self.Title
  75       if title:
  76          title = "<h1>%s</h1>" % title
  77
  78       subtitle = parts["subtitle"]
  79       if subtitle:
  80          subtitle = "<h2>%s</h2>" % subtitle
  81
  82       body = parts["body"]
  83       parts = []
  84       for part in (title, subtitle, body):
  85           if not part:
  86               continue
  87           if isinstance(part, unicode):
  88               part = part.encode('utf-8')
  89           parts.append(part)
  90       return "\n\n".join(parts)
  91
  92    def mkd2html(self):
  93       from markdown import markdown
  94       return markdown(self.body_mkd(), output_format="html")
  95
  96    def img_thumbnail_800_1024(self, img_name):
  97       return """\
  98 <img src="%(img_name)s-thumbnail.jpg" alt="%(img_name)s-thumbnail.jpg" /><br />
  99 <a href="%(img_name)s-800x600.jpg">800x600</a>, <a href="%(img_name)s-1024x800.jpg">1024x800</a>""" % {"img_name": img_name}
 100
 101    def wikipedia(self, query):
 102       return "http://en.wikipedia.org/wiki/%s" % quote_string(query.replace(' ', '_'), ext_safe=',')
 103
 104    def wikipedia_ru(self, query):
 105       return "http://ru.wikipedia.org/wiki/%s" % quote_string(query.replace(' ', '_'), ext_safe=',')
 106
 107    def nigma(self, query):
 108        return "http://www.nigma.ru/index.php?s=%s" % quote_string(query)
 109
 110    search = nigma
 111
 112    def yandex(self, query):
 113       return "http://www.yandex.ru/yandsearch?text=%s&rpt=rad" % quote_string(query, "cp1251")
 114
 115    def google(self, query):
 116       return "http://www.google.com/search?hl=en&ie=utf-8&oe=utf-8&q=%s" % quote_string(query)
 117
 118    def google_ru(self, query):
 119       return "http://www.google.ru/search?hl=ru&ie=utf-8&oe=utf-8&q=%s" % quote_string(query)
 120
 121 def quote_string(s, to_encoding="utf-8", ext_safe=''):
 122    return urllib.quote(unicode(s, "utf-8").encode(to_encoding), '/' + ext_safe)