Ejemplo n.º 1
0
    def handle(self, *args, **kwargs):
        if len(args) != 1:
            self.print_help('manage.py', self.subcommand)
            return
        filename = args[0]
        self.overwrite = kwargs.get('overwrite', False)
        self.limit = kwargs.get('limit', None)
        self.set_popularity = kwargs.get('set_popularity', False)

        if self.set_popularity:
            if self.limit:
                raise CommandError(
                    'Can not use --set_popularity with --limit, the full data '
                    'set is needed to set popularity, aborting.')
            self.websites = []
            self.ranking_per_region = collections.defaultdict(list)

        with translation.override('en-US'):
            self.languages = dict(LANGUAGES).keys()
            self.reversed_languages = {v['English'].lower(): k for k, v
                                       in LANGUAGES.items()}
            self.categories = CATEGORY_CHOICES_DICT.keys()
            self.reversed_categories = {unicode(v).lower(): k for k, v
                                        in CATEGORY_CHOICES_DICT.items()}
        data = self.parse(filename)
        created_count = self.create_instances(data)
        print 'Import phase done, created %d websites.' % created_count

        if self.set_popularity:
            self.assign_popularity()
            self.assign_last_updated()
    def handle(self, *args, **kwargs):
        if len(args) != 1:
            self.print_help('manage.py', self.subcommand)
            return
        filename = args[0]
        self.overwrite = kwargs.get('overwrite', False)
        self.limit = kwargs.get('limit', None)
        self.set_popularity = kwargs.get('set_popularity', False)

        if self.set_popularity:
            if self.limit:
                raise CommandError(
                    'Can not use --set_popularity with --limit, the full data '
                    'set is needed to set popularity, aborting.')
            self.websites = []
            self.ranking_per_region = collections.defaultdict(list)

        with translation.override('en-US'):
            self.languages = dict(LANGUAGES).keys()
            self.reversed_languages = {
                v['English'].lower(): k
                for k, v in LANGUAGES.items()
            }
            self.categories = CATEGORY_CHOICES_DICT.keys()
            self.reversed_categories = {
                unicode(v).lower(): k
                for k, v in CATEGORY_CHOICES_DICT.items()
            }
        data = self.parse(filename)
        created_count = self.create_instances(data)
        print 'Import phase done, created %d websites.' % created_count

        if self.set_popularity:
            self.assign_popularity()
            self.assign_last_updated()
Ejemplo n.º 3
0
 def handle(self, *args, **options):
     filename = options['filename'][0]
     self.dev = options['dev']
     with translation.override('en-US'):
         self.languages = dict(LANGUAGES).keys()
         self.reversed_languages = {v['English'].lower(): k for k, v
                                    in LANGUAGES.items()}
         self.categories = CATEGORY_CHOICES_DICT.keys()
         self.reversed_categories = {unicode(v).lower(): k for k, v
                                     in CATEGORY_CHOICES_DICT.items()}
     data = list(self.parse(filename))
     # Skip first line, since it's explanatory, not a site.
     created_count = self.create_instances(data[1:])
     print 'Import phase done, created %d websites.' % created_count
    def handle(self, *args, **kwargs):
        if len(args) != 1:
            self.print_help('manage.py', self.subcommand)
            return
        filename = args[0]
        self.overwrite = kwargs.get('overwrite', False)
        self.limit = kwargs.get('limit', None)
        self.set_popularity = kwargs.get('set_popularity', False)

        with translation.override('en-US'):
            self.languages = dict(LANGUAGES).keys()
            self.reversed_languages = {v['English'].lower(): k for k, v
                                       in LANGUAGES.items()}
            self.categories = CATEGORY_CHOICES_DICT.keys()
            self.reversed_categories = {unicode(v).lower(): k for k, v
                                        in CATEGORY_CHOICES_DICT.items()}
        data = list(self.parse(filename))
        # Skip first line, since it's explanatory, not a site.
        created_count = self.create_instances(data[1:])
        print 'Import phase done, created %d websites.' % created_count
Ejemplo n.º 5
0
    def handle(self, *args, **kwargs):
        if len(args) != 1:
            self.print_help('manage.py', self.subcommand)
            return
        filename = args[0]
        self.overwrite = kwargs.get('overwrite', False)
        self.limit = kwargs.get('limit', None)
        self.set_popularity = kwargs.get('set_popularity', False)

        with translation.override('en-US'):
            self.languages = dict(LANGUAGES).keys()
            self.reversed_languages = {
                v['English'].lower(): k
                for k, v in LANGUAGES.items()
            }
            self.categories = CATEGORY_CHOICES_DICT.keys()
            self.reversed_categories = {
                unicode(v).lower(): k
                for k, v in CATEGORY_CHOICES_DICT.items()
            }
        data = list(self.parse(filename))
        # Skip first line, since it's explanatory, not a site.
        created_count = self.create_instances(data[1:])
        print 'Import phase done, created %d websites.' % created_count
Ejemplo n.º 6
0
 def get_object(self, *args, **kwargs):
     cat = CATEGORY_CHOICES_DICT.get(self.kwargs['pk'])
     if cat is None:
         raise Http404
     else:
         return self.kwargs['pk'], cat
Ejemplo n.º 7
0
 def get_object(self, *args, **kwargs):
     cat = CATEGORY_CHOICES_DICT.get(self.kwargs['pk'])
     if cat is None:
         raise Http404
     else:
         return self.kwargs['pk'], cat