def training_program_list(request):
if request.method == 'GET':
with sqlite3.connect(Connection.db_path) as conn:
conn.row_factory = sqlite3.Row
db_cursor = conn.cursor()
# TODO: Add to query: e.department,
db_cursor.execute("""
select
tp.id,
tp.title,
tp.start_date,
tp.end_date,
tp.capacity
from hrapp_trainingprogram tp
order by tp.end_date asc
""")
all_training_programs = []
dataset = db_cursor.fetchall()
for row in dataset:
training_program = TrainingProgram()
training_program.id = row['id']
training_program.title = row['title']
training_program.start_date = row['start_date']
training_program.end_date = row['end_date']
training_program.capacity = row['capacity']
training_program.is_future = is_future_training(
training_program.start_date)
all_training_programs.append(training_program)
template = 'training_programs/training_programs_list.html'
context = {'training_programs': all_training_programs}
return render(request, template, context)
elif request.method == "POST":
form_data = request.POST
with sqlite3.connect(Connection.db_path) as conn:
db_cursor = conn.cursor()
db_cursor.execute(
"""
insert into hrapp_trainingprogram
(
title, start_date, end_date, capacity
)
values
(
?, ?, ?, ?
)
""",
(
form_data["title"],
form_data["start_date"],
form_data["end_date"],
form_data["capacity"],
),
)
return redirect(reverse("hrapp:training_program_list"))
def training_program_details(request, training_program_id):
if request.method == "GET":
training_program_details = get_training_program_details(
training_program_id)
employee_training = TrainingProgram()
for row in training_program_details:
if employee_training.title == "":
employee_training.id = row['id']
employee_training.title = row['title']
employee_training.start_date = row['start_date']
employee_training.end_date = row['end_date']
employee_training.capacity = row['capacity']
employee_training.is_future = is_future_training(
employee_training.start_date)
employee_training.attendees = []
if row['employee_id']:
employee_training.attendees.append(
(row['employee_id'],
f"{row['first_name']} {row['last_name']}"))
else:
employee_training.attendees.append(
(row['employee_id'],
f"{row['first_name']} {row['last_name']}"))
template = "training_programs/training_programs_detail.html"
context = {"training_details": employee_training}
return render(request, template, context)
if request.method == 'POST':
form_data = request.POST
if ("actual_method" in form_data
and form_data["actual_method"] == "DELETE"):
with sqlite3.connect(Connection.db_path) as conn:
db_cursor = conn.cursor()
db_cursor.execute(
"""
DELETE FROM hrapp_trainingprogram
WHERE id = ?
""", (training_program_id, ))
return redirect(reverse('hrapp:training_program_list'))
# Check if this POST is for editing a book
if "actual_method" in form_data and form_data["actual_method"] == "PUT":
with sqlite3.connect(Connection.db_path) as conn:
db_cursor = conn.cursor()
db_cursor.execute(
"""
UPDATE hrapp_trainingprogram
SET title = ?,
start_date = ?,
end_date = ?,
capacity = ?
WHERE id = ?
""",
(
form_data["title"],
form_data["start_date"],
form_data["end_date"],
form_data["capacity"],
training_program_id,
),
)
return redirect(
reverse("hrapp:training_program_details",
kwargs={'training_program_id': training_program_id}))