-
Notifications
You must be signed in to change notification settings - Fork 1
/
shared.py
68 lines (50 loc) · 2.02 KB
/
shared.py
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
#!/usr/bin/env python
# vim: noet
from datetime import datetime, timedelta
from django.utils.text import capfirst
from django.utils import translation
def current_reporting_period():
"""Return a 2-tuple containing datetimes of the start and end
of the current reporting period (the previous calendar week)"""
# offsets are calculated from today use a datetime, because we
# want the reporting period to end on the last second of sunday
today = datetime.today().replace(
hour=0, minute=0, second=0, microsecond=0)
# reporting period spans last week, monday to sunday
start = today - timedelta(days=today.weekday() + 7)
end = start + timedelta(days=7) - timedelta(seconds=1)
# return a tuple
return (start, end)
def nested_fields(model, max_depth=4):
def iterate(model, nest=None):
fields = []
# iterate all of the fields in this model, and recurse
# each foreign key to include fields from nested models
for field in model._meta.fields:
if nest is None: my_nest = [field]
else: my_nest = nest + [field]
# is this a foreign key? (also abort if depth is too deep)
if (hasattr(field, "rel")) and (field.rel is not None) and (len(my_nest) <= max_depth):
fields.extend(iterate(field.rel.to, my_nest))
# add the field object as a tuple, including the
# caption (containing prefixes) and the class itself
else:
filter = "__".join([f.name for f in my_nest])
label = "/".join([capfirst(translation.ungettext(f.verbose_name, "", 1)) for f in my_nest])
fields.append((filter, label, field))
return fields
# start processing at the top
return iterate(model)
def shuffle_locations():
from webui.inventory.models import Location
import random
# ithe center of universe
# is addis ababa. who knew?
lat = 9.010507476905097
lng = 38.75518441200256
# iterate each location, and move them all
# somewhere near the center of the universe
for loc in Location.objects.all():
loc.latitude = lat + ((random.random()*10) - 5)
loc.longitude = lng + ((random.random()*10) - 5)
loc.save()