flyapps/fir_ser/api/views/apps.py

#!/usr/bin/env python
# -*- coding:utf-8 -*-
# project: 3月 
# author: liuyu
# date: 2020/3/4

from rest_framework.views import APIView
from api.utils.response import BaseResponse
from api.utils.auth import ExpiringTokenAuthentication
from rest_framework.response import Response
from rest_framework.viewsets import ModelViewSet
from django.db.models import Sum,aggregates
import os
from fir_ser import settings
from api.utils.randomstrings import make_from_user_uuid

import redis
from api.models import Apps,UserInfo,AppReleaseInfo
from api.utils.serializer import AppsSerializer,AppReleaseSerializer,UserInfoSerializer
from api.utils.app.analyze import delete_apps_storage,delete_apps_icon_storage
from rest_framework.pagination import PageNumberPagination, LimitOffsetPagination
REDIS_CONN = redis.Redis(decode_responses=True)


class AppsPageNumber(PageNumberPagination):
    page_size = 20  # 每页显示多少条
    page_size_query_param = 'size'  # URL中每页显示条数的参数
    page_query_param = 'page'  # URL中页码的参数
    max_page_size = None  # 最大页码数限制



class AppsView(APIView):

    authentication_classes = [ExpiringTokenAuthentication, ]

    def get(self, request):

        app_type = request.query_params.get("type", None)
        act_type = request.query_params.get("act", None)

        res = BaseResponse()
        res.hdata={"all_hits_count":0}
        res.hdata["upload_domain"] = request.user.domain_name
        res.hdata["ios_count"] = Apps.objects.filter(type=1,user_id=request.user).values('app_id').count()
        res.hdata["android_count"] = Apps.objects.filter(type=0,user_id=request.user).values('app_id').count()
        all_hits_obj = Apps.objects.filter(user_id=request.user).aggregate(count_hits=Sum('count_hits'))
        if all_hits_obj :
            count_hits=all_hits_obj.get("count_hits",0)
            if  count_hits:
                if  count_hits > 0:
                    request.user.all_download_times +=count_hits
                    request.user.save()
            else:
                count_hits = 0
            res.hdata["all_hits_count"]=count_hits
        else:
            res.hdata["all_hits_count"]=0
        res.hdata["android_today_hits_count"] = 0
        res.hdata["ios_today_hits_count"] = 0


        if app_type == "android":
            filter_data={"user_id":request.user,"type":0}

        elif app_type == "ios":
            filter_data={"user_id":request.user,"type":1}
        else:
            filter_data={"user_id":request.user}

        if act_type == "combo":
            filter_data["has_combo"]=None

        apps_obj = Apps.objects.filter(**filter_data)

        page_obj = AppsPageNumber()
        app_page_serializer = page_obj.paginate_queryset(queryset=apps_obj.order_by("-updated_time"), request=request, view=self)

        # app_serializer = AppsSerializer(apps_obj.order_by("-updated_time"),many=True)
        app_serializer = AppsSerializer(app_page_serializer,many=True)

        userserializer = UserInfoSerializer(request.user)
        res.userinfo={}
        res.has_next={}
        res.userinfo = userserializer.data
        res.data = app_serializer.data
        res.has_next = page_obj.page.has_next()
        return Response(res.dict)




class AppinfoView(APIView):
    authentication_classes = [ExpiringTokenAuthentication, ]
    def get(self, request,app_id):
        res = BaseResponse()
        if app_id:
            apps_obj = Apps.objects.filter(user_id=request.user,app_id=app_id).first()
            if apps_obj:
                app_serializer = AppsSerializer(apps_obj)
                res.data = app_serializer.data

            else:
                res.msg="未找到该应用"
                res.code=1003
            userserializer = UserInfoSerializer(request.user)
            res.userinfo = {}
            res.userinfo = userserializer.data
        return Response(res.dict)
    def delete(self,request,app_id):
        res = BaseResponse()
        if app_id:
            apps_obj = Apps.objects.filter(user_id=request.user,app_id=app_id).first()
            if apps_obj:
                for appreleaseobj in AppReleaseInfo.objects.filter(app_id=apps_obj).all():
                    delete_apps_storage(appreleaseobj.release_id,appreleaseobj.release_type)
                    appreleaseobj.delete()

                has_combo = apps_obj.has_combo
                if has_combo:
                    apps_obj.has_combo.has_combo=None
                apps_obj.delete()

        return Response(res.dict)

    def put(self,request,app_id):
        res = BaseResponse()
        if app_id:
            data = request.data

            has_combo = data.get("has_combo",None)
            if has_combo:
                actions = has_combo.get("action",None)
                hcombo_id = has_combo.get("hcombo_id",None)

                if actions and hcombo_id:
                    has_combo = Apps.objects.filter(user_id=request.user,app_id=hcombo_id)
                    apps_obj = Apps.objects.filter(user_id=request.user, app_id=app_id)
                    try:
                        if actions == "UNCOMBO":
                            if apps_obj.filter(has_combo=has_combo.first()).first():
                                apps_obj.update(**{"has_combo":None})
                                has_combo.update(**{"has_combo":None})

                        elif actions == "COMBO":
                            apps_obj.update(**{"has_combo": has_combo.first()})
                            has_combo.update(**{"has_combo": apps_obj.first()})
                        else:
                            pass
                    except Exception as e:
                        res.code = 1004
                        res.msg = "该应用已经关联"
            else:
                try:
                    apps_obj = Apps.objects.filter(user_id=request.user, app_id=app_id).first()
                    apps_obj.description= data.get("description",apps_obj.description)
                    apps_obj.short= data.get("short",apps_obj.short)
                    apps_obj.name= data.get("name",apps_obj.name)
                    apps_obj.save()
                except Exception as e:
                    res.code = 1005
                    res.msg = "短连接已经存在"

        return Response(res.dict)


    def post(self, request,app_id):
        res = BaseResponse()

        # 获取多个file
        files = request.FILES.getlist('file', None)
        for file_obj in files:
            # 将文件缓存到本地后上传
            try:
                app_type = file_obj.name.split(".")[-1]
                if app_type in ['png','jpeg','jpg']:
                    #上传图片
                    pass
                else:
                    raise
            except Exception as e:
                res.code = 1003
                res.msg = "错误的类型"
                return Response(res.dict)

            if app_id:
                apps_obj = Apps.objects.filter(user_id=request.user, app_id=app_id).first()
                if apps_obj:
                    release_obj = AppReleaseInfo.objects.filter(app_id = apps_obj,is_master=True).first()
                    if release_obj:

                        random_file_name = make_from_user_uuid(request.user)



                        old_file_name = os.path.basename(release_obj.icon_url)

                        local_file = os.path.join(settings.MEDIA_ROOT,"icons",random_file_name+ "." + old_file_name.split(".")[1])
                        # 读取传入的文件
                        try:
                            destination = open(local_file, 'wb+')
                            for chunk in file_obj.chunks():
                                # 写入本地文件
                                destination.write(chunk)
                            destination.close()
                        except Exception as e:
                            res.code = 1003
                            res.msg = "数据写入失败"
                            return Response(res.dict)
                        release_obj.icon_url = release_obj.icon_url.replace(old_file_name.split(".")[0],random_file_name)
                        release_obj.save()
                        delete_apps_icon_storage(old_file_name,'icons')

                    else:
                        res.code = 1003
                else:
                    res.code = 1003
        return Response(res.dict)


