def test_add_project_to_db(client, clean_db): entry = Project() entry.client = "project client" entry.name = "project name" entry.job_number = "project job number" entry.timestamp = datetime.datetime.now() entry.last_active = datetime.datetime.now() db.session.add(entry) db.session.commit() db_entry = Project.query.first_or_404() assert entry == db_entry
def add(): form = AddProject() if form.validate_on_submit(): entry = Project() entry.job_number = form.job_number.data entry.client = form.client.data entry.name = form.project.data entry.timestamp = datetime.datetime.now() entry.company = current_user.company.id db.session.add(entry) db.session.commit() current_user.company.add_asset(entry.asset) db.session.commit() flash("Project added") return redirect(url_for("project.view", asset=entry.asset)) return render_template("project/add.html", form=form)