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