'''

set just created medias and contents public

'''

import csv

from django.conf import settings

from django.core.management.base import BaseCommand

from django.contrib.auth.models import Group

from ldt.security.cache import cached_assign

from ldt.ldt_utils.models import Content, Media

class Command(BaseCommand):

    '''

    command

    '''

    def add_arguments(self, parser):

        '''

        add arguments

        '''

        parser.add_argument('-p',

                            nargs='+',

                            dest='pathes',

                            default=None)

    def handle(self, *args, **options):

        '''

        handle command

        '''

        sources = []

        for path in options['pathes']:

            self.stdout.write(path)

            try:

                csvfile = open(path, 'r')

            except IOError:

                self.stdout.write('file can\'t be opened')

                continue

            files = csv.reader(csvfile)

            sources += list([content[0] for content in files])

            csvfile.close()

        everyone = Group.objects.get(name=settings.PUBLIC_GROUP_NAME)

        for source in sources:

            media = Media.objects.get(src=source)

            content = Content.objects.get(media_obj_id=media.id)

            content.is_public = True

            content.save()

            cached_assign('view_content', everyone, content)

            cached_assign('view_media', everyone, media)