def add_instance(request, model_name):
def select_model_by_name(name):
from catalog.admin.utils import get_connected_models
connected_models = get_connected_models()
model_list = [item[0].__name__.lower() for item in connected_models]
try:
return connected_models[model_list.index(name)][0]
except ValueError:
return None
parent_tree_item = TreeItem.manager.json(request.REQUEST.get('parent', 'root'))
model_cls = select_model_by_name(model_name)
if model_cls is None:
raise Http404
instance = model_cls(name=u'%s' % model_cls._meta.verbose_name)
instance.save(save_tree_id=False)
tree_item = TreeItem(parent=parent_tree_item, content_object=instance)
tree_item.save()
instance.save()
return HttpResponseRedirect('/admin/%s/%s/%d/?_popup=1'
% (instance.__module__.rsplit('.', 2)[-2], model_name, instance.id))
def _update_or_create_item(self, **kwds):
item_options_list = [
'price', 'wholesale_price', 'quantity', 'barcode',
'identifier', 'short_description', 'name', 'slug',
]
item_options = {}
for k, v in kwds.iteritems():
if k in item_options_list:
item_options[k] = v
item_options['slug'] = urlify(kwds['name'])
if kwds['identifier'] in self.data['item_by_identifier']:
item = self.data['item_by_identifier'][kwds['identifier']]
for k, v in item_options.iteritems():
setattr(item, k, v)
item.save()
self.data['item_by_identifier'].update({item.identifier: item})
logging.debug('[U] %s' % kwds['name'])
# True if created
return False
else:
item = Item(**item_options)
item.save()
tree_item = TreeItem(parent=kwds['parent'], content_object=item)
tree_item.save()
self.data['item_by_identifier'].update({item.identifier: item})
logging.debug('[S] %s' % item.name)
# True if created
return True
def insert_in_tree(sender, instance, **kwargs):
"""
Create TreeItem object after content object created
"""
created = kwargs.pop('created', False)
if created:
tree_item = TreeItem(parent=None, content_object=instance)
tree_item.save()
else:
tree_item = instance.tree.get()
if tree_item.get_slug() and \
instance.full_path() != cache.get(instance.cache_url_key()):
instance.clear_cache()
def insert_in_tree(sender, instance, **kwargs):
"""
Create TreeItem object after content object created
"""
created = kwargs.pop('created', False)
if created:
tree_item = TreeItem(parent=None, content_object=instance)
tree_item.save()
else:
tree_item = instance.tree.get()
if tree_item.get_slug() and \
instance.full_path() != cache.get(instance.cache_url_key()):
instance.clear_cache()
def _get_or_create_section(self, name, parent):
if name in self.data['section_by_name']:
return self.data['section_by_name'][name].tree.get()
else:
section = Section(name=name, slug=urlify(name))
section.save()
section_tree_item = TreeItem(parent=parent, content_object=section)
section_tree_item.save()
self.data['section_by_name'].update({section.name: section})
logging.debug('[S] === %s ===' % section)
return section_tree_item
def _get_or_create_section(self, name, parent):
if name in self.data['section_by_name']:
return self.data['section_by_name'][name].tree.get()
else:
section = Section(
name=name, slug = urlify(name)
)
section.save()
section_tree_item = TreeItem(parent=parent, content_object=section)
section_tree_item.save()
self.data['section_by_name'].update({section.name: section})
logging.debug('[S] === %s ===' % section)
return section_tree_item
def _update_or_create_item(self, **kwds):
item_options_list = [
'price',
'wholesale_price',
'quantity',
'barcode',
'identifier',
'short_description',
'name',
'slug',
]
item_options = {}
for k, v in kwds.iteritems():
if k in item_options_list:
item_options[k] = v
item_options['slug'] = urlify(kwds['name'])
if kwds['identifier'] in self.data['item_by_identifier']:
item = self.data['item_by_identifier'][kwds['identifier']]
for k, v in item_options.iteritems():
setattr(item, k, v)
item.save()
self.data['item_by_identifier'].update({item.identifier: item})
logging.debug('[U] %s' % kwds['name'])
# True if created
return False
else:
item = Item(**item_options)
item.save()
tree_item = TreeItem(parent=kwds['parent'], content_object=item)
tree_item.save()
self.data['item_by_identifier'].update({item.identifier: item})
logging.debug('[S] %s' % item.name)
# True if created
return True
def create_item(self, item_params):
try:
target_tree_item = TreeItem.objects.get(name=u'Импорт')
except TreeItem.DoesNotExist:
target_tree_item = TreeItem(parent_id=None, name=u'Импорт')
target_tree_item.save()
target_section = Section(is_meta_item=False)
target_section.save()
target_tree_item.section = target_section
target_tree_item.save()
new_tree_item = TreeItem(name=item_params['name'][:200],
parent=target_tree_item)
new_item = Item(
price=item_params['price'],
identifier=item_params['identifier'],
quantity=item_params['quantity'],
)
new_tree_item.save()
new_item.save()
new_tree_item.item = new_item
new_tree_item.save()
def create_item(self, item_params):
try:
target_tree_item = TreeItem.objects.get(name=u'Импорт')
except TreeItem.DoesNotExist:
target_tree_item = TreeItem(parent_id=None, name=u'Импорт')
target_tree_item.save()
target_section = Section(is_meta_item=False)
target_section.save()
target_tree_item.section = target_section
target_tree_item.save()
new_tree_item = TreeItem(name = item_params['name'][:200],
parent=target_tree_item)
new_item = Item(
price = item_params['price'],
identifier = item_params['identifier'],
quantity = item_params['quantity'],
)
new_tree_item.save()
new_item.save()
new_tree_item.item = new_item
new_tree_item.save()
