def weeklyreport_edit(request, wr_id):
wr = WeeklyReport.objects.get(id=wr_id)
if wr.xiangmujingli == Xiangmujingli.objects.get(user=request.user):
if request.method == "POST":
form = WeeklyReportForm(request.POST, request.FILES, instance=wr)
if form.is_valid():
form.save()
return HttpResponseRedirect("/weeklyreport/")
else:
form = WeeklyReportForm(instance=wr)
return render_to_response("weeklyreport/form.html", {"form": form}, context_instance=RequestContext(request))
else:
return HttpResponseRedirect("/weeklyreport/")
def weeklyreport_add(request):
if request.method == "POST":
form = WeeklyReportForm(request.POST, request.FILES)
if form.is_valid():
wr = form.save(commit=False)
xiangmujingli = Xiangmujingli.objects.get(user=request.user)
wr.xiangmujingli = xiangmujingli
if request.POST.get("week") == "this_week":
wr.week = time.strftime("%Y-%m-%d", time.localtime(get_week_begin()))
elif request.POST.get("week") == "last_week":
wr.week = time.strftime("%Y-%m-%d", time.localtime(get_week_begin(N=-1)))
wr.save()
return HttpResponseRedirect("/weeklyreport/")
else:
week_begin_day = time.strftime("%Y-%m-%d", time.localtime(get_week_begin()))
week_end_day = time.strftime("%Y-%m-%d", time.localtime(get_week_end()))
last_week_begin_day = time.strftime("%Y-%m-%d", time.localtime(get_week_begin(N=-1)))
last_week_end_day = time.strftime("%Y-%m-%d", time.localtime(get_week_end(N=-1)))
form = WeeklyReportForm()
return render_to_response("weeklyreport/form.html", locals(), context_instance=RequestContext(request))