def crud_event(id=0): form = EventForm() if form.validate_on_submit(): # POST, PUT if request.method == 'POST': # Create e = Event(name=form.name.data, start_date=form.from_date.data, end_date=form.end_date.data, city=form.city.date, state_abbr=form.state_abbr.data) db.session.add(e) db.session.commit() flash('Event created') return redirect(url_for('/crud_event')) elif request.method == 'PUT': # Update e = Event.query.filter_by(id=id) e.name = form.name.data e.start_date = form.start_date.data e.end_date = form.end_date.data e.city = form.city.data e.state_abbr = form.state_abbr.data db.session.commit() flash('Event updated') return redirect(url_for('/crud_event')) else: if request.method == 'GET': if id == 0: # Empty form to Create return render_template('/crud_event.html', form=form, action='Create', route='event') else: # Pre-populated form to Update form = EventForm(id) e = Event.query.filter_by(id=id).first_or_404() form.name.data = e.name form.start_date.data = e.start_date form.end_date.data = e.end_date form.city.data = e.city form.state_abbr.date = e.state_abbr return render_template('/crud_event.html', form=form, action='Update', route='event') elif request.method == 'DELETE': # Delete e = Event.query.filter_by(id=id).first_or_404() db.session.delete(e) db.session.commit() flash('Event deleted') return redirect(url_for('/crud_event'))
def create(userid): user = User.query.get(userid) data = request.get_json() event = Event() event.name = data.get('name') event.category = data.get('category') event.place = data.get('place') event.address = data.get('address') event.start_date = data.get('start_dadte') event.end_date = data.get('end_date') event.method = True if data.get('method') == 'true' else False event.owner = user db.session.add(event) db.session.commit() return jsonify(event.serialize())
def test_saving_and_retrieving_models(self): company = Company() company.name = 'PSL' company.email = '*****@*****.**' company.telephone = 7777777 company.save() first_room = Room() first_room.name = 'Peter Santamaria' first_room.company = company first_room.security_level = 2 first_room.description = 'Conference room' first_room.id = 'r00m1' first_room.save() second_room = Room() second_room.name = '201' second_room.company = company second_room.security_level = 1 second_room.description = 'Class room' second_room.id = 'r00m2' second_room.save() third_room = Room() third_room.name = 'rest' third_room.company = company third_room.security_level = 1 third_room.description = 'Rest room' third_room.id = 'r00m3' third_room.save() event = Event() event.name = 'My event' event.company = company event.start_date = datetime(2017, 3, 16) event.end_date = datetime(2017, 3, 21) event.event_id = '3v3nt' event.rooms.add(first_room) event.rooms.add(second_room) event.rooms.add(third_room) event.save() enduser = EndUser() enduser.id = 'u53r' enduser.name = 'Sebastian' enduser.last_name = 'Villegas' enduser.email = '*****@*****.**' enduser.save() first_permission = Permission() first_permission.user_id = enduser first_permission.event = event first_permission.id = '93rm151' first_permission.save() second_permission = Permission() second_permission.user_id = enduser second_permission.event = event second_permission.id = '93rm152' second_permission.save() saved_company = Company.objects.first() self.assertEqual(saved_company, company) saved_rooms = Room.objects.all() self.assertEqual(saved_rooms.count(), 3) self.assertEqual(saved_rooms[0], first_room) self.assertEqual(saved_rooms[1], second_room) self.assertEqual(saved_rooms[2], third_room) saved_event = Event.objects.first() self.assertEqual(saved_event, event) saved_enduser = EndUser.objects.first() self.assertEqual(saved_enduser, enduser) saved_permissions = Permission.objects.all() self.assertEqual(saved_permissions.count(), 2) self.assertEqual(saved_permissions[0], first_permission) self.assertEqual(saved_permissions[1], second_permission)