mirror of
https://github.com/nottinghamtec/PyRIGS.git
synced 2026-01-17 13:32:15 +00:00
134 lines
4.3 KiB
Python
134 lines
4.3 KiB
Python
from django.shortcuts import render, get_object_or_404
|
|
from django.contrib.auth.decorators import login_required
|
|
from django.contrib.auth.mixins import LoginRequiredMixin
|
|
from django.http import HttpResponse, QueryDict
|
|
from django.core import serializers
|
|
from django.views import generic
|
|
from django.contrib.auth import views as auth_views
|
|
from django.urls import reverse_lazy, reverse
|
|
from django.db.models import Q
|
|
import datetime
|
|
from dateutil import parser
|
|
# import json
|
|
import simplejson as json
|
|
from assets import models, forms
|
|
|
|
|
|
class Login(auth_views.LoginView):
|
|
template_name = 'registration/login.html'
|
|
|
|
|
|
class Logout(auth_views.LogoutView):
|
|
template_name = 'registration/logout.html'
|
|
|
|
|
|
class PasswordChange(auth_views.PasswordChangeView):
|
|
template_name = 'registration/password_change.html'
|
|
|
|
def get_success_url(self):
|
|
return reverse_lazy('profile_detail', kwargs={'pk': self.request.user.id})
|
|
|
|
|
|
class Index(LoginRequiredMixin, generic.TemplateView):
|
|
template_name = 'index.html'
|
|
|
|
|
|
class AssetList(LoginRequiredMixin, generic.ListView):
|
|
model = models.Asset
|
|
template_name = 'asset_list.html'
|
|
paginate_by = 40
|
|
ordering = ['-pk']
|
|
|
|
|
|
class AssetDetail(LoginRequiredMixin, generic.DetailView):
|
|
model = models.Asset
|
|
template_name = 'asset_update.html'
|
|
|
|
|
|
# class AssetCreate(LoginRequiredMixin, generic.TemplateView):
|
|
# fields = '__all__'
|
|
# template_name = 'asset_update.html'
|
|
# # success_url = reverse_lazy('asset_list')
|
|
|
|
|
|
class AssetEdit(LoginRequiredMixin, generic.TemplateView):
|
|
template_name = 'asset_update.html'
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super(AssetEdit, self).get_context_data(**kwargs)
|
|
if self.kwargs:
|
|
context['object'] = get_object_or_404(models.Asset, pk=self.kwargs['pk'])
|
|
context['form'] = forms.AssetForm
|
|
# context['asset_names'] = models.Asset.objects.values_list('asset_id', 'description').order_by('-date_acquired')[]
|
|
context['edit'] = True
|
|
|
|
return context
|
|
|
|
|
|
@login_required()
|
|
def asset_update(request):
|
|
context = dict()
|
|
|
|
if request.method == 'POST' and request.is_ajax():
|
|
defaults = QueryDict(request.POST['form'].encode('ASCII')).dict()
|
|
defaults.pop('csrfmiddlewaretoken')
|
|
|
|
asset_pk = int(defaults.pop('id'))
|
|
|
|
if defaults['date_acquired']:
|
|
defaults['date_acquired'] = parser.parse(defaults.pop('date_acquired'))
|
|
else:
|
|
defaults['date_acquired'] = None
|
|
|
|
if defaults['date_sold']:
|
|
defaults['date_sold'] = parser.parse(defaults.pop('date_sold'))
|
|
else:
|
|
defaults['date_sold'] = None
|
|
|
|
# if defaults['parent']:
|
|
# defaults['parent'] = models.Asset.objects.get(asset_id=defaults.pop('parent'))
|
|
|
|
form = forms.AssetForm(defaults)
|
|
context['valid'] = form.is_valid()
|
|
context['errors'] = form.errors.as_json()
|
|
|
|
if asset_pk == 0:
|
|
asset = models.Asset.objects.create(**form.cleaned_data)
|
|
else:
|
|
asset, created = models.Asset.objects.update_or_create(pk=asset_pk, defaults=form.cleaned_data)
|
|
|
|
context['url'] = reverse('asset_detail', args=[asset.pk])
|
|
|
|
return HttpResponse(json.dumps(context), content_type='application/json')
|
|
|
|
|
|
@login_required()
|
|
def asset_delete(request):
|
|
context = dict()
|
|
if request.method == 'POST' and request.is_ajax():
|
|
asset = get_object_or_404(models.Asset, pk=request.POST.get('asset_id', None))
|
|
asset.delete()
|
|
|
|
context['url'] = reverse('asset_list')
|
|
|
|
return HttpResponse(json.dumps(context), content_type='application/json')
|
|
|
|
|
|
@login_required()
|
|
def asset_filter(request):
|
|
context = dict()
|
|
|
|
if request.method == 'POST' and request.is_ajax():
|
|
defaults = QueryDict(request.POST['form'].encode('ASCII')).dict()
|
|
defaults.pop('csrfmiddlewaretoken')
|
|
|
|
context['object_list'] = models.Asset.objects.filter(
|
|
Q(pk__icontains=defaults.get('asset_id')) |
|
|
Q(asset_id__icontains=defaults.get('asset_id'))
|
|
)
|
|
|
|
if request.POST.get('sender', None) == 'asset_update':
|
|
return render(request, template_name='asset_update_search_results.html', context=context)
|
|
else:
|
|
return render(request, template_name='asset_list_table_body.html', context=context)
|