reindex_blog.py

   1 #! /usr/local/bin/python -O
   2 # -*- coding: koi8-r -*-
   3
   4 __version__ = "$Revision$"[11:-2]
   5 __revision__ = "$Id$"[5:-2]
   6 __date__ = "$Date$"[7:-2]
   7 __author__ = "Oleg BroytMann <phd@phd.pp.ru>"
   8 __copyright__ = "Copyright (C) 2006 PhiloSoft Design"
   9
  10
  11 import sys, os
  12
  13 blog_data_root = sys.argv[1]
  14 blog_root = sys.argv[2]
  15 blog_filename = os.path.join(blog_data_root, "blog_dict.pickle")
  16
  17 try:
  18    import cPickle as pickle
  19 except ImportError:
  20    import pickle
  21
  22 from Cheetah.Template import Template
  23
  24
  25 # Load old blog
  26
  27 try:
  28    blog_file = open(blog_filename, "rb")
  29 except IOError:
  30    old_blog = {}
  31 else:
  32    old_blog = pickle.load(blog_file)
  33    blog_file.close()
  34
  35
  36 # blog is a dictionary mapping
  37 # (year, month, day) => [list of (file, title, lead, tags)]
  38
  39 blog = {}
  40 years = {}
  41
  42 # Walk the directory recursively
  43 for dirpath, dirs, files in os.walk(blog_root):
  44    d = os.path.basename(dirpath)
  45    if not d.startswith("20") and not d.isdigit():
  46       continue
  47    for file in files:
  48       if not file.endswith(".tmpl"):
  49          continue
  50       fullpath = os.path.join(dirpath, file)
  51       template = Template(file=fullpath)
  52       title_parts = template.Title.split()
  53       title = ' '.join(title_parts[6:])
  54       lead = getattr(template, "Lead", None)
  55
  56       tags = template.Tag
  57       if isinstance(tags, basestring):
  58          tags = (tags,)
  59
  60       if title:
  61          key = year, month, day = tuple(dirpath[len(blog_root):].split(os.sep)[1:])
  62          if key in blog:
  63             days = blog[key]
  64          else:
  65             days = blog[key] = []
  66          days.append((file, title, lead, tags))
  67
  68          if year in years:
  69             months = years[year]
  70          else:
  71             months = years[year] = {}
  72
  73          if month in months:
  74             days = months[month]
  75          else:
  76             days = months[month] = []
  77
  78          if day not in days: days.append(day)
  79
  80
  81 # Need to save the blog?
  82 if blog <> old_blog:
  83    blog_file = open(blog_filename, "wb")
  84    pickle.dump(blog, blog_file, pickle.HIGHEST_PROTOCOL)
  85    blog_file.close()
  86
  87
  88 # Localized month names
  89
  90 import locale
  91 locale.setlocale(locale.LC_ALL, '')
  92 from calendar import _localized_day, _localized_month
  93
  94 locale.setlocale(locale.LC_TIME, 'C')
  95 months_names_en = list(_localized_month('%B'))
  96 months_abbrs_en = list(_localized_month('%b'))
  97
  98 locale.setlocale(locale.LC_TIME, '')
  99 months_names_ru = [month.lower() for month in _localized_month('%B')]
 100
 101 months_names_ru0 = ['', "январь", "февраль", "март", "апрель", "май", "июнь",
 102    "июль", "август", "сентябрь", "октябрь", "ноябрь", "декабрь"
 103 ]
 104
 105 from news import write_if_changed
 106
 107
 108 def write_template(level, year, month, day, titles, tags=None):
 109    path = [blog_root]
 110    if level >= 1:
 111       path.append(year)
 112    if level >= 2:
 113       path.append(month)
 114    if level == 3:
 115       path.append(day)
 116    path.append("index.tmpl")
 117    index_name = os.path.join(*path)
 118
 119    new_text = ["""\
 120 ## THIS FILE IS AUTOMATICALLY GENERATED. DO NOT EDIT.
 121 #extends phd_pp_ru
 122 #implements respond
 123 """]
 124
 125    if level == 0:
 126       new_text.append("""\
 127 #attr $Title = "Oleg BroytMann's blog"
 128 #attr $Description = "BroytMann Russian Blog Index Document"
 129 #attr $Copyright = %(cyear)s
 130 #attr $alternates = (("application/atom+xml", "News [Atom 1.0]", "atom_10.xml"),
 131                      ("application/rss+xml",  "News [RSS 2.0]",  "rss_20.xml")
 132 )
 133 ##
 134 #def body_html
 135 <h1>Журнал</h1>
 136 """ % {"cyear": year or 2005})
 137
 138    elif level == 1:
 139       new_text.append("""\
 140 #attr $Title = "Oleg BroytMann's blog: %(year)s"
 141 #attr $Description = "BroytMann Russian Blog %(year)s Index Document"
 142 #attr $Copyright = %(cyear)s
 143 ##
 144 #def body_html
 145 <h1>Журнал: %(year)s</h1>
 146 """ % {"year": year, "cyear": year or 2005})
 147
 148    elif level == 2:
 149       imonth = int(month)
 150       new_text.append("""\
 151 #attr $Title = "Oleg BroytMann's blog: %(month_abbr_en)s %(year)s"
 152 #attr $Description = "BroytMann Russian Blog %(month_name_en)s %(year)s Index Document"
 153 #attr $Copyright = %(cyear)s
 154 ##
 155 #def body_html
 156 <h1>Журнал: %(month_name_ru0)s %(year)s</h1>
 157 """ % {
 158       "year": year, "cyear": year or 2005,
 159       "month_abbr_en": months_abbrs_en[imonth], "month_name_en": months_names_en[imonth],
 160       "month_name_ru0": months_names_ru0[imonth],
 161    })
 162
 163    elif level == 3:
 164       iday = int(day)
 165       imonth = int(month)
 166
 167       new_text.append("""\
 168 #attr $Next = "%s"
 169 """ % titles[0][3])
 170
 171
 172       if len(titles) == 1:
 173          new_text.append("""\
 174 #attr $refresh = "0; URL=%s"
 175 """ % titles[0][3])
 176
 177       new_text.append("""\
 178 #attr $Title = "Oleg BroytMann's blog: %(day)d %(month_abbr_en)s %(year)s"
 179 #attr $Description = "BroytMann Russian Blog %(day)d %(month_name_en)s %(year)s Index Document"
 180 #attr $Copyright = %(cyear)s
 181 ##
 182 #def body_html
 183 <h1>Журнал: %(day)d %(month_name_ru)s %(year)s</h1>
 184 """ % {
 185       "year": year, "cyear": year or 2005,
 186       "month_abbr_en": months_abbrs_en[imonth], "month_name_en": months_names_en[imonth],
 187       "month_name_ru": months_names_ru[imonth],
 188       "day": iday
 189    })
 190
 191    save_titles = titles[:]
 192    titles.reverse()
 193
 194    save_day = None
 195    for year, month, day, file, title, lead in titles:
 196       href = []
 197       if level == 0:
 198          href.append(year)
 199       if level <= 1:
 200          href.append(month)
 201       if level <= 2:
 202          href.append(day)
 203       href.append(file)
 204       href = '/'.join(href)
 205       if day[0] == '0': day = day[1:]
 206       if save_day <> day:
 207          if level == 0:
 208             new_text.append('\n<h2>%s %s %s</h2>' % (day, months_names_ru[int(month)], year))
 209          else:
 210             new_text.append('\n<h2>%s %s</h2>' % (day, months_names_ru[int(month)]))
 211          save_day = day
 212       if lead:
 213          lead = lead + ' '
 214       else:
 215          lead = ''
 216       new_text.append('''
 217 <p class="head">
 218    %s<a href="%s">%s</a>.
 219 </p>
 220 ''' % (lead, href, title))
 221
 222    if level == 0:
 223       new_text.append("""
 224 <hr>
 225
 226 <p class="head">Новостевая лента в форматах
 227 <A HREF="atom_10.xml">Atom 1.0 <img src="../../Graphics/atom_10.jpg" border=0></A>
 228 и <A HREF="rss_20.xml">RSS 2.0 <img src="../../Graphics/rss_20.jpg" border=0></A>.
 229 </p>
 230 """)
 231
 232       years = {}
 233       for year, month, day, file, title, lead in save_titles:
 234          years[year] = True
 235       new_text.append('''
 236 <p class="head"><a href="tags/">Теги</a>:
 237 ''')
 238       first_tag = True
 239       for count, tag, links in all_tags:
 240          if first_tag:
 241             first_tag = False
 242          else:
 243             new_text.append(' - ')
 244          new_text.append("""<a href="tags/%s.html">%s (%d)</a>""" % (tag, tag, count))
 245       new_text.append('''
 246 </p>
 247 ''')
 248
 249       new_text.append('''
 250 <p class="head">По годам:
 251 ''')
 252       first_year = True
 253       for year in sorted(years.keys()):
 254          if first_year:
 255             first_year = False
 256          else:
 257             new_text.append(' - ')
 258          new_text.append('<a href="%s/">%s</a>' % (year, year))
 259       new_text.append('''
 260 </p>
 261 ''')
 262
 263       new_text.append("""
 264 <hr>
 265 <p class="head"><a href="http://phd.livejournal.com/">ЖЖ</a>
 266 """)
 267
 268    new_text.append("""\
 269 #end def
 270 $phd_pp_ru.respond(self)
 271 """)
 272
 273    write_if_changed(index_name, ''.join(new_text))
 274
 275
 276 all_tags = {}
 277 all_titles = []
 278 all_titles_tags = []
 279
 280 for year in sorted(years.keys()):
 281    year_titles = []
 282    months = years[year]
 283    for month in sorted(months.keys()):
 284       month_titles = []
 285       for day in sorted(months[month]):
 286          day_titles = []
 287          key = year, month, day
 288          if key in blog:
 289             for file, title, lead, tags in blog[key]:
 290                if file.endswith(".tmpl"): file = file[:-len("tmpl")] + "html"
 291                value = (year, month, day, file, title, lead)
 292                all_titles_tags.append((year, month, day, file, title, lead, tags))
 293                all_titles.append(value)
 294                year_titles.append(value)
 295                month_titles.append(value)
 296                day_titles.append(value)
 297                for tag in tags:
 298                   if tag in all_tags:
 299                      tag_links = all_tags[tag]
 300                   else:
 301                      tag_links = all_tags[tag] = []
 302                   tag_links.append(value)
 303          write_template(3, year, month, day, day_titles)
 304       write_template(2, year, month, day, month_titles)
 305    write_template(1, year, month, day, year_titles)
 306
 307 def by_count_rev_tag_link(t1, t2):
 308    """Sort all_tags by count in descending order,
 309    and by tags and links in ascending order
 310    """
 311    r = cmp(t1[0], t2[0])
 312    if r:
 313       return -r
 314    return cmp((t1[1], t1[2]), (t2[1], t2[2]))
 315
 316 all_tags = [(len(links), tag, links) for (tag, links) in all_tags.items()]
 317 all_tags.sort(by_count_rev_tag_link)
 318
 319 write_template(0, year, month, day, all_titles[-20:], all_tags)
 320
 321 new_text = ["""\
 322 ## THIS FILE IS AUTOMATICALLY GENERATED. DO NOT EDIT.
 323 #extends phd_pp_ru
 324 #implements respond
 325 #attr $Title = "Oleg BroytMann's blog: tags"
 326 #attr $Description = "BroytMann Russian Blog Tags Index Document"
 327 #attr $Copyright = 2006
 328 ##
 329 #def body_html
 330 <h1>Теги</h1>
 331
 332 <p class="head">
 333 <dl>
 334 """]
 335
 336 for count, tag, links in all_tags:
 337    new_text.append("""\
 338    <dt><a href="%s.html">%s (%d)</a></dt>
 339 """ % (tag, tag, count))
 340
 341    tag_text = ["""\
 342 ## THIS FILE IS AUTOMATICALLY GENERATED. DO NOT EDIT.
 343 #extends phd_pp_ru
 344 #implements respond
 345 #attr $Title = "Oleg BroytMann's blog: tag %s"
 346 #attr $Description = "BroytMann Russian Blog Tag %s Index Document"
 347 #attr $Copyright = 2006
 348 ##
 349 #def body_html
 350 <h1>%s</h1>
 351
 352 <p class="head">
 353 <ul>
 354 """ % (tag, tag, tag)]
 355
 356    count = 0
 357    for year, month, day, filename, title, lead in reversed(links):
 358       if lead:
 359          lead = lead + ' '
 360       else:
 361          lead = ''
 362       link = "../%s/%s/%s/%s" % (year, month, day, filename)
 363       item_text = """<li><a href="%s">%s/%s/%s: %s%s</a></li>""" % (link, year, month, day, lead, title)
 364
 365       count += 1
 366       if count <= 5:
 367          new_text.append("      <dd>%s</dd>\n" % item_text)
 368
 369       tag_text.append("   %s\n" % item_text)
 370
 371    tag_text.append("""\
 372 </ul>
 373 </p>
 374 #end def
 375 $phd_pp_ru.respond(self)
 376 """)
 377    write_if_changed(os.path.join(blog_root, "tags", tag+".tmpl"), ''.join(tag_text))
 378
 379 new_text.append("""\
 380 </dl>
 381 </p>
 382 #end def
 383 $phd_pp_ru.respond(self)
 384 """)
 385 write_if_changed(os.path.join(blog_root, "tags", "index.tmpl"), ''.join(new_text))
 386
 387
 388 from atom_10 import atom_10
 389 from rss_20 import rss_20
 390 from news import NewsItem
 391
 392 if blog_root:
 393    baseURL = "http://phd.pp.ru/%s/" % blog_root
 394 else:
 395    baseURL = "http://phd.pp.ru/"
 396
 397 items = []
 398 for item in tuple(reversed(all_titles_tags))[:10]:
 399    year, month, day, file, title, lead, tags = item
 400    if lead:
 401       lead = lead + ' '
 402    else:
 403       lead = ''
 404    item = NewsItem(
 405       "%s-%s-%s" % (year, month, day),
 406       "%s%s" % (lead, title),
 407       "%s/%s/%s/%s" % (year, month, day, file)
 408    )
 409    items.append(item)
 410    item.baseURL = baseURL
 411    item.categoryList = tags
 412
 413 namespace = {
 414    "title": "Oleg Broytmann's blog",
 415    "baseURL": baseURL,
 416    "indexFile": "",
 417    "description": "",
 418    "lang": "ru",
 419    "author": "Oleg Broytmann",
 420    "email": "phd@phd.pp.ru",
 421    "generator": os.path.basename(sys.argv[0]),
 422    "posts": items,
 423 }
 424
 425 # For english dates
 426 locale.setlocale(locale.LC_TIME, 'C')
 427
 428 atom_tmpl = str(atom_10(searchList=[namespace]))
 429 write_if_changed(os.path.join(blog_root, "atom_10.xml"), atom_tmpl)
 430 rss_tmpl = str(rss_20(searchList=[namespace]))
 431 write_if_changed(os.path.join(blog_root, "rss_20.xml"), rss_tmpl)