]> git.phdru.name Git - phdru.name/phdru.name.git/blobdiff - reindex_blog.py
Blog feeds.
[phdru.name/phdru.name.git] / reindex_blog.py
index 4d0ad0a052e3b45abd34255e22c65483d28c5aab..74717438d9e99d5629ae04266b10db48be40a595 100755 (executable)
@@ -10,6 +10,9 @@ __copyright__ = "Copyright (C) 2006 PhiloSoft Design"
 
 import sys, os
 
+blog_filename = sys.argv[1]
+blog_root = sys.argv[2]
+
 try:
    import cPickle as pickle
 except ImportError:
@@ -20,7 +23,6 @@ from Cheetah.Template import Template
 
 # Load old blog
 
-blog_filename = sys.argv[1]
 try:
    blog_file = open(blog_filename, "rb")
 except IOError:
@@ -31,13 +33,16 @@ else:
 
 
 # blog is a dictionary mapping
-# (year, month, day) => [list of (file, title, lead)]
+# (year, month, day) => [list of (file, title, lead, tags)]
 
 blog = {}
 years = {}
 
 # Walk the directory recursively
-for dirpath, dirs, files in os.walk(os.curdir):
+for dirpath, dirs, files in os.walk(blog_root):
+   d = os.path.basename(dirpath)
+   if not d.startswith("20") and not d.isdigit():
+      continue
    for file in files:
       # Ignore index.tmpl and *.html files; supose all other files are *.tmpl
       if file == "index.tmpl" or file.endswith(".html"):
@@ -48,13 +53,17 @@ for dirpath, dirs, files in os.walk(os.curdir):
       title = ' '.join(title_parts[6:])
       lead = getattr(template, "Lead", None)
 
+      tags = getattr(template, "Tag", None)
+      if isinstance(tags, basestring):
+         tags = (tags,)
+
       if title:
-         key = year, month, day = tuple(dirpath.split(os.sep)[1:])
+         key = year, month, day = tuple(dirpath[len(blog_root):].split(os.sep)[1:])
          if key in blog:
             days = blog[key]
          else:
             days = blog[key] = []
-         days.append((file, title, lead))
+         days.append((file, title, lead, tags))
 
          if year in years:
             months = years[year]
@@ -93,9 +102,11 @@ months_names_ru0 = ['', "
    "июль", "август", "сентябрь", "октябрь", "ноябрь", "декабрь"
 ]
 
+from news import write_if_changed
 
-def write_template(level, year, month, day, titles):
-   path = []
+
+def write_template(level, year, month, day, titles, tags=None):
+   path = [blog_root]
    if level >= 1:
       path.append(year)
    if level >= 2:
@@ -104,12 +115,6 @@ def write_template(level, year, month, day, titles):
       path.append(day)
    path.append("index.tmpl")
    index_name = os.path.join(*path)
-   try:
-      index_tmpl = open(index_name, 'r')
-      old_text = index_tmpl.read()
-      index_tmpl.close()
-   except IOError:
-      old_text = None
 
    new_text = ["""\
 ## THIS FILE IS AUTOMATICALLY GENERATED. DO NOT EDIT.
@@ -215,12 +220,26 @@ def write_template(level, year, month, day, titles):
       years = {}
       for year, month, day, file, title, lead in save_titles:
          years[year] = True
-      first_year = True
       new_text.append('''
 <hr>
 
-<p class="years">
+<p class="noindent"><a href="tags/">Теги</a>:
+''')
+      first_tag = True
+      for count, tag, links in all_tags:
+         if first_tag:
+            first_tag = False
+         else:
+            new_text.append(' - ')
+         new_text.append("""<a href="tags/%s.html">%s (%d)</a>""" % (tag, tag, count))
+      new_text.append('''
+</p>
+''')
+
+      new_text.append('''
+<p class="noindent">По годам:
 ''')
+      first_year = True
       for year in sorted(years.keys()):
          if first_year:
             first_year = False
@@ -236,15 +255,13 @@ def write_template(level, year, month, day, titles):
 $phd_pp_ru.respond(self)
 """)
 
-   new_text = ''.join(new_text)
-   if old_text <> new_text:
-      print "Writing", index_name
-      index_tmpl = open(index_name, 'w')
-      index_tmpl.write(new_text)
-      index_tmpl.close()
+   write_if_changed(index_name, ''.join(new_text))
 
 
+all_tags = {}
 all_titles = []
+all_titles_tags = []
+
 for year in sorted(years.keys()):
    year_titles = []
    months = years[year]
