wg-backend-django/wg_api/views.py

from django.shortcuts import render
from django.http import HttpResponse
# from django.core import serializer
from rest_framework import viewsets
from rest_framework.views import APIView
from rest_framework.authentication import SessionAuthentication, BasicAuthentication
from rest_framework.permissions import IsAuthenticated
from rest_framework.response import Response
from rest_framework.permissions import IsAdminUser
from .serializers import wgSerializer 
from .models import user_model
from .utils.html_parser import get_logs, get_profiles
import json
from .utils.statistics import count_clients, count_traffic, count_connection, count_percentage, count_broken
from .utils.reqs_handler import req_edit_state, req_edit_user, req_apply_sys
from django.middleware.csrf import get_token
from django.views.decorators.csrf import csrf_exempt

class wgLogin(APIView):
    authentication_classes = [SessionAuthentication, BasicAuthentication]
    permission_classes = [IsAuthenticated]

    def get(self, request, format=None):
        content = {
            'user': str(request.user),  # `django.contrib.auth.User` instance.
            'auth': str(request.auth),  # None
        }
        return Response(content)

# Create your views here.
class wgView(viewsets.ModelViewSet):

    # queryset =  user_model.objects.all()
    queryset =  user_model.objects.all()
    serializer_class = wgSerializer

    # print(">"*100," ",inf_data)

    def get(self, request):
        return self.inf_connector()
       # iserializer = wgSerializer
      # queryset = user_model.objects.all()
      # serializers_class = wgSerializer
      # serializer = wgSerializer
      # permission_classes = [IsAdminUser]

def wgViewLogs(request):

    param = request.GET.get('param', '')

    data = get_logs()
    data = json.loads(data)

    final_data = [] 


    if  param == "d2201":
        print("x"*100)
        i = 1
        for x in data:
            if "DTS22_PNUP_01" in str(x["wgName"]):
                print(x)
                x["data_id"]=i
                final_data.append(x)
                i = i+1
    elif  param == "d2202":
        print("x"*100)
        i = 1
        for x in data:
            if "DTS22_PNUP_02" in str(x["wgName"]):
                print(x)
                x["data_id"]=i
                final_data.append(x)
                i = i+1
    elif  param == "d2203":
        print("x"*100)
        i = 1
        for x in data:
            if "DTS22_PNUP_03" in str(x["wgName"]):
                print(x)
                x["data_id"]=i
                final_data.append(x)
                i = i+1
    elif  param == "d2204":
        print("x"*100)
        i = 1
        for x in data:
            if "DTS22_PNUP_04" in str(x["wgName"]):
                print(x)
                x["data_id"]=i
                final_data.append(x)
                i = i+1
    elif  param == "d2205":
        print("x"*100)
        i = 1
        for x in data:
            if "DTS22_PNUP_05" in str(x["wgName"]):
                print(x)
                x["data_id"]=i
                final_data.append(x)
                i = i+1
    elif  param == "d2206":
        print("x"*100)
        i = 1
        for x in data:
            if "DTS22_TELU" in str(x["wgName"]):
                print(x)
                x["data_id"]=i
                final_data.append(x)
                i = i+1
    elif  param == "GCI":
        print("x"*100)
        i = 1
        for x in data:
            if "GCI" in str(x["wgName"]):
                x["data_id"]=i
                print(x)
                final_data.append(x)
                i = i+1
    elif  param == "META":
        print("x"*100)
        i = 1 
        for x in data:
            if "NIS" in str(x["wgName"]):
                print(x["data_id"])
                x["data_id"]=i
                print(x["data_id"])
                final_data.append(x)
                i = i+1
    else:
        final_data = data

    final_data = json.dumps(final_data) 

    return HttpResponse(final_data, content_type='application')

