view legacy/management/commands/import_old_news_comments.py @ 757:f36864bc919e

Remove NSSR banner from base template.
author Brian Neal <bgneal@gmail.com>
date Sun, 05 Jan 2014 13:06:53 -0600 (2014-01-05)
parents ee87ea74d46b
children
line wrap: on
line source
"""
import_old_news_comments.py - For importing comments on news stories from SG101 1.0 as csv files.
"""
from __future__ import with_statement
import csv
import optparse
import sys
from datetime import datetime

from django.core.management.base import LabelCommand, CommandError
from django.contrib.auth.models import User
from django.contrib.contenttypes.models import ContentType

from comments.models import Comment
from news.models import Story
import legacy.data
from legacy.html2md import MarkdownWriter


class Command(LabelCommand):
    args = '<filename filename ...>'
    help = 'Imports news story comments from the old database in CSV format'
    option_list = LabelCommand.option_list + (
        optparse.make_option("-p", "--progress", action="store_true",
            help="Output a . after every 20 comments to show progress"),
    )
    md_writer = MarkdownWriter()

    def handle_label(self, filename, **options):
        """
        Process each line in the CSV file given by filename by
        creating a new story comment.

        """
        self.show_progress = options.get('progress')
        self.users = {}

        try:
            with open(filename, "rb") as f:
                self.reader = csv.DictReader(f)
                num_rows = 0
                try:
                    for row in self.reader:
                        self.process_row(row)
                        num_rows += 1
                        if self.show_progress and num_rows % 20 == 0:
                            sys.stdout.write('.')
                            sys.stdout.flush()
                except csv.Error, e:
                    raise CommandError("CSV error: %s %s %s" % (
                        filename, self.reader.line_num, e))

                print

        except IOError:
            raise CommandError("Could not open file: %s" % filename)

    def process_row(self, row):
        """
        Process one row from the CSV file: create a Comment object for
        the row and save it in the database.

        """
        row = dict((k, v if v != 'NULL' else '') for k, v in row.iteritems())

        try:
            user = self._get_user(row['name'])
        except User.DoesNotExist:
            print "Could not find user %s for comment %s; skipping." % (
                    row['name'], row['tid'])
            return

        try:
            story = Story.objects.get(id=int(row['sid']))
        except Story.DoesNotExist:
            print "Could not find story %s for comment %s; skipping." % (
                    row['sid'], row['tid'])
            return

        comment = Comment(
            id=int(row['tid']),
            content_type = ContentType.objects.get_for_model(story),
            object_id = story.id,
            user = user,
            comment = self.to_markdown(row['comment']),
            creation_date = datetime.strptime(row['date'], "%Y-%m-%d %H:%M:%S"),
            ip_address = row['host_name'],
            is_public = True,
            is_removed = False,
        )

        comment.save()

    def _get_user(self, username):
        """
        Returns the user object with the given username.
        Throws User.DoesNotExist if not found.

        """
        try:
            return self.users[username]
        except KeyError:
            pass

        try:
            user = User.objects.get(username=username)
        except User.DoesNotExist:
            old_name = username.lower()
            try:
                user = User.objects.get(
                        username=legacy.data.KNOWN_USERNAME_CHANGES[old_name])
            except KeyError:
                raise User.DoesNotExist

        self.users[username] = user
        return user

    def to_markdown(self, s):
        self.md_writer.reset()

        if not isinstance(s, unicode):
            s = s.decode('latin-1', 'replace')

        self.md_writer.feed(s)
        return self.md_writer.markdown()