def add(self): post = self.context.POST route = route_ui(self.context) route_choises = self.u.choises_gen(route.routes(), 'number') form = BusForm(post, route_choises=route_choises) if self.context.method == "POST": if form.is_valid(): route = Route.objects.get_or_create(number = form.cleaned_data['route_hide'], city = self.user_profile.adress.city)[0] one_bus = Bus.objects.get_or_create(route = route, model = form.cleaned_data['model'], licensePlate = form.cleaned_data['licensePlate'])[0] one_bus.save() cleared_form = BusForm() cleared_form.fields['route'].choices = route_choises return cleared_form
def routes(request): response_context = { 'title': 'Route', 'login_form': LoginForm(), 'is_auth': False, 'active': 'route', } 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 = route_ui(request) genui = generic_ui(request) response_context['city'] = genui.brand() response_context['routes'] = ui.routes() return direct_to_template(request, 'ui/route/routes_page.html', response_context)
def add(request): response_context = { 'title': 'Route', 'login_form': LoginForm(), 'is_auth': False, 'active': 'route', } 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 = route_ui(request) genui = generic_ui(request) response_context['city'] = genui.brand() response_context['form'] = ui.add() if request.method == 'POST': return redirect('/routes/') else: return direct_to_template(request, 'ui/route/route_add.html', response_context)
def edit(self, id = 0): post = self.context.POST route = route_ui(self.context) route_choises = self.u.choises_gen(route.routes(), 'number') form = BusForm(post, route_choises=route_choises) bus = Bus.objects.get(id = id) if self.context.method == "POST": if form.is_valid(): route = Route.objects.get_or_create(number = form.cleaned_data['route_hide'], city = self.user_profile.adress.city)[0] bus.route = route bus.model = form.cleaned_data['model'] bus.licensePlate = form.cleaned_data['licensePlate'] bus.save() cleared_form = BusForm() cleared_form.fields['route'].choices = route_choises cleared_form.fields['route'].initial = bus.route.number cleared_form.fields['model'].initial = bus.model cleared_form.fields['licensePlate'].initial = bus.licensePlate return cleared_form
def remove(request, id = 0): ui = route_ui(request) ui.remove(id) return redirect('/routes/')