github/pelican
1
0
Fork 1
mirror of https://github.com/getpelican/pelican.git synced 2025-10-15 20:28:56 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Refactor building of header part.

Browse source
This commit is contained in:
FELD Boris 2011-01-25 00:02:00 +01:00
commit a5203ae363
1 changed files with 15 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

17
tools/wp2pelican.py
View file

@ -4,6 +4,7 @@ from BeautifulSoup import BeautifulStoneSoup
from codecs import open
import os
import argparse
import time
def wp2html(xml):
xmlfile = open(xml, encoding='utf-8').read()
@ -27,6 +28,17 @@ def wp2html(xml):
yield (title, content, filename, date, author, categories, tags)
def buildHeader(title, date, author, categories, tags):
header = '%s\n%s\n' % (title, '#' * len(title))
if (date != None):
header += ':date: %s\n' % (date)
if (categories != []):
header += ':category: %s\n' % (', '.join(categories))
if (tags != []):
header += ':tags: %s\n' % (', '.join(tags))
header += '\n'
return header
if __name__ == '__main__':
parser = argparse.ArgumentParser(description="""Transform a wordpress xml export into rst files """)
@ -55,5 +67,6 @@ if __name__ == '__main__':
with open(rst_filename, 'r', encoding='utf-8') as fs:
content = fs.read()
with open(rst_filename, 'w', encoding='utf-8') as fs:
header = '%s\n%s\n\n:date: %s\n:author: %s\n\n' % (title, '#' * len(title) ,date, author)
fs.write(header+content)
categories = [x[1] for x in categories]
header = buildHeader(title, date, author, categories, tags)
fs.write(header+content)