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

wordpress importer support for draft article

Browse source
This commit is contained in:
Arul 2015-02-01 14:01:35 +05:30 • committed by Arul
commit 77231e97c0
2 changed files with 24 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

4
pelican/tests/test_importer.py
View file

@ -254,13 +254,13 @@ class TestBuildHeader(unittest.TestCase):
def test_galleries_added_to_header(self): def test_galleries_added_to_header(self):
header = build_header('test', None, None, None, None, header = build_header('test', None, None, None, None,
None, ['output/test1', 'output/test2']) None, attachments=['output/test1', 'output/test2'])
self.assertEqual(header, 'test\n####\n' + ':attachments: output/test1, ' self.assertEqual(header, 'test\n####\n' + ':attachments: output/test1, '
+ 'output/test2\n\n') + 'output/test2\n\n')
def test_galleries_added_to_markdown_header(self): def test_galleries_added_to_markdown_header(self):
header = build_markdown_header('test', None, None, None, None, None, header = build_markdown_header('test', None, None, None, None, None,
['output/test1', 'output/test2']) attachments=['output/test1', 'output/test2'])
self.assertEqual(header, 'Title: test\n' + 'Attachments: output/test1, ' self.assertEqual(header, 'Title: test\n' + 'Attachments: output/test1, '
+ 'output/test2\n\n') + 'output/test2\n\n')

34
pelican/tools/pelican_import.py
View file

