def addPlacement(companyId): form = PlacementForm() form['csrf_token'].data = request.cookies['csrf_token'] if form.validate_on_submit(): newPlacement = Placement( contractorId_fk=form.data['contractorId'], companyId_fk=companyId, companyContactId_fk=form.data['companyContactId'], startDate=datetime.strptime(form.data['startDate'], '%Y-%m-%d %H:%M:%S'), endDate=datetime.strptime(form.data['endDate'], '%Y-%m-%d %H:%M:%S')) db.session.add(newPlacement) db.session.commit() placements = Placement.query.order_by(Placement.startDate).all() placementTableStruct = createContractorPlacementTableInfo( [placement.to_dict() for placement in placements]) return {"placements": placementTableStruct}
def seed_placements(): pl1 = Placement(contractorId_fk=2, companyId_fk=1, companyContactId_fk=1, startDate=datetime.strptime('12/20/20', '%m/%d/%y'), endDate=datetime.strptime('12/24/20', '%m/%d/%y')) pl2 = Placement(contractorId_fk=2, companyId_fk=1, companyContactId_fk=1, startDate=datetime.strptime('12/19/20', '%m/%d/%y'), endDate=datetime.strptime('12/19/20', '%m/%d/%y')) pl3 = Placement(contractorId_fk=2, companyId_fk=2, companyContactId_fk=5, startDate=datetime.strptime('12/29/20', '%m/%d/%y'), endDate=datetime.strptime('12/29/20', '%m/%d/%y')) pl4 = Placement(contractorId_fk=5, companyId_fk=2, companyContactId_fk=5, startDate=datetime.strptime('12/21/20', '%m/%d/%y'), endDate=datetime.strptime('12/23/20', '%m/%d/%y')) pl5 = Placement(contractorId_fk=5, companyId_fk=1, companyContactId_fk=4, startDate=datetime.strptime('12/27/20', '%m/%d/%y'), endDate=datetime.strptime('12/28/20', '%m/%d/%y')) pl6 = Placement(contractorId_fk=5, companyId_fk=1, companyContactId_fk=3, startDate=datetime.strptime('12/29/20', '%m/%d/%y'), endDate=datetime.strptime('12/30/20', '%m/%d/%y')) db.session.add(pl1) db.session.add(pl2) db.session.add(pl3) db.session.add(pl4) db.session.add(pl5) db.session.add(pl6) db.session.commit()