class AppReleaseinfoView(APIView):

    authentication_classes = [ExpiringTokenAuthentication, ]

    def get(self, request,app_id,act):
        res = BaseResponse()
        res.data={}
        if app_id:
            apps_obj = Apps.objects.filter(user_id=request.user,app_id=app_id).first()
            if apps_obj:
                app_serializer = AppsSerializer(apps_obj)
                res.data["currentapp"] = app_serializer.data

                app_release_obj = AppReleaseInfo.objects.filter(app_id=apps_obj).all().order_by("-created_time")
                app_release_serializer = AppReleaseSerializer(app_release_obj,many=True)
                res.data["release_apps"] = app_release_serializer.data

            else:
                res.msg="未找到该应用"
                res.code=1003
        return Response(res.dict)

    def delete(self,request,app_id,act):
        res = BaseResponse()
        if app_id:
            apps_obj = Apps.objects.filter(user_id=request.user,app_id=app_id).first()
            if apps_obj:
                apprelease_count = AppReleaseInfo.objects.filter(app_id=apps_obj).values("release_id").count()
                appreleaseobj =AppReleaseInfo.objects.filter(app_id=apps_obj,release_id=act).first()
                if not appreleaseobj.is_master :
                    delete_apps_storage(appreleaseobj.release_id,appreleaseobj.release_type)
                    appreleaseobj.delete()
                elif appreleaseobj.is_master and apprelease_count <2:
                    delete_apps_storage(appreleaseobj.release_id,appreleaseobj.release_type)
                    appreleaseobj.delete()

                    has_combo = apps_obj.has_combo
                    if has_combo:
                        apps_obj.has_combo.has_combo=None
                    apps_obj.delete()
                else:
                    pass

        return Response(res.dict)

    def put(self,request,app_id,act):
        res = BaseResponse()
        res.data={}
        if app_id:
            apps_obj = Apps.objects.filter(user_id=request.user, app_id=app_id).first()
            if apps_obj:
                appreleaseobj = AppReleaseInfo.objects.filter(app_id=apps_obj, release_id=act)

                make_master =  request.data.get("make_master",None)
                try:
                    if make_master and make_master == act:
                        AppReleaseInfo.objects.filter(app_id=apps_obj).update(**{"is_master":False})
                        appreleaseobj.update(**{"is_master":True})
                    else:
                        appreleaseobj.update(**{"changelog":request.data.get("changelog",appreleaseobj.first().changelog)})
                        binary_url = request.data.get("binary_url",None)
                        if binary_url !='':
                            if binary_url:
                                if not binary_url.startswith('http'):
                                    binary_url='http://%s'%binary_url
                            else:
                                binary_url =appreleaseobj.first().binary_url

                        print(binary_url)
                        appreleaseobj.update(**{"binary_url":binary_url})
                except Exception as e:
                    res.code = 1006
                    res.msg = "更新失败"
                    return Response(res.dict)


                app_serializer = AppsSerializer(apps_obj)
                res.data["currentapp"] = app_serializer.data

                app_release_obj = AppReleaseInfo.objects.filter(app_id=apps_obj).all().order_by("-created_time")
                app_release_serializer = AppReleaseSerializer(app_release_obj,many=True)
                res.data["release_apps"] = app_release_serializer.data

        return Response(res.dict)