django_coreprotect/activity.py

from django_coreprotect.models import SessionResult, ActivityResult
from django_coreprotect.utils import safe_int
from datetime import datetime
from typing import List


class ActivityForm:

    def __init__(self):
        self.players = ""
        self.date_from = ""
        self.date_to = ""


def activity_data(request):
    request_data = request.GET
    form = ActivityForm()

    # Players
    form.players = request_data["players"] if "players" in request_data else ""

    # Date and Time
    form.date_from = request_data["date_from"] if "date_from" in request_data else ""
    form.date_to = request_data["date_to"] if "date_to" in request_data else ""

    return form


def activity_results(form) -> List[ActivityResult]:
    query = ""

    players_clause = ""
    if form.players:
        players = []
        for player in form.players.split(","):
            players.append(player.strip())
        clause = " OR ".join([f"cu.user LIKE '%%{p}%%'" for p in players])
        players_clause = f" WHERE ({clause}) "

    time_clause = ""
    if form.date_from or form.date_to:
        df, dt = form.date_from, form.date_to
        if form.date_from and not form.date_to:
            dt = datetime.now().timestamp()
        if form.date_to and not form.date_from:
            df = datetime.now().timestamp()
        time_clause = " HAVING unix BETWEEN {} AND {} ".format(df, dt)

    if players_clause or time_clause:
        query = f'''SELECT 
                0 AS id, cs.time AS unix, cu.user AS player, cs.action 
                FROM co_session cs 
                JOIN co_user cu ON cs.user = cu.rowid 
                {players_clause}
                {time_clause}
                '''

    if query:
        sessions = SessionResult.objects.raw(query)
        activity = {}
        last_session = {}
        for session in sessions:
            time = -safe_int(session.unix) if session.action == 1 else safe_int(session.unix)
            if session.player in activity:
                if last_session[session.player] == session.action:
                    continue
                activity[session.player] += time
                last_session[session.player] = session.action
            else:
                activity[session.player] = time
                last_session[session.player] = session.action
        results = []
        for player, time in activity.items():
            results.append(ActivityResult(player, time))
        return results
    return []