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())
