]> git.phdru.name Git - phdru.name/phdru.name.git/blobdiff - phd.py
Fix(phd.py): Change URL for Lurk
[phdru.name/phdru.name.git] / phd.py
diff --git a/phd.py b/phd.py
index e12fb902cb88599af9dfc49a13a361df6c4d9f0d..4aa206c47fc1637a83cc65b1b6955a9efbc1fbdd 100644 (file)
--- a/phd.py
+++ b/phd.py
-import os, re, time, urllib
+import os, re, time
+try:
+    from urllib import quote as url_quote
+except ImportError:
+    from urllib.parse import quote as url_quote
 from Cheetah.Template import Template
+from Cheetah.compat import PY2
+from blog_db import load_blog
 
 
 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/]"
 
+
 def _url2href(match):
-   url = match.group(0)
-   return '<a href="%s">%s</a>' % (url, url)
+    url = match.group(0)
+    return '<a href="%s">%s</a>' % (url, url)
 
 
 full_dirs = len(os.getcwd().split('/')) + 1
 
+
 class phd(Template):
-   def __init__(self, *args, **kw):
-      if not hasattr(self, "_fileBaseName"):
-         self._fileDirName, self._fileBaseName = os.path.split(os.path.abspath(self._CHEETAH_src))
-      Template.__init__(self, *args, **kw)
-      directories = self._fileDirName.split('/')[full_dirs:] # remove directories up to "./files"
-      dirs_to_root = len(directories)
-      if dirs_to_root:
-         root = "../"*dirs_to_root
-      else:
-         root = ''
-      self.root = root
-      path = '/'.join(directories) + '/' + \
-         self._fileBaseName.replace(".tmpl", ".html")
-      if path[0] <> '/': path = '/' + path
-      self.path = path
-
-   def copyright(self, start_year):
-      this_year = time.localtime()[0]
-      if start_year >= this_year:
-         return this_year
-      if start_year == this_year - 1:
-         return "%s, %s" % (start_year, this_year)
-      return "%s-%s" % (start_year, this_year)
-
-
-   def body(self):
-      if hasattr(self, "body_html"):
-         return self.body_html()
-      if hasattr(self, "body_text"):
-         return self.text2html()
-      if hasattr(self, "body_rst"):
-         return self.rst2html()
-
-   def text2html(self):
-      body = re.sub(url_re, _url2href, self.body_text())
-
-      paragraphs = body.split("\n\n")
-
-      new_paras = []
-      for p in paragraphs:
-         parts = p.split("\n   ")
-         parts[0] = parts[0].strip()
-         new_paras.append('\n</p>\n<p>\n'.join(parts))
-
-      if self.Title:
-         title = "<h1>%s</h1>\n\n" % self.Title
-      else:
-         title = ''
-
-      body = '\n</p>\n\n<p class="head">\n'.join(new_paras)
-      return "%s<p>%s</p>" % (title, body)
-
-   def rst2html(self):
-      from docutils.core import publish_parts
-      from m_lib.defenc import default_encoding as encoding
-
-      parts = publish_parts(self.body_rst(), writer_name="html")
-
-      title = parts["title"].encode(encoding) or self.Title
-      if title:
-         title = "<h1>%s</h1>" % title
-
-      subtitle = parts["subtitle"].encode(encoding)
-      if subtitle:
-         subtitle = "<h2>%s</h2>" % subtitle
-
-      body = parts["body"].encode(encoding)
-      parts = [part for part in (title, subtitle, body) if part]
-      return "\n\n".join(parts)
-
-
-   def img_thumbnail_800_1024(self, img_name):
-      return """\
+    def __init__(self, *args, **kw):
+        if not hasattr(self, "_fileBaseName"):
+            self._fileDirName, self._fileBaseName = \
+                os.path.split(os.path.abspath(self._CHEETAH_src))
+        Template.__init__(self, *args, **kw)
+        # remove directories up to "./files"
+        directories = self._fileDirName.split('/')[full_dirs:]
+        dirs_to_root = len(directories)
+        if dirs_to_root:
+            root = "../"*dirs_to_root
+        else:
+            root = ''
+        self.root = root
+        path = '/'.join(directories) + '/' + \
+            self._fileBaseName.replace(".tmpl", ".html")
+        if path[0] != '/': path = '/' + path
+        self.path = path
+
+    def copyright(self, start_year):
+        this_year = time.localtime()[0]
+        if start_year >= this_year:
+            return this_year
+        if start_year == this_year - 1:
+            return "%s, %s" % (start_year, this_year)
+        return "%s-%s" % (start_year, this_year)
+
+    def body(self):
+        if hasattr(self, "body_html"):
+            return self.body_html()
+        if hasattr(self, "body_text"):
+            return self.text2html()
+        if hasattr(self, "body_rst"):
+            return self.rst2html()
+        if hasattr(self, "body_mkd"):
+            return self.mkd2html()
+
+    def text2html(self):
+        body = re.sub(url_re, _url2href, self.body_text())
+
+        paragraphs = body.split("\n\n")
+
+        new_paras = []
+        for p in paragraphs:
+            if PY2 and not isinstance(p, bytes):
+                p = p.encode('utf-8')
+            parts = p.split("\n  ")
+            parts[0] = parts[0].strip()
+            new_paras.append('\n</p>\n<p>\n'.join(parts))
+
+        if self.Title:
+            title = "<h1>%s</h1>\n\n" % self.Title
+        else:
+            title = ''
+
+        body = '\n</p>\n\n<p class="head">\n'.join(new_paras)
+        return "%s<p>%s</p>" % (title, body)
+
+    def rst2html(self):
+        from docutils.core import publish_parts
+
+        parts = publish_parts(self.body_rst(), writer_name="html")
+
+        title = parts["title"] or self.Title
+        if title:
+            title = "<h1>%s</h1>" % title
+
+        subtitle = parts["subtitle"]
+        if subtitle:
+            subtitle = "<h2>%s</h2>" % subtitle
+
+        body = parts["body"]
+        parts = []
+        for part in (title, subtitle, body):
+            if not part:
+                continue
+            if PY2 and not isinstance(part, bytes):
+                part = part.encode('utf-8')
+            parts.append(part)
+        return "\n\n".join(parts)
+
+    def mkd2html(self):
+        from markdown import markdown
+        return markdown(self.body_mkd(), output_format="html")
+
+    def find_near_blog_posts(self):
+        if not self.path.startswith("/Russian/blog/"):
+            return None, None, None, None
+        dirs = self.path.split('/')
+        # blog post is ['', 'Russian', 'blog', year, month, day, filename]
+        if len(dirs) != 7:  # Not a blog post
+            return None, None, None, None
+        ymd = tuple(dirs[3:6])
+        filename = self._fileBaseName
+
+        prev_key = prev_blog_post = current_key = current_blog_post = \
+            next_key = next_blog_post = None
+        blog = load_blog()
+        for key in sorted(blog):
+            for blog_post in sorted(blog[key]):
+                if current_blog_post:
+                    prev_key = current_key
+                    prev_blog_post = current_blog_post
+                if next_blog_post:
+                    current_key = next_key
+                    current_blog_post = next_blog_post
+                next_key = key
+                next_blog_post = blog_post
+                if current_blog_post and (current_key == ymd) and \
+                        (current_blog_post[0] == filename):  # Found!
+                    return prev_key, prev_blog_post, next_key, next_blog_post
+        return current_key, current_blog_post, None, None
+
+    def img_thumbnail_800_1024(self, img_name):
+        return """\
 <img src="%(img_name)s-thumbnail.jpg" alt="%(img_name)s-thumbnail.jpg" /><br />
 <a href="%(img_name)s-800x600.jpg">800x600</a>, <a href="%(img_name)s-1024x800.jpg">1024x800</a>""" % {"img_name": img_name}
 
-   def wikipedia(self, query):
-      return "http://en.wikipedia.org/wiki/%s" % quote_string(query.replace(' ', '_'), ext_safe=',')
+    def wikipedia(self, query):
+        return "https://en.wikipedia.org/wiki/%s" % quote_string(
+            query.replace(' ', '_'), ext_safe=',')
+
+    def wikipedia_ru(self, query):
+        return "https://ru.wikipedia.org/wiki/%s" % quote_string(
+            query.replace(' ', '_'), ext_safe=',')
+
+    def startpage(self, query):
+        return "https://startpage.com/do/search?q=%s" % quote_string(query)
+
+    search = startpage
 
-   def wikipedia_ru(self, query):
-      return "http://ru.wikipedia.org/wiki/%s" % quote_string(query.replace(' ', '_'), ext_safe=',')
+    def nigma(self, query):
+        return "http://www.nigma.ru/index.php?s=%s" % quote_string(query)
 
-   def nigma(self, query):
-       return "http://www.nigma.ru/index.php?s=%s" % quote_string(query)
+    def yandex(self, query):
+        return "http://www.yandex.ru/yandsearch?text=%s&rpt=rad" \
+            % quote_string(query, "cp1251")
 
-   search = nigma
+    def google(self, query):
+        return "http://www.google.com/search?hl=en&ie=utf-8&oe=utf-8&q=%s" \
+            % quote_string(query)
 
-   def yandex(self, query):
-      return "http://www.yandex.ru/yandsearch?text=%s&rpt=rad" % quote_string(query, "cp1251")
+    def google_ru(self, query):
+        return "http://www.google.ru/search?hl=ru&ie=utf-8&oe=utf-8&q=%s" \
+            % quote_string(query)
 
-   def google(self, query):
-      return "http://www.google.com/search?hl=en&ie=utf-8&oe=utf-8&q=%s" % quote_string(query)
+    def lurkmoar(self, query):
+        return "http://lurklurk.com/%s" % quote_string(
+            query.replace(' ', '_'), ext_safe=',')
 
-   def google_ru(self, query):
-      return "http://www.google.ru/search?hl=ru&ie=utf-8&oe=utf-8&q=%s" % quote_string(query)
 
 def quote_string(s, to_encoding="utf-8", ext_safe=''):
-   return urllib.quote(unicode(s, "koi8-r").encode(to_encoding), '/' + ext_safe)
+    if isinstance(s, bytes):
+        s = s.decode("utf-8")
+        if PY2:
+            s = s.encode(to_encoding)
+    return url_quote(s, '/' + ext_safe)