def add(self):
post = self.context.POST
bus = bus_ui(self.context)
bus_choises = self.u.choises_gen(bus.buses(), 'busDescription')
company = company_ui(self.context)
company_choises = self.u.choises_gen(company.companies(), 'name')
caseform = CaseForm(post, bus_choises=bus_choises, company_choises=company_choises)
personform = PersonForm(self.context.POST)
if self.context.method == "POST":
if caseform.is_valid() and personform.is_valid():
driver = Person.objects.get_or_create(
firstName = personform.cleaned_data['firstName'],
middleName = personform.cleaned_data['middleName'],
lastName = personform.cleaned_data['lastName'],
person_position = u'Driver'
)[0]
bus = Bus.objects.get(id = caseform.cleaned_data['bus'])
company = Company.objects.get(id = caseform.cleaned_data['company'])
intruder = Intruder.objects.get_or_create(
company = company,
driver = driver,
bus = bus
)[0]
# start = BusStation.objects.get_or_create(
# adress = Adress(),
# name = ''
# )[0]
# end = BusStation.objects.get_or_create(
# adress = Adress(),
# name = ''
# )[0]
date = datetime.datetime.now()
case = Case.objects.create(
intruder = intruder,
dateTime = date
)
cleared_form = {'case': caseform, 'person': PersonForm()}
return cleared_form
def buses(request):
response_context = {
'title': 'Buses',
'login_form': LoginForm(),
'is_auth': False,
'active': 'bus',
}
if request.user.is_authenticated():
response_context['is_auth'] = request.user.is_authenticated()
response_context['username'] = request.user.username
if request.user.is_staff:
response_context['mode'] = 'admin'
ui = bus_ui(request)
genui = generic_ui(request)
response_context['city'] = genui.brand()
response_context['buses'] = ui.buses()
return direct_to_template(request, 'ui/bus/buses_page.html', response_context)
def add(request):
response_context = {
'title': 'Buses',
'login_form': LoginForm(),
'is_auth': False,
'active': 'bus',
}
if request.user.is_authenticated():
response_context['is_auth'] = request.user.is_authenticated()
response_context['username'] = request.user.username
if request.user.is_staff:
response_context['mode'] = 'admin'
ui = bus_ui(request)
genui = generic_ui(request)
response_context['city'] = genui.brand()
response_context['form'] = ui.add()
if request.method == 'POST':
return redirect('/buses/')
else:
return direct_to_template(request, 'ui/bus/bus_add.html', response_context)
def remove(request, id = 0):
ui = bus_ui(request)
ui.remove(id)
return redirect('/buses/')
