view news/management/commands/import_old_news.py @ 113:8a0076d7d041

Bootstrap: press.
author Brian Neal <bgneal@gmail.com>
date Sun, 20 Oct 2013 11:20:24 -0500
parents b7cdfdde3999
children
line wrap: on
line source
"""
import_old_news.py - For importing news stories from the older version of this
website.

"""
import datetime
import json

from django.conf import settings
from django.core.management.base import LabelCommand
from django.utils.html import linebreaks
import textile

from news.models import News

IMG_TAG = u"""\
<img src="{src}" class="floatLeftBox" alt="{title}" title="{title}" border="0" />"""


class Command(LabelCommand):
    args = '<filename filename ...>'
    help = 'Imports older news stories in JSON format'

    def handle_label(self, filename, **options):
        """
        Process the file of older news stories in JSON. Convert to the new model
        scheme.

        """
        with open(filename, 'rb') as f:
            items = json.load(f)

        for item in items:
            if item['model'] == 'band.news':
                self.process_item(item)

    def process_item(self, item):

        fields = item['fields']

        content = fields['text'].strip()
        if fields['markup_enabled']:
            content = textile.textile(content, encoding='utf-8', output='utf-8')
        else:
            content = linebreaks(fields['text'])

        author = fields['author'].strip()
        if author:
            content += u"<p>\u2013&nbsp;%s" % author

        image = fields['photo'].strip()
        if image:
            caption = fields['photo_caption'].strip()
            caption = caption if caption else 'Image'

            src = u"%s%s" % (settings.MEDIA_URL, image)

            content = IMG_TAG.format(src=src, title=caption) + content

        news = News(id=item['pk'],
                title=fields['title'].strip(),
                date=datetime.datetime.strptime(fields['date'], '%Y-%m-%d'),
                content=content)

        news.save()