コード例 #1
0
ファイル: test_news_import.py プロジェクト: Mindaug/akl.lt
    def test_manager(self):
        export_dir = fixture('')
        root = get_default_site().root_page
        manager = ImportManager(root, export_dir)
        manager.add_importers([NewsImporter('Naujienos', 'naujienos')])
        for importer, item in manager.iterate():
            importer.import_item(item)

        slugs = Page.objects.values_list('slug', flat=True)
        self.assertEqual(sorted(slugs), [
            'home', 'naujiena_0001', 'naujiena_1016', 'naujienos', 'root',
        ])

        page = Page.objects.get(slug='naujiena_0001')
        self.assertEqual(page.title, 'Konkursas')
コード例 #2
0
    def test_manager(self):
        export_dir = fixture('')
        root = get_default_site().root_page
        manager = ImportManager(root, export_dir)
        manager.add_importers([NewsImporter('Naujienos', 'naujienos')])
        for importer, item in manager.iterate():
            importer.import_item(item)

        slugs = Page.objects.values_list('slug', flat=True)
        self.assertEqual(sorted(slugs), [
            'home',
            'naujiena_0001',
            'naujiena_1016',
            'naujienos',
            'root',
        ])

        page = Page.objects.get(slug='naujiena_0001')
        self.assertEqual(page.title, 'Konkursas')
コード例 #3
0
ファイル: akllt_importzope.py プロジェクト: Mindaug/akl.lt
    def handle(self, export_dir, *args, **options):
        verbosity = int(options['verbosity'])

        if not User.objects.filter(username='******').exists():
            User.objects.create_superuser('admin', 'admin@localhost', 'admin')

        root = Page.objects.get(url_path='/')
        if not Page.objects.filter(url_path='/akl/').exists():
            site_root = root.add_child(instance=IndexPage(
                title='AKL',
                slug='akl',
            ))
            site = Site.objects.get(is_default_site=True)
            site.root_page = site_root
            site.save()
        else:
            site_root = Page.objects.get(url_path='/akl/')

        manager = ImportManager(site_root, export_dir)
        manager.add_importers([
            NewsImporter('Naujienos', 'naujienos', in_menu=False),
            PagesImporter('Atviras kodas', 'ak', in_menu=True),
            PagesImporter('Apie AKL', 'apie', in_menu=True),
            PagesImporter('Projektai', 'projektai', in_menu=True),
            PagesImporter('Skaitykla', 'skaitykla', in_menu=True),
            PagesImporter('Rėmėjai', 'remejai', in_menu=True),
            PagesImporter('Nuorodos', 'nuorodos', in_menu=True),
            PagesImporter('Balsavimas', 'balsavimas', in_menu=False),
            PagesImporter('Programos', 'programos', in_menu=False),
            PagesImporter('2004', '2004', in_menu=False),
            PagesImporter('2005', '2005', in_menu=False),
            PagesImporter('2006', '2006', in_menu=False),
            PagesImporter('2009', '2009', in_menu=False),
            PagesImporter('2010', '2010', in_menu=False),
        ])

        if verbosity == 1:
            items = tqdm.tqdm(manager.iterate(), total=manager.get_total())
        else:
            items = manager.iterate()

        n_created = 0
        n_updated = 0

        for importer, item in items:
            if verbosity > 1:
                self.stdout.write(str(item.path))
            try:
                item = importer.import_item(item)
            except:
                self.stdout.write((
                    '\n\nError occured while importing {path} news file.'
                ).format(path=item.path))
                raise

            if item.created:
                n_created += 1
            else:
                n_updated += 1

        self.stdout.write((
            'Successfully imported {n_created} and updated {n_updated} '
            'items. Total {total} items processed.\n'
        ).format(
            n_created=n_created, n_updated=n_updated,
            total=(n_updated + n_created),
        ))

        pages = (p.specific for p in Page.objects.all())
        if verbosity == 1:
            total = Page.objects.count()
            generator = tqdm.tqdm(convert_links(pages), total=total)
        else:
            generator = convert_links(pages)

        total = sum(1 for _ in generator)

        self.stdout.write((
            'Successfully converted links in {total} pages.'
        ).format(total=total))
コード例 #4
0
    def handle(self, export_dir, *args, **options):
        verbosity = int(options['verbosity'])

        if not User.objects.filter(username='******').exists():
            User.objects.create_superuser('admin', 'admin@localhost', 'admin')

        root = Page.objects.get(url_path='/')
        if not Page.objects.filter(url_path='/akl/').exists():
            site_root = root.add_child(instance=IndexPage(
                title='AKL',
                slug='akl',
            ))
            site = Site.objects.get(is_default_site=True)
            site.root_page = site_root
            site.save()
        else:
            site_root = Page.objects.get(url_path='/akl/')

        manager = ImportManager(site_root, export_dir)
        manager.add_importers([
            NewsImporter('Naujienos', 'naujienos', in_menu=False),
            PagesImporter('Atviras kodas', 'ak', in_menu=True),
            PagesImporter('Apie AKL', 'apie', in_menu=True),
            PagesImporter('Projektai', 'projektai', in_menu=True),
            PagesImporter('Skaitykla', 'skaitykla', in_menu=True),
            PagesImporter('Rėmėjai', 'remejai', in_menu=True),
            PagesImporter('Nuorodos', 'nuorodos', in_menu=True),
            PagesImporter('Balsavimas', 'balsavimas', in_menu=False),
            PagesImporter('Programos', 'programos', in_menu=False),
            PagesImporter('2004', '2004', in_menu=False),
            PagesImporter('2005', '2005', in_menu=False),
            PagesImporter('2006', '2006', in_menu=False),
            PagesImporter('2009', '2009', in_menu=False),
            PagesImporter('2010', '2010', in_menu=False),
        ])

        if verbosity == 1:
            items = tqdm.tqdm(manager.iterate(), total=manager.get_total())
        else:
            items = manager.iterate()

        n_created = 0
        n_updated = 0

        for importer, item in items:
            if verbosity > 1:
                self.stdout.write(str(item.path))
            try:
                item = importer.import_item(item)
            except:
                self.stdout.write(
                    ('\n\nError occured while importing {path} news file.'
                     ).format(path=item.path))
                raise

            if item.created:
                n_created += 1
            else:
                n_updated += 1

        self.stdout.write(
            ('Successfully imported {n_created} and updated {n_updated} '
             'items. Total {total} items processed.\n').format(
                 n_created=n_created,
                 n_updated=n_updated,
                 total=(n_updated + n_created),
             ))

        pages = (p.specific for p in Page.objects.all())
        if verbosity == 1:
            total = Page.objects.count()
            generator = tqdm.tqdm(convert_links(pages), total=total)
        else:
            generator = convert_links(pages)

        total = sum(1 for _ in generator)

        self.stdout.write(
            ('Successfully converted links in {total} pages.').format(
                total=total))