← Back to branch summary

~widelands-dev/widelands-website/django_staticfiles

« back to all changes in this revision

Viewing changes to threadedcomments/management/commands/migratecomments.py

  • Committer: franku
  • Date: 2018-05-06 08:20:39 UTC
  • mto: This revision was merged to the branch mainline in revision 494.
  • Revision ID: somal@arcor.de-20180506082039-v8n40alffhi2ulct
This is an Attribute error

Show diffs side-by-side

added added

removed removed

Lines of Context:
threadedcomments/management/commands/migratecomments.py
 
1
from django.core.management.base import BaseCommand
 
2
from django.contrib.comments.models import Comment, FreeComment
 
3
from threadedcomments.models import ThreadedComment, FreeThreadedComment
 
4
 
 
5
 
 
6
class Command(BaseCommand):
 
7
    help = "Migrates Django's built-in django.contrib.comments data to threadedcomments data"
 
8
 
 
9
    output_transaction = True
 
10
 
 
11
    def handle(self, *args, **options):
 
12
        """Converts all legacy ``Comment`` and ``FreeComment`` objects into
 
13
        ``ThreadedComment`` and ``FreeThreadedComment`` objects,
 
14
        respectively."""
 
15
        self.handle_free_comments()
 
16
        self.handle_comments()
 
17
 
 
18
    def handle_free_comments(self):
 
19
        """Converts all legacy ``FreeComment`` objects into
 
20
        ``FreeThreadedComment`` objects."""
 
21
        comments = FreeComment.objects.all()
 
22
        for c in comments:
 
23
            new = FreeThreadedComment(
 
24
                content_type=c.content_type,
 
25
                object_id=c.object_id,
 
26
                comment=c.comment,
 
27
                name=c.person_name,
 
28
                website='',
 
29
                email='',
 
30
                date_submitted=c.submit_date,
 
31
                date_modified=c.submit_date,
 
32
                date_approved=c.submit_date,
 
33
                is_public=c.is_public,
 
34
                ip_address=c.ip_address,
 
35
                is_approved=c.approved
 
36
            )
 
37
            new.save()
 
38
 
 
39
    def handle_comments(self):
 
40
        """Converts all legacy ``Comment`` objects into ``ThreadedComment``
 
41
        objects."""
 
42
        comments = Comment.objects.all()
 
43
        for c in comments:
 
44
            new = ThreadedComment(
 
45
                content_type=c.content_type,
 
46
                object_id=c.object_id,
 
47
                comment=c.comment,
 
48
                user=c.user,
 
49
                date_submitted=c.submit_date,
 
50
                date_modified=c.submit_date,
 
51
                date_approved=c.submit_date,
 
52
                is_public=c.is_public,
 
53
                ip_address=c.ip_address,
 
54
                is_approved=not c.is_removed
 
55
            )
 
56
            new.save()