def get(self):
point_categories = PointCategory.query(ancestor=PointCategory.root_key())
out = {}
for p in point_categories:
if p.parent is None:
out[p.name] = {
"name": p.name,
"baby_requirement": p.baby_requirement,
"member_requirement": p.member_requirement,
"sub_categories": [],
}
invalid_keys = []
for key in p.sub_categories:
sub = key.get()
# The key in sub_categories is no longer valid
if sub is None:
invalid_keys.append(key)
else:
sub_cat = {
"name": sub.name,
"baby_requirement": sub.baby_requirement,
"member_requirement": sub.member_requirement,
}
out[p.name]['sub_categories'].append(sub_cat)
for key in invalid_keys:
p.sub_categories.remove(key)
if invalid_keys:
p.put()
return jsonify(**out)
def post(self):
""" Creates a new point category or updates a duplicate one
If this function is given a point category with the same name as one
that already exists, it will update the existing point category. It
handles removing and adding necessary keys of parent categories.
"""
data = request.form
new_category = None
new_key = None
# TODO (phillip): we shouldn't put anything in the datastore if there is an error.
# Therefore, we should wait until the end of the function to call a
# put if possible.
# Check to see if the category already exists
for category in PointCategory.query():
if category.name == data['name']:
new_category = category
new_key = new_category.key
parent = new_category.parent
if parent is not None:
# Update old parent sub categories
parent.sub_categories.remove(new_key)
parent.put()
# If the category doesn't exist, create a new one
if not new_category:
new_category = PointCategory(parent=PointCategory.root_key())
new_category.name = data['name']
new_key = new_category.put()
# If necessary, add the proper key to the parent category
parent = data.get('parent', None)
# TODO (phillip): Should I really also be checking for "none"? Is there a
# better way?
if parent is not None and parent != "none":
parent = PointCategory.get_from_name(parent)
parent.sub_categories.append(new_key)
parent.put()
response = jsonify()
response.status_code = 201
response.headers['location'] = "/api/point-categories/" + new_category.name.replace(" ", "")
return response
def reshape_categories(cat_info):
""" Reshape flat list of catory information to include structure.
Args:
cat_info (dict): A flat dictionary of all categories, their point
requirements, and their points earned.
Returns:
A structured dictionary with each category, its sub-categoires,
points required, and points earned.
"""
out = {}
# TODO (phillip): this code is probably duplicated from the PointCategoryListAPI
point_categories = PointCategory.query(ancestor=PointCategory.root_key())
for cat in point_categories:
if cat.parent is None:
out[cat.name] = {
"required": cat_info["categories"][cat.name]["required"],
"received": cat_info["categories"][cat.name]["received"],
"level": 1,
"sub_categories": {},
}
invalid_keys = []
for key in cat.sub_categories:
sub = key.get()
# The key in sub_categories is no longer valid
if sub is None:
invalid_keys.append(key)
else:
sub_cat = {
"required": cat_info["categories"][sub.name]["required"],
"received": cat_info["categories"][sub.name]["received"],
"level": 2,
}
out[cat.name]["sub_categories"][sub.name] = sub_cat
for key in invalid_keys:
cat.sub_categories.remove(key)
if invalid_keys:
cat.put()
return out
def get(self, user_id):
output = {u"categories": {}}
user = UserData.get_user_from_id(user_id)
point_exceptions = {exc.point_category: exc.points_needed for exc in user.point_exceptions}
categories = PointCategory.query(ancestor=PointCategory.root_key())
for cat in categories:
# TODO (phillip): Write tests to test the behavior when calculating requirement for
# a category with children having a higher requirement. (i.e. max(self, children))
# TODO (phillip): remove some of the ugly repeated code below
# Add each category and the required number of points
if user.is_baby():
requirement = cat.baby_requirement
if requirement is None:
requirement = 0
sub_req_list = []
for sub in cat.sub_categories:
req = sub.get().member_requirement
if req is None:
sub_req_list.append(0)
else:
sub_req_list.append(req)
sub_req = sum(sub_req_list)
requirement = max(requirement, sub_req)
else:
requirement = cat.member_requirement
if requirement is None:
requirement = 0
# TODO (phillip): add test when one of the requirements is None
sub_req_list = []
for sub in cat.sub_categories:
req = sub.get().member_requirement
if req is None:
sub_req_list.append(0)
else:
sub_req_list.append(req)
sub_req = sum(sub_req_list)
requirement = max(requirement, sub_req)
if cat.name in point_exceptions:
requirement = point_exceptions[cat.name]
output["categories"][cat.name] = {u"required": requirement}
output["categories"][cat.name]["received"] = 0
if cat.parent is not None:
output["categories"][cat.name]["level"] = 2
else:
output["categories"][cat.name]["level"] = 1
# NOTE: At this point I am assuming each category has been added to
# the output. If this is not true, the following code will fail
# miserably.
records = PointRecord.query()
# username = user.username
records = records.filter(PointRecord.username == user.username)
for record in records:
event = Event.get_from_name(record.event_name)
if event is None:
logging.error("Uknown event " + record.event_name + " requested for point record: " + str(record))
record.key.delete()
continue
category = event.point_category.get()
if record.points_earned is None:
record.points_earned = 0
record.put()
output["categories"][category.name]["received"] += record.points_earned
# Make sure to also count the points for the parent category
if category.parent is not None:
output["categories"][category.parent.name]["received"] += record.points_earned
output["categories"][category.name]["level"] = 2
output = self.reshape_categories(output)
return jsonify(**output)
