wrigit.py

   1 #!/usr/bin/env python
   2 # -*- mode: python -*-
   3
   4 import os, time, datetime
   5 from os.path   import join, abspath
   6
   7 import rst
   8 from cfg  import o, setup_cfg
   9 cfg = setup_cfg([
  10         o('-c', '--conf', dest='conf',
  11           help='Configuration file'),
  12         o('-r', '--rstdir', dest='rstdir',
  13           help='Directory where source .rst files are found'),
  14         o('-t', '--templatedir', dest='templatedir',
  15           help='Directory where mako templates can be found'),
  16         o('-b', '--baseurl', dest='baseurl',
  17           help='The base URL of the site'),
  18         o('-T', '--title', dest='title',
  19           help='Site title that goes into <title>'),
  20         o('', '--regen', dest='regen', action="store_true",
  21           help='Regenerate the whole site (all files)'),
  22         o('-d', '--disqus-forum', dest='disqus_forum',
  23           help='The name of the disqus forum. For example, ' +
  24           'if http://foo.disqus.com is the forum, then do --disqus-forum=foo'),
  25         o('', '--blog-title', dest='blog.title',
  26           help='Blog title'),
  27         o('', '--blog-subtitle', dest='blog.subtitle',
  28           help='Blog title'),
  29         o('', '--blog-feedburner-url', dest='blog.feedburner_url',
  30           help='FeedBurner URL to use instead of local atom.xml'),
  31         ], {'regen': False})
  32
  33 cfg['rstdir'] = abspath(cfg['rstdir'])
  34 cfg['templatedir'] = abspath(cfg['templatedir'])
  35
  36 from util import make_mako, last_modified, rfc3339
  37
  38
  39 class Page(object):
  40     """A rst page"""
  41
  42     def __init__(self, rstfile):
  43         parts, meta = rst.process(rstfile)
  44
  45         self.rstfile   = rstfile
  46         self.title     = parts['title']
  47         self.fragment  = parts['fragment']
  48         self.meta      = meta
  49
  50     def title_html(self):
  51         return '%s - %s' % (self.title, cfg['title'])
  52
  53     def date(self):
  54         if '|' in self.meta['date']:
  55             format = '%Y-%m-%d|%H:%M'
  56         else:
  57             format = '%Y-%m-%d'
  58
  59         return datetime.datetime(
  60             *time.strptime(self.meta['date'],
  61                            format)[0:6])
  62
  63     def updated_date(self):
  64         return datetime.datetime.fromtimestamp(
  65             last_modified(self.rstfile))
  66
  67     def tags(self):
  68         tags = self.meta.get('tags', '').split()
  69         tags.sort()
  70         return tags
  71
  72     def url(self):
  73         path_info = rst.htmlpath(self.rstfile[len(cfg['rstdir']):])
  74         return cfg['baseurl'] + path_info
  75
  76     def commentable(self):
  77         return False
  78
  79 class BlogPost(Page):
  80
  81     def title_html(self):
  82         return self.title
  83
  84     def commentable(self):
  85         return True
  86
  87     def __cmp__(self, other):
  88         return cmp(self.date(), other.date())
  89
  90     def __repr__(self):
  91         return '<BlogPost "%s" on %s>' % (self.title, self.date())
  92
  93
  94 if __name__ == '__main__':
  95     # Import Psyco if available
  96     try:
  97         import psyco
  98         psyco.full()
  99     except ImportError:
 100         pass
 101
 102     print '[%s - %s] rstdir:%s [using templatedir:%s]' % (
 103         cfg['baseurl'], cfg['title'], cfg['rstdir'], cfg['templatedir'])
 104
 105     rstdir = cfg['rstdir']
 106     blog_posts = []
 107     blog_path = join(rstdir, 'blog')
 108     feed_url = cfg['blog.feedburner_url'] or cfg['baseurl'] + '/blom/atom.xml'
 109
 110     for f in rst.rstfiles(rstdir):
 111         if blog_path in f:
 112             page = BlogPost(f)
 113             blog_posts.append(page)
 114         else:
 115             page = Page(f)
 116
 117         make_mako(rst.htmlpath(f),
 118                   [f],
 119                   'page.mako')
 120
 121     blog_posts.sort()
 122     blog_posts.reverse()
 123
 124     # generate blog index
 125     make_mako(join(rstdir, 'blog', 'index.html'),
 126               [p.rstfile for p in blog_posts],
 127               'blog/index.mako')
 128
 129     # generate blog feed
 130     recent_blog_posts = blog_posts
 131     updated = max([p.updated_date()
 132                    for p in recent_blog_posts])
 133     recent_blog_posts = recent_blog_posts[:10]
 134     make_mako(join(rstdir, 'blog', 'atom.xml'),
 135               [p.rstfile for p in recent_blog_posts],
 136               'blog/atom.xml.mako')
 137
 138