@@ -254,14 +271,130 @@ for year in sorted(years.keys()):
          day_titles = []
          key = year, month, day
          if key in blog:
-            for file, title, lead in blog[key]:
+            for file, title, lead, tags in blog[key]:
                if file.endswith(".tmpl"): file = file[:-len("tmpl")] + "html"
                value = (year, month, day, file, title, lead)
+               all_titles_tags.append((year, month, day, file, title, lead, tags))
                all_titles.append(value)
                year_titles.append(value)
                month_titles.append(value)
                day_titles.append(value)
+               for tag in tags:
+                  if tag in all_tags:
+                     tag_links = all_tags[tag]
+                  else:
+                     tag_links = all_tags[tag] = []
+                  tag_links.append(value)
          write_template(3, year, month, day, day_titles)
       write_template(2, year, month, day, month_titles)
    write_template(1, year, month, day, year_titles)
-write_template(0, year, month, day, all_titles[-20:])
+
+all_tags = [(len(links), tag, links) for (tag, links) in all_tags.items()]
+all_tags.sort()
+
+write_template(0, year, month, day, all_titles[-20:], all_tags)
+
+new_text = ["""\
+## THIS FILE IS AUTOMATICALLY GENERATED. DO NOT EDIT.
+#extends phd_pp_ru
+#implements respond
+#attr $Title = "Oleg BroytMann's blog: tags"
+#attr $Description = "BroytMann Russian Blog Tags Index Document"
+#attr $Copyright = 2006
+##
+#def body_html
+<H1>Теги</H1>
+
+<p class="head">
+<dl>
+"""]
+
+for count, tag, links in all_tags:
+   new_text.append("""\
+   <dt><a href="%s.html">%s (%d)</a></dt>
+""" % (tag, tag, count))
+
+   tag_text = ["""\
+## THIS FILE IS AUTOMATICALLY GENERATED. DO NOT EDIT.
+#extends phd_pp_ru
+#implements respond
+#attr $Title = "Oleg BroytMann's blog: tag %s"
+#attr $Description = "BroytMann Russian Blog Tag %s Index Document"
+#attr $Copyright = 2006
+##
+#def body_html
+<H1>%s</H1>
+
+<p class="head">
+<ul>
+""" % (tag, tag, tag)]
+
+   count = 0
+   for year, month, day, filename, title, lead in reversed(links):
+      if lead:
+         lead = lead + ' '
+      else:
+         lead = ''
+      link = "../%s/%s/%s/%s" % (year, month, day, filename)
+      item_text = """<li><a href="%s">%s/%s/%s: %s%s</a></li>""" % (link, year, month, day, lead, title)
+
+      count += 1
+      if count <= 5:
+         new_text.append("      <dd>%s</dd>\n" % item_text)
+
+      tag_text.append("   %s\n" % item_text)
+
+   tag_text.append("""\
+</ul>
+</p>
+#end def
+$phd_pp_ru.respond(self)
+""")
+   write_if_changed(os.path.join(blog_root, "tags", tag+".tmpl"), ''.join(tag_text))
+
+new_text.append("""\
+</dl>
+</p>
+#end def
+$phd_pp_ru.respond(self)
+""")
+write_if_changed(os.path.join(blog_root, "tags", "index.tmpl"), ''.join(new_text))
+
+
+from atom_10 import atom_10
+from rss_20 import rss_20
+from news import NewsItem
+
+baseURL = "http://phd.pp.ru/Russian/blog/"
+
+items = []
+for item in tuple(reversed(all_titles_tags))[:10]:
+   year, month, day, file, title, lead, tags = item
+   if lead:
+      lead = lead + ' '
+   else:
+      lead = ''
+   item = NewsItem(
+      "%s-%s-%s" % (year, month, day),
+      "%s%s" % (lead, title),
+      "%s/%s/%s/%s" % (year, month, day, file)
+   )
+   items.append(item)
+   item.baseURL = baseURL
+   item.categoryList = tags
+
+namespace = {
+   "title": "Oleg Broytmann's blog",
+   "baseURL": baseURL,
+   "indexFile": "",
+   "description": "",
+   "lang": "ru",
+   "author": "Oleg Broytmann",
+   "email": "phd@phd.pp.ru",
+   "posts": items,
+}
+
+atom_tmpl = str(atom_10(searchList=[namespace]))
+write_if_changed(os.path.join(blog_root, "atom_10.xml"), atom_tmpl)
+rss_tmpl = str(rss_20(searchList=[namespace]))
+write_if_changed(os.path.join(blog_root, "rss_20.xml"), rss_tmpl)