def demoralize(run_transaction=True, verbosity=0):
begin_work(run_transaction)
cur = connection.cursor()
cur.execute('DELETE FROM %s' % DenormalizedCategoryUserRole._meta.db_table)
group_category = dict()
denormalized = None
'''5181183-core.delete_placement-223-1-63'''
for role in CategoryUserRole.objects.all():
# Optimalization -- create dict for set of categories and certain group. Then copy this denorm. data and change only user field.
key = u'%s_' % role.group
for cat in role.category.all():
key += u'%s-' % cat.title
if key in group_category:
printv('Saving denormalized %s (fast)' % role, STD_VERB)
user_id = role.user.pk
for d in group_category[key]:
#copy denorm. data
nd = DenormalizedCategoryUserRole(
contenttype_id=d.contenttype_id,
user_id=user_id,
permission_codename=d.permission_codename,
permission_id=d.permission_id,
category_id=d.category_id,
root_category_id=d.root_category_id
)
try:
nd.save()
except Exception, e:
printv(str(e))
else:
printv('Saving denormalized %s' % role, STD_VERB)
denormalized = role.sync_denormalized()
group_category[key] = denormalized
def denormalize(username_list, run_transaction=True, verbosity=0):
cur = connection.cursor()
group_category = dict()
denormalized = None
prev_user = None
queryset = CategoryUserRole.objects.select_related().order_by('user')
if username_list:
queryset = queryset.filter(user__username__in=username_list)
for role in queryset.iterator():
# Optimalization -- create dict for set of categories and certain group. Then copy this denorm. data and change only user field.
key = u'%s_' % role.group
for cat in role.category.order_by('-title').all():
key += u'%s-' % cat.title
# Transaction handling, deleting all rows with actual role.user_id
if prev_user != role.user:
# ommit commit when processing first user
if prev_user is not None:
commit_work(run_transaction)
reset_queries()
begin_work(run_transaction)
begin = time()
cur.execute(
'DELETE FROM %s WHERE user_id = %d;' %
(DenormalizedCategoryUserRole._meta.db_table, role.user.pk)
)
printv('Delete took %f sec.' % (time() - begin))
if key in group_category:
printv('Saving denormalized %s (fast)' % role, STD_VERB)
user_id = role.user.pk
begin = time()
for d in group_category[key]:
#copy denorm. data
nd = DenormalizedCategoryUserRole(
contenttype_id=d.contenttype_id,
user_id=user_id,
permission_codename=d.permission_codename,
permission_id=d.permission_id,
category_id=d.category_id,
root_category_id=d.root_category_id
)
try:
nd.save()
except Exception, e:
printv(str(e))
printv('Took %f sec' % (time() - begin))
else:
printv('Saving denormalized %s' % role, STD_VERB)
begin = time()
denormalized = role.sync_denormalized()
printv('Took %f sec' % (time() - begin))
group_category[key] = denormalized
prev_user = role.user
