from django.shortcuts import render from django.contrib.auth import logout from django.views.generic.list import ListView from django.views.generic.detail import DetailView from rest_framework.authentication import TokenAuthentication from rest_framework.views import APIView from rest_framework.response import Response from rest_framework.permissions import IsAuthenticated from rest_framework import status, permissions from django.contrib.auth.models import User from django.contrib import messages from django.contrib.auth.mixins import LoginRequiredMixin, UserPassesTestMixin from .models import sistemas_por_cliente, DeviceHistory,Device from .serializers import DeviceSerializer from .permissions import HasAuthorizationHeader from django.utils import timezone import re from rest_framework.authtoken.models import Token from django.utils import timezone from datetime import timedelta from django.contrib.sessions.models import Session def get_logged_in_users(): sessions = Session.objects.filter(expire_date__gte=timezone.now()) # Get all non-expired tokens #tokens = Token.objects.filter(created__gte=timezone.now() - timedelta(hours=1)) tokens = Token.objects.all() # Get a list of logged-in user ids uid_list = [token.user_id for token in tokens] # Get the logged-in users users = User.objects.filter(id__in=uid_list) for user,session in zip(users,sessions): user.session_data = session.get_decoded() user.session_expire = session.expire_date return users class UsersConnectedList(UserPassesTestMixin,LoginRequiredMixin,ListView): model = User template_name= 'Sistemas/Usuarios/lista.html' def get_queryset(self): return get_logged_in_users() def test_func(self): res = self.request.user.groups.filter(name= 'admin_soft') if not res: messages.error(self.request, f'Lo sentimos. La página que buscas no está disponible, no cuentas con los permisos.') return res class SistemasXCliente_ListView(UserPassesTestMixin,LoginRequiredMixin, ListView): model = sistemas_por_cliente paginate_by = 5 template_name = 'Sistemas/Xclientes/lista.html' def test_func(self): res = self.request.user.groups.filter(name= 'admin_soft') if not res: messages.error(self.request, f'Lo sentimos. La página que buscas no está disponible, no cuentas con los permisos.') return res class SistemasXCliente_DetailView(UserPassesTestMixin,LoginRequiredMixin, DetailView): model = sistemas_por_cliente template_name= 'Sistemas/Xclientes/detail.html' def test_func(self): res = self.request.user.groups.filter(name= 'admin_soft') if not res: messages.error(self.request, f'Lo sentimos. La página que buscas no está disponible, no cuentas con los permisos.') return res ''' ========================= API Views ''' class RegisterDeviceView(APIView): permissions_classes = (permissions.AllowAny,) def post(self,request): try: serializer = DeviceSerializer(data=request.data,context={'request':request}) if serializer.is_valid(): serializer.save() return Response(serializer.data, status=status.HTTP_200_OK) else: return Response({'Error':f'{serializer.errors}','isError':True}, status=status.HTTP_200_OK) except Exception as e: return Response( {'Error':f'{e}','isError':True} , status=status.HTTP_200_OK) class AuthenticateDeviceView(APIView): authentication_classes= [TokenAuthentication] permissions_classes=[IsAuthenticated, HasAuthorizationHeader] def get(self, request): try: obj, created = DeviceHistory.objects.get_or_create( device=request.user.device, ip_address=request.META.get('REMOTE_ADDR'), ) obj.last_authentication=timezone.now() obj.save() device_data=DeviceSerializer(request.user.device).data if device_data.serializer.is_valid: return Response(device_data.serializer.data, status=status.HTTP_200_OK) else: return Response( {'Error':f'{device_data.serializer.errors}','isError':True} , status=status.HTTP_200_OK) except Exception as e: return Response( {'Error':f'{e}','isError':True} , status=status.HTTP_200_OK) class LogoutView(APIView): authentication_classes = (TokenAuthentication,) permission_classes = (IsAuthenticated,HasAuthorizationHeader,) def post(self, request): logout(request) return Response({'OK':'Dispositivo desautenticado'},status=200)