def index(request): """ handle top level pages. """ page_name = request.path[1:][:-1] view_objects = {} if page_name != "home" and not challenge_mgr.is_page_unlock(request.user, page_name): return HttpResponseForbidden('<h1>Permission denied</h1>') # get the view_objects is_page_defined = supply_view_objects(request, page_name, view_objects) if not is_page_defined: raise Http404 # get user resource rank and usage team = request.user.get_profile().team if team: if challenge_mgr.is_page_unlock(request.user, "energy"): view_objects["energy_rank_info"] = resource_mgr.resource_team_rank_info(team, "energy") if challenge_mgr.is_page_unlock(request.user, "water"): view_objects["water_rank_info"] = resource_mgr.resource_team_rank_info(team, "water") #time_start = datetime.datetime.now() response = render_to_response("%s.html" % page_name, { "view_objects": view_objects, }, context_instance=RequestContext(request)) #time_end = datetime.datetime.now() #print "%s time: %s" % ("render", (time_end - time_start)) return response
def testEnergy(self): """test Energy.""" date = datetime.date.today() EnergyUsage( team=self.team, date=date, time=datetime.time(hour=15), usage=100, ).save() rank = resource_mgr.resource_team_rank_info(self.team, "energy")["rank"] usage = resource_mgr.team_resource_usage(date=date, team=self.team, resource="energy") self.assertEqual(rank, 1, "The team should be first rank.") self.assertEqual(usage, 100, "The team usage is not correct.")