@ -126,7 +126,7 @@ def wp2fields(xml, wp_custpost=False):
items = get_items(xml) items = get_items(xml)
for item in items: for item in items:
if item.find('status').string == "publish": if item.find('status').string in ["publish", "draft"]:
try: try:
# Use HTMLParser due to issues with BeautifulSoup 3 # Use HTMLParser due to issues with BeautifulSoup 3
@ -149,6 +149,8 @@ def wp2fields(xml, wp_custpost=False):
# caturl = [cat['nicename'] for cat in item.find(domain='category')] # caturl = [cat['nicename'] for cat in item.find(domain='category')]
tags = [tag.string for tag in item.findAll('category', {'domain' : 'post_tag'})] tags = [tag.string for tag in item.findAll('category', {'domain' : 'post_tag'})]
# To publish a post the status should be 'published'
status = 'published' if item.find('status').string == "publish" else item.find('status').string
kind = 'article' kind = 'article'
post_type = item.find('post_type').string post_type = item.find('post_type').string
@ -165,7 +167,7 @@ def wp2fields(xml, wp_custpost=False):
pass pass
else: else:
kind = post_type kind = post_type
yield (title, content, filename, date, author, categories, tags, yield (title, content, filename, date, author, categories, tags, status,
kind, "wp-html") kind, "wp-html")
def dc2fields(file): def dc2fields(file):
@ -296,9 +298,10 @@ def dc2fields(file):
post_format = "html" post_format = "html"
kind = 'article' # TODO: Recognise pages kind = 'article' # TODO: Recognise pages
status = 'published' # TODO: Find a way for draft posts
yield (post_title, content, slugify(post_title), post_creadt, author, yield (post_title, content, slugify(post_title), post_creadt, author,
categories, tags, kind, post_format) categories, tags, status, kind, post_format)
def posterous2fields(api_token, email, password): def posterous2fields(api_token, email, password):
@ -347,9 +350,10 @@ def posterous2fields(api_token, email, password):
date_object -= delta date_object -= delta
date = date_object.strftime("%Y-%m-%d %H:%M") date = date_object.strftime("%Y-%m-%d %H:%M")
kind = 'article' # TODO: Recognise pages kind = 'article' # TODO: Recognise pages
status = 'published' # TODO: Find a way for draft posts
yield (post.get('title'), post.get('body_cleaned'), slug, date, yield (post.get('title'), post.get('body_cleaned'), slug, date,
post.get('user').get('display_name'), [], tags, kind, "html") post.get('user').get('display_name'), [], tags, status, kind, "html")
def tumblr2fields(api_key, blogname): def tumblr2fields(api_key, blogname):
@ -426,8 +430,10 @@ def tumblr2fields(api_key, blogname):
content = content.rstrip() + '\n' content = content.rstrip() + '\n'
kind = 'article' kind = 'article'
status = 'published' # TODO: Find a way for draft posts
yield (title, content, slug, date, post.get('blog_name'), [type], yield (title, content, slug, date, post.get('blog_name'), [type],
tags, kind, format) tags, status, kind, format)
offset += len(posts) offset += len(posts)
posts = get_tumblr_posts(api_key, blogname, offset) posts = get_tumblr_posts(api_key, blogname, offset)
@ -444,10 +450,10 @@ def feed2fields(file):
slug = slugify(entry.title) slug = slugify(entry.title)
kind = 'article' kind = 'article'
yield (entry.title, entry.description, slug, date, author, [], tags, yield (entry.title, entry.description, slug, date, author, [], tags, None,
kind, "html") kind, "html")
def build_header(title, date, author, categories, tags, slug, attachments=None): def build_header(title, date, author, categories, tags, slug, status=None, attachments=None):
from docutils.utils import column_width from docutils.utils import column_width
"""Build a header from a list of fields""" """Build a header from a list of fields"""
@ -462,13 +468,15 @@ def build_header(title, date, author, categories, tags, slug, attachments=None):
header += ':tags: %s\n' % ', '.join(tags) header += ':tags: %s\n' % ', '.join(tags)
if slug: if slug:
header += ':slug: %s\n' % slug header += ':slug: %s\n' % slug
if status:
header += ':status: %s\n' % status
if attachments: if attachments:
header += ':attachments: %s\n' % ', '.join(attachments) header += ':attachments: %s\n' % ', '.join(attachments)
header += '\n' header += '\n'
return header return header
def build_markdown_header(title, date, author, categories, tags, slug, def build_markdown_header(title, date, author, categories, tags, slug, status=None,
attachments=None): attachments=None):
"""Build a header from a list of fields""" """Build a header from a list of fields"""
header = 'Title: %s\n' % title header = 'Title: %s\n' % title
if date: if date:
@ -481,6 +489,8 @@ def build_markdown_header(title, date, author, categories, tags, slug,
header += 'Tags: %s\n' % ', '.join(tags) header += 'Tags: %s\n' % ', '.join(tags)
if slug: if slug:
header += 'Slug: %s\n' % slug header += 'Slug: %s\n' % slug
if status:
header += 'Status: %s\n' % status
if attachments: if attachments:
header += 'Attachments: %s\n' % ', '.join(attachments) header += 'Attachments: %s\n' % ', '.join(attachments)
header += '\n' header += '\n'
@ -606,7 +616,7 @@ def fields2pelican(fields, out_markup, output_path,
dircat=False, strip_raw=False, disable_slugs=False, dircat=False, strip_raw=False, disable_slugs=False,
dirpage=False, filename_template=None, filter_author=None, dirpage=False, filename_template=None, filter_author=None,
wp_custpost=False, wp_attach=False, attachments=None): wp_custpost=False, wp_attach=False, attachments=None):
for (title, content, filename, date, author, categories, tags, for (title, content, filename, date, author, categories, tags, status,
kind, in_markup) in fields: kind, in_markup) in fields:
if filter_author and filter_author != author: if filter_author and filter_author != author:
continue continue
@ -624,11 +634,11 @@ def fields2pelican(fields, out_markup, output_path,
ext = get_ext(out_markup, in_markup) ext = get_ext(out_markup, in_markup)
if ext == '.md': if ext == '.md':
header = build_markdown_header(title, date, author, categories, header = build_markdown_header(title, date, author, categories,
tags, slug, attached_files) tags, slug, status, attached_files)
else: else:
out_markup = "rst" out_markup = "rst"
header = build_header(title, date, author, categories, header = build_header(title, date, author, categories,
tags, slug, attached_files) tags, slug, status, attached_files)
out_filename = get_out_filename(output_path, filename, ext, out_filename = get_out_filename(output_path, filename, ext,
kind, dirpage, dircat, categories, wp_custpost) kind, dirpage, dircat, categories, wp_custpost)