You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

62 lines
2.0 KiB
Python

from osinacore.api.utils import *
from osichat.models import *
from rest_framework.decorators import api_view
from .serializers import *
from django.shortcuts import get_object_or_404
from django.db.models import Q
from django.utils.dateparse import parse_date
@api_view(['GET'])
def get_chat_rooms(request):
chat_rooms = ChatRoom.objects.annotate(last_update=Max('chatmessage__date_sent')).order_by('-last_update', '-date_created')
chat_rooms_serializer = ChatRoomSerializer(chat_rooms, many=True)
return successRes(chat_rooms_serializer.data)
@api_view(['GET'])
def get_visitors(request):
start_date = request.query_params.get("start_date")
end_date = request.query_params.get("end_date")
if start_date:
start_date = parse_date(start_date)
if end_date:
end_date = parse_date(end_date)
if start_date and end_date: # Range of time
visitors = Visitor.objects.filter(Q(visitorlog__visit_date__date__range=[start_date, end_date]) | Q(visitorlog__left_date__date__range=[start_date, end_date])).distinct()
elif start_date and not end_date: # Specific date
visitors = Visitor.objects.filter(Q(visitorlog__visit_date__date=start_date) | Q(visitorlog__left_date__date=start_date)).distinct()
else: # Active visitors
visitors = Visitor.objects.filter(visitorlog__left_date__isnull=True).distinct()
visitors_count = visitors.count()
visitors_data = []
for visitor in visitors:
visitor_data = {
"id": visitor.id,
"ip": visitor.ip_address,
"flag": visitor.flag_image_url,
"is_online": visitor.is_online,
"duration": visitor.total_duration,
"last_log": VisitorLog.objects.filter(visitor=visitor).last().title
}
visitors_data.append(visitor_data)
response_data = {
"count": visitors_count,
"visitors": visitors_data
}
return successRes(response_data)