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)