def wgViewClients(request):

    param = request.GET.get('param', '')
    filter = request.GET.get('filter', '')

    data = get_profiles()
    data = json.loads(data)

    
    final_data = []

    if filter :
        filter = str(filter).split(" ")
        for x in filter:
            print("-> ",x)
        final_data = data
        if not (filter[0] == "" and filter[1] == "undefined" and filter[2] == "undefined" and filter[3] == "undefined"):
            if filter[0] != "":
                f_name = filter[0]
                i = 0
                tmp_data = []
                for x in final_data:
                    if str(f_name).lower() in str(x["name"]).lower():
                        x["data_id"]=i
                        tmp_data.append(x)
                        i = i+1
                final_data = tmp_data

            if filter[1] != "undefined":
                f_orgs = filter[1]
                i = 0
                tmp_data = []
                for x in final_data:
                    if str(f_orgs).lower() in str(x["name"]).lower():
                        x["data_id"]=i
                        tmp_data.append(x)
                        i = i+1
                final_data = tmp_data
                        
            if filter[2] != "undefined":
                i = 0
                f_email = filter[2]
                tmp_data = []

                if f_email == "" :
                    print("Empty mails...")
                    for x in final_data:
                        if str(x["email"]).lower() == "":
                            x["data_id"]=i
                            tmp_data.append(x)
                            i = i+1
                else:
                    for x in final_data:
                        if str(f_email).lower() in str(x["email"]).lower():
                            x["data_id"]=i
                            tmp_data.append(x)
                            i = i+1

                final_data = tmp_data

            if filter[3] != "undefined":
                i = 0
                f_state = filter[3]
                tmp_data = []

                if f_state == "" :
                    print("Empty State...")
                    for x in final_data:
                        if str(x["enabled"]).lower() == "":
                            x["data_id"]=i
                            tmp_data.append(x)
                            i = i+1
                else:
                    for x in final_data:
                        if str(f_state).lower() in str(x["enabled"]).lower():
                            x["data_id"]=i
                            tmp_data.append(x)
                            i = i+1

                final_data = tmp_data

        else:
            final_data = data

        final_data = json.dumps(final_data) 

    elif len(param)>0:

        if param == "email":
            groups = []
            for x in data:
                if str("@").lower() in str(x["email"]).lower():
                    tmp_val = x["email"]
                    tmp_val = str(tmp_val).split("@")[1]
                    tmp_val = tmp_val.lower()
                    if {"email": tmp_val} not in groups:
                        # print("-> ",tmp_val)
                        groups.append({"email": tmp_val})

            final_data = json.dumps(groups)
              
        if  param == "orgs":
            groups = []
            for x in data:
                if str("-").lower() in str(x["name"]).lower():
                    tmp_val = x["name"]
                    tmp_val = str(tmp_val).split("-")[0]
                    tmp_val = tmp_val.upper()

                    if {"orgs": tmp_val} not in groups:
                        # print("-> ",tmp_val)
                        groups.append({"orgs": tmp_val})

            final_data = json.dumps(groups)

        if  param == "stats":
            data_logs = get_logs()
            data = json.loads(data_logs)

            sum_data = count_clients(data)
            connected_data = count_connection(data)
            traffic_data = len(data)

            percentage_data = count_percentage(data)
            broken_data = count_broken(data)

            traffic_data = count_traffic(data)
            if traffic_data > 1000000000:
                traffic_data="{:.2f}".format(traffic_data/1000000000)+" GB"
            else:
                traffic_data="{:.2f}".format(traffic_data/1000000)+" MB"

            data = {}

            data["clienttotal"]=sum_data
            data["traffictotal"]=traffic_data
            data["connectedtotal"]=connected_data
            data["percentagetotal"]=percentage_data
            data["brokentotal"]=broken_data
            
            final_data = json.dumps(data)

    else :
        final_data = json.dumps(data)
     
    # print(request.GET.get('param', ''))

    return HttpResponse(final_data, content_type='application')

def get_ApplyConfig(request):

    # id = request.GET.get('id', '')
    # state = request.GET.get('state', '')
    # print("Edit Request...")

   req_response = req_apply_sys() 

   return HttpResponse(req_response)

def get_EditState(request):

    id = request.GET.get('id', '')
    state = request.GET.get('state', '')
    print("Edit Request...")

    if state and id:
        if id != "bulk":
            print("[ Edit ]=======")
            print(id)
            print(state)

            req_edit_state(id, state)
        else:
            print("[ Bulk handler ]=======")
            print(request.body)

            target_ids = request.body

            for x in json.loads(target_ids):
                req_edit_state(str(x), state)
                print(x)
    
    final_data = json.dumps({"state":"Success"})

    return HttpResponse(final_data, content_type='application')

# def get_Cookie(request):
# 
#     return get_token()

# @csrf_exempt
def get_EditClient(request):

    my_token = get_token()

    print("TOKEN ...")
    print(my_token)

    id = request.GET.get('id', '')

    if request.method == "POST":
        print(request)

    if id:
        print(id)
        ret_val = req_edit_user(id)


    else:
        ret_val = {"data": "Failed"}

    # ret_val = json.dumps(ret_val)

    ret_val = ret_val["Client"]

    ret_val = json.dumps(ret_val)


    return HttpResponse(ret_val, content_type='application')