def import_categories(): res = csv_load_file( os.path.dirname(os.path.abspath(__file__)) + "/categories.csv") #print res parent_categories = dict() for r in res: if r[1] == '': continue level = int(r[1]) cat = Category.objects.filter(pk=r[0]) if cat.count() == 0: print cat item = Category() item.id = r[0] item.name = r[2] if level == 1: print item item = Category.add_root(name=r[2], id=r[0]) parent_categories[level] = item else: item = parent_categories[level - 1].add_child(name=r[2], id=r[0]) parent_categories[level] = item else: parent_categories[level] = cat[0]
def import_categories(): res = csv_load_file(os.path.dirname(os.path.abspath (__file__)) + "/categories.csv") #print res parent_categories = dict() for r in res: if r[1] == '': continue level = int(r[1]) cat = Category.objects.filter(pk=r[0]) if cat.count() == 0: print cat item = Category() item.id = r[0] item.name = r[2] if level == 1: print item item = Category.add_root(name=r[2], id=r[0]) parent_categories[level] = item else: item = parent_categories[level-1].add_child(name=r[2], id=r[0]) parent_categories[level] = item else: parent_categories[level] = cat[0]
def handle(self, *args, **options): print('Loading category') Category.objects.all().delete() Journal.objects.all().delete() names = [ 'ЖЕНСКИЕ', 'МУЖСКИЕ', 'ДЕТСКИЕ', 'ДОМ И СЕМЬЯ', 'АВТО', 'СПОРТ', 'РЕЛИГИЯ', 'ОБЩЕСТВО', 'ПОЛИТИКА', 'БИЗНЕС', 'ФИНАНСЫ', 'НАУЧНО-ПОПУЛЯРНЫЕ' ] for name in names: c = Category() c.name = name c.save() print('Saving %s' % name) journal_list = [ 'Наука и техника', 'Здоровье', 'Моделист-конструктор', 'Лиза', 'Прибой', 'Забой', 'Внуки и правнуки', 'В мире муравья', 'Крышка стола', 'Восемь' ] cnt = 0 for jname in journal_list: cnt += 1 j = Journal() j.name = jname j.save() j.category.add(c) path = os.path.join(settings.BASE_DIR, 'init_data', 'journal', f'{cnt}.jpg') with open(path, 'rb') as doc_file: j.default_cover.save(f'{cnt}.jpeg', File(doc_file), save=True) for ni in range(1, 20): name = f'выпуск {ni}' i = Issue() i.name = name i.journal = j i.save() for pi in range(1, 10): p = IssuePage() p.paper = i p.page = pi p.save() path = os.path.join(settings.BASE_DIR, 'init_data', 'journal', f'{pi}.jpg') with open(path, 'rb') as doc_file: p.file_low.save(f'{pi}.jpeg', File(doc_file), save=True) p.file_middle.save(f'{pi}.jpeg', File(doc_file), save=True) p.file_high.save(f'{pi}.jpeg', File(doc_file), save=True)
def test_can_create_new_category(self): category = Category() category.name = 'New category' category.save() self.assertIn(category, Category.objects.all())