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/')