def add_edit_contact(request, pk=None):
_form = ContractorContactForm
_model = ContractorContact
if pk is None:
record = _model()
form = _form(request.POST or None)
else:
record = get_object_or_404(_model, pk=pk)
form = _form(initial=record.__dict__)
if request.method == 'POST':
form = _form(request.POST)
if form.is_valid():
form.get_cleaned_data('pic')
populate_obj(form.cleaned_data, record)
record.save()
messages.info(request, "Successfully Updated the Database")
return redirect('contract_mgt:table_contact')
context = {
'forms' : form,
'form_title': 'Contractor Contact'
}
return render(request, 'default/add_form.html', context)
def post(self, request, *args, **kwargs):
pk = kwargs.pop('pk', None)
if pk is None:
record = self.model()
else:
record = get_object_or_404(self.model, pk=pk)
form = self.form_class(request.POST, request.FILES)
if form.is_valid():
cleaned_data = form.clean()
populate_obj(cleaned_data, record)
spec_date = cleaned_data['spec_date']
record.filename = spec_date.strftime('%Y-%b.xlsx')
record.data = {}
record.save()
kpi_calc = KpiCalculator(kpi_file=record.file.path)
kpi_calc.kpi_xl_to_spec()
record.data = kpi_calc._all_specs
record.save()
messages.success(request, "Successfully Updated the Database")
return redirect(self.success_redirect_link)
return render(request, self.template_name, {'forms': form})
def add_edit_attendance_reason(request, pk=None):
_form = ReasonForm
_model = Attendance
if pk is None:
record = _model()
form = _form(request.POST or None)
else:
record = get_object_or_404(_model, pk=pk)
form = _form(initial=record.__dict__)
if request.method == 'POST':
form = _form(request.POST)
if form.is_valid():
cleaned_data = form.clean()
populate_obj(cleaned_data, record)
record.save()
summarize_attendance(date=record.date_day, user_pk=request.user.id)
messages.info(request, "Successfully Updated the Database")
return redirect('user_mgt:table_attendance')
context = {'forms': form, 'form_title': 'Attendance'}
return render(request, 'default/add_form.html', context)
def post(self, request, *args, **kwargs):
pk = request.GET.get('pk', None)
if pk is None:
record = self.model()
else:
record = get_object_or_404(self.model, pk=pk)
form = self.form_class(request.POST)
if form.is_valid():
cleaned_data = form.clean()
tags = cleaned_data.pop('tags')
populate_obj(cleaned_data, record)
try:
record.save()
record.tags.set(*tags)
messages.success(request, "Successfully Updated the Database")
return redirect(self.success_redirect_link)
except IntegrityError:
messages.error(request,
"{} already exist".format(record.task_no))
return render(request, self.template_name, {
'forms': form,
'form_title': self.model.__name__
})
def add_edit_team_task(request, pk=None):
_form = TeamTaskForm
_model = TeamTask
if pk is None:
record = _model()
form = _form(request.POST or None)
else:
record = get_object_or_404(_model, pk=pk)
form = _form(initial=record.get_initials())
if request.method == 'POST':
form = _form(request.POST)
if form.is_valid():
cleaned_data = form.clean()
users = cleaned_data.pop('user')
populate_obj(cleaned_data, record)
record.save()
user_list = []
for element in users:
user = User.objects.get(pk=int(element))
user_list.append(user)
record.user = user_list
summarize_team_task(pk=record.id)
messages.info(request, "Successfully Updated the Database")
return redirect('team_mgt:table_team_task')
context = {'forms': form, 'form_title': 'Team Task'}
return render(request, 'default/add_form.html', context)
def post(self, request, *args, **kwargs):
pk = kwargs.pop('pk', None)
if pk is None:
record = self.model()
else:
record = get_object_or_404(self.model, pk=pk)
form = self.form_class(request.POST, request.FILES)
if form.is_valid():
cleaned_data = form.clean()
populate_obj(cleaned_data, record)
spec_date = cleaned_data['spec_date']
record.filename = spec_date.strftime('%Y-%b.xlsx')
record.data = {}
record.save()
kpi_calc = KpiCalculator(kpi_file=record.file.path)
kpi_calc.kpi_xl_to_spec()
record.data = kpi_calc._all_specs
record.save()
messages.success(request, "Successfully Updated the Database")
return redirect(self.success_redirect_link)
return render(request, self.template_name, {'forms': form})
def add_edit_attendance_reason(request, pk=None):
_form = ReasonForm
_model = Attendance
if pk is None:
record = _model()
form = _form(request.POST or None)
else:
record = get_object_or_404(_model, pk=pk)
form = _form(initial=record.__dict__)
if request.method == 'POST':
form = _form(request.POST)
if form.is_valid():
cleaned_data = form.clean()
populate_obj(cleaned_data, record)
record.save()
summarize_attendance(date=record.date_day, user_pk=request.user.id)
messages.info(request, "Successfully Updated the Database")
return redirect('user_mgt:table_attendance')
context = {
'forms' : form,
'form_title': 'Attendance'
}
return render(request, 'default/add_form.html', context)
def post(self, request, *args, **kwargs):
pk = kwargs.pop('pk', None)
if pk is None:
record = self.model()
else:
record = get_object_or_404(self.model, pk=pk)
form = self.form_class(request.POST)
if form.is_valid():
cleaned_data = form.clean()
populate_obj(cleaned_data, record)
record.save()
messages.success(request, "Successfully Updated the Database")
return self.success_redirect_link
return render(request, self.template_name, {'forms': form})
def edit_team_task_history(request, pk=None):
_form = TeamTaskHistoryEditForm
record = get_object_or_404(TeamTaskHistory, pk=pk)
form = _form(initial=record.__dict__)
if request.method == 'POST':
form = _form(request.POST)
if form.is_valid():
cleaned_data = form.clean()
populate_obj(cleaned_data, record)
record.save()
summarize_team_task(pk=record.team_task_id)
messages.info(request, "Successfully Updated the Database")
return redirect(
reverse('team_mgt:timeline') + '%s' % record.team_task_id)
context = {'forms': form, 'form_title': 'Team Task'}
return render(request, 'default/add_form.html', context)
def post(self, request, *args, **kwargs):
pk = request.GET.get('pk', None)
if pk is None:
record = self.model()
else:
record = get_object_or_404(self.model, pk=pk)
form = self.form_class(request.POST)
if form.is_valid():
cleaned_data = form.clean()
populate_obj(cleaned_data, record)
record.save()
# summarize_invoice(task_pk=record.task_id)
messages.success(request, "Successfully Updated the Database")
return redirect(self.success_redirect_link)
return render(request, self.template_name, {
'forms': form,
'form_title': self.model.__name__
})
