def edit_news(id):
form = JobsForm()
if request.method == "GET":
session = db_session.create_session()
job = session.query(Jobs).filter(Jobs.id == id).first()
if job and (current_user.id == job.creator or current_user.id == 1):
form.job.data = job.job
form.team_leader.data = job.team_leader
form.work_size.data = job.work_size
form.collaborators.data = job.collaborators
form.start_date.data = job.start_date
form.end_date.data = job.end_date
form.is_finished.data = job.is_finished
else:
abort(404)
if form.validate_on_submit():
session = db_session.create_session()
job = session.query(Jobs).filter(Jobs.id == id,
((Jobs.creator == current_user.id) |
(current_user.id == 1))).first()
if job:
session.delete(job)
job = Jobs()
job.team_leader = form.team_leader.data
job.job = form.job.data
job.work_size = form.work_size.data
job.collaborators = form.collaborators.data
job.start_date = form.start_date.data
job.end_date = form.end_date.data
job.is_finished = form.is_finished.data
session.add(job)
session.commit()
return redirect('/page')
else:
abort(404)
return render_template('edit_job.html',
title='Редактирование работы',
form=form)
def add_job():
form = JobsForm()
if request.method == 'POST' and form.validate_on_submit():
s = db_session.create_session()
job = Jobs()
job.team_leader = form.team_leader.data
job.user = s.query(User).filter(User.id == job.team_leader).first()
job.job = form.job.data
job.work_size = form.work_size.data
job.collaborators = form.collaborators.data
job.start_date = form.start_date.data
job.end_date = form.end_date.data
job.is_finished = form.is_finished.data
job.creator = current_user.id
categories = HazardCategory(name=form.categories.data)
job.categories.append(categories)
s.add(job)
s.commit()
return redirect('/page')
return render_template('add_job.html',
title='Добавление работы',
form=form)
self.assertEqual({"success": "OK"}, resp.json())
if __name__ == "__main__":
db_session.global_init("db/test_database.db")
job_1 = Jobs()
job_1.id = 1
job_1.team_leader = 2
job_1 = "Repair broken TV"
job_1.work_size = 1
job_1.collaborators = "1,2,3"
job_1.start_date = datetime.datetime.now()
job_1.end_date = datetime.datetime.now() + datetime.timedelta(1, 2, 3)
job_1.is_finished = False
job_2 = Jobs()
job_1.id = 2
job_1.team_leader = 1
job_1 = "Repair broken spaceship"
job_1.work_size = 5
job_1.collaborators = "2,3,4"
job_1.start_date = datetime.datetime.now()
job_1.end_date = datetime.datetime.now() + datetime.timedelta(1, 2, 3)
job_1.is_finished = True
