<reponame>Ishma59/molo<filename>molo/core/management/commands/add_language_to_pages.py
from __future__ import absolute_import, unicode_literals

from django.core.management.base import BaseCommand
from molo.core.models import LanguageRelation
from molo.core.models import Page


class Command(BaseCommand):

    def handle(self, *args, **options):
        for relation in LanguageRelation.objects.all():
            if relation.page and relation.language:
                page = Page.objects.get(pk=relation.page.pk).specific
                page.language = relation.language
                page.save()
            else:
                self.stdout.write(self.style.NOTICE(
                    'Relation with pk "%s" is missing either page/language'
                    % (relation.pk)))
