def add_new_entry(request, template_name, page_title):
form_data = get_form_data(request)
form = PomEntryDescForm(form_data)
difficulty_form = PomEntryDifficultyForm(form_data) # added difficulty
catnameform = PomCategoryNameForm(form_data)
errorlist = []
context = {
"page_title": page_title,
"difficulty_form": difficulty_form,
"entryform": form,
"categoryform": catnameform,
"errorlist": errorlist,
"start_time_string": "",
"stop_time_string": "",
"form_errors": False,
}
if request.method == "POST":
start_time_string = request.POST[u"timerstarted"]
stop_time_string = request.POST[u"timerstopped"]
context["start_time_string"] = start_time_string
context["stop_time_string"] = stop_time_string
if form.is_valid() and catnameform.is_valid() and difficulty_form.is_valid():
desc = form.cleaned_data["description"]
difficulty = difficulty_form.cleaned_data["difficulty"] # added difficulty
# print 'cleaned_data difficulty=',difficulty,'type=',type(difficulty)
# print 'start_time_string=',start_time_string,len(start_time_string)
# print 'stop_time_string=',stop_time_string,len(stop_time_string)
if len(start_time_string) == 0 or len(stop_time_string) == 0:
# print 'starttime and endtime cannot be empty'
errorlist.append("starttime and endtime cannot be empty")
return custom_render(request, context, template_name)
start_time = long(start_time_string)
stop_time = long(stop_time_string)
if not (start_time < stop_time):
# need to put this in errors
errorlist.append("starttime should be less than endtime")
return custom_render(request, context, template_name)
try:
start_ttpl, stop_ttpl = get_timetuples(start_time, stop_time)
start_timeval = datetime.time(*start_ttpl)
stop_timeval = datetime.time(*stop_ttpl)
except ValueError:
errorlist.append("format of time entries not correct")
return custom_render(request, context, template_name)
newentry = PomEntry(description=desc)
newentry.save()
newentry.start_time = start_timeval
newentry.end_time = stop_timeval
catnames = catnameform.cleaned_data["categories"]
catnames = get_list_of_names(catnames)
cats = get_categories(request.user, catnames) # added user as creator
"""
for x in cats:
if request.user not in x.users.all():
x.users.add(request.user)#need to append, not assign
"""
newentry.categories = cats
newentry.author = request.user
newentry.difficulty = difficulty # added difficulty
# print 'before save with diff=',difficulty
newentry.save()
remove_index_key_from_cache(request) # removes index_key from cache
remove_archive_index_key_from_cache(request)
remove_current_month_chart_key_from_cache(request)
# print 'newentry saved with diff=',difficulty
return redirect("pomlog_entry_archive_index")
else:
context["form_errors"] = True
return custom_render(request, context, template_name)
elif request.method == "GET":
return custom_render(request, context, template_name)
def test_duration_for_categories_in_entry(self):
category1=PomCategory(name='magic',description='magical')
category1.save()
category2=PomCategory(name='astronomy',description='stars,planets')
category2.save()
dummy=User.objects.get(id=1)
entry1=PomEntry(today=date(2010,2,18),start_time=time(1,0,0),end_time=time(1,30,0),description='testentry1',author= dummy)
entry1.save()
entry1.categories=[category1,category2]
entry1.save()
entry2=PomEntry(today=date(2010,2,18),start_time=time(1,10,0),end_time=time(1,30,0),description='testentry2',author= dummy)
entry2.save()
entry2.categories=[category1]
entry2.save()
entryset=[entry1,entry2]
result_duration_dict=get_duration_for_categories(entryset)
expected={u'astronomy': 30,u'magic': 50}
self.assertEqual(expected,result_duration_dict)