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):
fixture = options['fixture']
if path.exists(fixture):
root_el = etree.parse(fixture).getroot()
self.cleanup()
root_record = Category.add_root(title='Каталог')
root_record = Category.objects.get(pk=root_record.pk)
iter_tree(root_el, root_record, add_book)
else:
raise CommandError('Fixture file %s does not exist' % fixture)
