1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
from django.apps import apps
from django.core.management import BaseCommand, CommandError
from django.db import transaction
from ordered_model.models import OrderedModelBase
class Command(BaseCommand):
help = "Re-do the ordering of a certain Model"
def add_arguments(self, parser):
parser.add_argument("model_name", type=str, nargs="*")
def handle(self, *args, **options):
"""
Sometimes django-ordered-models ordering goes wrong, for various reasons,
try re-ordering to a working state.
"""
self.verbosity = options["verbosity"]
orderedmodels = [
m._meta.label for m in apps.get_models() if issubclass(m, OrderedModelBase)
]
candidates = "\n {}".format("\n ".join(orderedmodels))
if not options["model_name"]:
return self.stdout.write("No model specified, try: {}".format(candidates))
for model_name in options["model_name"]:
if model_name not in orderedmodels:
self.stdout.write(
"Model '{}' is not an ordered model, try: {}".format(
model_name, candidates
)
)
break
model = apps.get_model(model_name)
if not issubclass(model, OrderedModelBase):
raise CommandError(
"{} does not inherit from OrderedModel or OrderedModelBase".format(
str(model)
)
)
self.reorder(model)
def reorder(self, model):
owrt = model.get_order_with_respect_to()
if owrt:
rel_kwargs = dict([("{}__isnull".format(k), False) for k in owrt])
relation_to_list = (
model.objects.order_by(*owrt)
.values_list(*owrt)
.filter(**rel_kwargs)
.distinct()
)
for relation_to in relation_to_list:
kwargs = dict([(k, v) for k, v in zip(owrt, relation_to)])
# print('re-ordering: {}'.format(kwargs))
self.reorder_queryset(model.objects.filter(**kwargs))
else:
self.reorder_queryset(model.objects.all())
@transaction.atomic
def reorder_queryset(self, queryset):
model = queryset.model
order_field_name = model.order_field_name
bulk_update_list = []
for order, obj in enumerate(queryset):
if getattr(obj, order_field_name) != order:
if self.verbosity:
self.stdout.write(
"changing order of {} ({}) from {} to {}".format(
model._meta.label,
obj.pk,
getattr(obj, order_field_name),
order,
)
)
setattr(obj, order_field_name, order)
bulk_update_list.append(obj)
model.objects.bulk_update(bulk_update_list, [order_field_name])
|
