/
util.py
164 lines (131 loc) · 6.39 KB
/
util.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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
import datetime
from models import Snippet
from models import User
# Functions for retrieving a user
def get_user(email):
"""Return the user object with the given email, or None if not found."""
q = User.all()
q.filter('email = ', email)
return q.get()
def get_user_or_die(email):
user = get_user(email)
if not user:
raise ValueError('User "%s" not found' % email)
return user
def snippets_for_user(user_email):
"""Return all snippets for a given user, oldest snippet first."""
snippets_q = Snippet.all()
snippets_q.filter('email = ', user_email)
snippets_q.order('week') # this puts oldest snippet first
return snippets_q.fetch(1000) # good for many years...
def most_recent_snippet_for_user(user_email):
"""Return the most recent snippet for a given user, or None."""
snippets_q = Snippet.all()
snippets_q.filter('email = ', user_email)
snippets_q.order('-week') # this puts newest snippet first
return snippets_q.get()
# Functions around filling in snippets
def newsnippet_monday(today):
"""Return a datetime.date object: the monday for new snippets.
We just return the monday for this week. Saturday and Sunday
map to the previous monday.
Note that this means when you look at snippets for monday, you're
offered to enter snippets for the week that has just started, even
though not much has happened yet! This is for people who like to
enter snippets as they go along. For those people who wait until
monday to fill in the previous week's snippets, they can still do
so; the second snippet box will be marked 'DUE TODAY'.
Arguments:
today: the current day as a datetime.datetime object, used to
calculate the best monday.
Returns:
The Monday that we are accepting new snippets for, by default,
as a datetime.date (not datetime.datetime) object.
"""
today_weekday = today.weekday() # monday == 0, sunday == 6
end_monday = today - datetime.timedelta(today_weekday)
return end_monday.date()
def existingsnippet_monday(today):
"""Return a datetime.date object: the monday for existing snippets.
The rule is that we show the snippets for the previous week. We
declare a week starts on Monday...well, actually, Sunday at 11pm.
The reason for this is that (for quota reasons) we sent out a
reminder email Sunday at 11:50pm rather than Monday morning, and
we want that to count as 'Monday' anyway...
Arguments:
today: the current day as a datetime.datetime object, used to
calculate the best monday.
Returns:
The Monday that we are accepting new snippets for, by default,
as a datetime.date (not datetime.datetime) object.
"""
today_weekday = today.weekday() # monday == 0, sunday == 6
if today_weekday == 6 and today.hour >= 23:
end_monday = today - datetime.timedelta(today_weekday)
else:
end_monday = today - datetime.timedelta(today_weekday + 7)
return end_monday.date()
_ONE_WEEK = datetime.timedelta(7)
def _backfill_missing_snippets(user, all_snippets,
current_monday, first_allowed_monday):
monday_ptr = current_monday - _ONE_WEEK
while monday_ptr >= first_allowed_monday:
all_snippets.insert(0,
Snippet(
email=user.email,
week=monday_ptr,
private=user.private_snippets,
is_markdown=user.uses_markdown))
monday_ptr -= _ONE_WEEK
def fill_in_missing_snippets(existing_snippets, user, user_email, today):
"""Make sure that the snippets array has a Snippet entry for every week.
The db may have holes in it -- weeks where the user didn't write a
snippet. Augment the given snippets array so that it has no holes,
by adding in default snippet entries if necessary. Also backfill empty
snippets up until one week before user's registration week.
Note it does not add these entries to the db, it just adds them
to the array.
Arguments:
existing_snippets: a list of Snippet objects for a given user.
The first snippet in the list is assumed to be the oldest
snippet from that user (at least, it's where we start filling
from).
user: a User object for the person writing this snippet.
user_email: the email of the person whose snippets it is.
today: a datetime.datetime object representing the current day.
We fill up to then. If today is wed or before, then we
fill up to the previous week. If it's thurs or after, we
fill up to the current week.
Returns:
A new list of Snippet objects, without any holes,
up to one week before user's registration week.
"""
end_monday = newsnippet_monday(today)
first_allowed_monday = newsnippet_monday(user.created) - _ONE_WEEK
# No snippets at all? Fill empty snippets up to one week before user's
# registration week.
if not existing_snippets:
all_snippets = [Snippet(email=user_email, week=end_monday,
private=user.private_snippets,
is_markdown=user.uses_markdown)]
_backfill_missing_snippets(user, all_snippets,
end_monday, first_allowed_monday)
return all_snippets
# Add a sentinel, one week past the last week we actually want.
# We'll remove it at the end.
existing_snippets.append(Snippet(email=user_email,
week=end_monday + _ONE_WEEK))
all_snippets = [existing_snippets[0]] # start with the oldest snippet
if all_snippets[0].week - first_allowed_monday >= _ONE_WEEK:
_backfill_missing_snippets(user, all_snippets,
all_snippets[0].week, first_allowed_monday)
for snippet in existing_snippets[1:]:
while snippet.week - all_snippets[-1].week > _ONE_WEEK:
missing_week = all_snippets[-1].week + _ONE_WEEK
all_snippets.append(Snippet(email=user_email, week=missing_week,
private=user.private_snippets,
is_markdown=user.uses_markdown))
all_snippets.append(snippet)
# Get rid of the sentinel we added above.
del all_snippets[-1]
return all_snippets