app.py

import os
import traceback
from pathlib import Path
from datetime import datetime
from flask import current_app, jsonify, request
from flask_restful import Resource
from flask_restful.reqparse import RequestParser
from flask_jwt_extended import ( jwt_required, get_jwt_identity )
from werkzeug.datastructures import FileStorage
from werkzeug.utils import secure_filename
from application.config import config
from application.signal_manager import signalManager
from models.app import  postAppSchema, deleteAppSchema, getListAppSchema, getListAppsSchema, getAppSchema, putAppSchema
from webcreator.log import logger
from webcreator.response import ResponseCode, response_result

class AppResourceList(Resource):
    def __init__(self):
        # 特殊参数，即不是从json获取参数的接口，可以将这个注释打开
        self.parser = RequestParser()

    @jwt_required(locations=["headers"])
    def get(self):
        # 特殊参数，即不是从json获取参数的接口，可以将这个注释打开
        self.parser.add_argument('User-Agent', location='headers')
        self.parser.add_argument("app", type=str, location="args", nullable=True, required=False)
        self.parser.add_argument("scope", type=str, location="args", nullable=True, required=False)
        self.parser.add_argument("app_name", type=str, location="args", nullable=True, required=False)
        self.parser.add_argument("app_version", type=str, location="args", nullable=True, required=False)
        self.parser.add_argument("category", type=str, location="args", nullable=True, required=False)
        self.parser.add_argument("app_screen_size", type=str, location="args", nullable=True, required=False)
        self.parser.add_argument("app_arch", type=str, location="args", nullable=True, required=False)
        self.parser.add_argument("page", type=int, location="args", default=1)
        self.parser.add_argument("pageSize", type=int, location="args", default=15)
        args = self.parser.parse_args()

        try:
            jwt = get_jwt_identity()
            # data = getListAppSchema.load(args)
            data = dict()
            for key, value in args.items():
                if value != None:
                    data[key] = value
            result, message = signalManager.actionGetlistApp.emit(data, jwt)
            json_dumps = getListAppSchema.dump(result)
            if result:
                if isinstance(result, list):
                    json_dumps = result
                    return response_result(ResponseCode.HTTP_SUCCESS, data=json_dumps)
                else:
                    json_dumps = getListAppsSchema.dump(result.items)
                    for item in json_dumps:
                        item['file_dir'] = os.path.dirname(item['download_url'])
                    return response_result(ResponseCode.HTTP_SUCCESS, data=json_dumps, total=result.total, pageSize=args.pageSize)
            return response_result(message)
        except Exception as e:
            traceback.print_exc()
            current_app.logger.error(e)
            return response_result(ResponseCode.HTTP_SERVER_ERROR)

    @jwt_required(locations=["headers"])
    def post(self):
        self.parser.add_argument('User-Agent', location='headers')
        self.parser.add_argument("meta_data", type=str, location="form", default="{}", required=False)
        self.parser.add_argument("app_name", type=str, location="form", required=True)
        self.parser.add_argument("app_version", type=str, location="form", required=True)
        self.parser.add_argument("category", type=str, location="form", required=True)
        self.parser.add_argument("app_screen_size", type=str, location="form", required=True)
        self.parser.add_argument("app_arch", type=str, location="form", required=True)
        self.parser.add_argument("algorithm", type=str, location="form", required=True)
        self.parser.add_argument("remarks", type=str, location="form", required=True)
        self.parser.add_argument("logo", type=FileStorage, location="files", required=True)
        self.parser.add_argument("fileList", type=FileStorage, location="files", required=True)
        args = self.parser.parse_args()

        try:
            jwt = get_jwt_identity()

            # 获取request.files参数
            # json_payload = request.json
            params = postAppSchema.load(args)

            now_str = datetime.now().strftime("%Y%m%d%H%M%S")
            # 获取相对路径
            dirname = "{}-{}-{}-{}".format(params["app_name"], params["app_version"], params["category"], now_str)
            relative_path = Path(config.UPLOAD_ROOT_DIR)
            # 获取最终存储的绝对路径
            upload_path = Path(config.EPK_DIR).joinpath(dirname)

            if not upload_path.exists():
                os.makedirs(upload_path.resolve().as_posix())

            files = []
            # 应用logo
            logo = request.files.get("logo") # args.get('picture')
            if logo:
                filename = secure_filename(logo.filename)
                file_path = upload_path.joinpath(filename)
                logo.save(file_path)
                params.update({ "app_icon": file_path.relative_to(relative_path).as_posix() })

            # 应用源文件
            fileList = request.files.getlist('fileList')
            if fileList:
                upload_path = upload_path.joinpath("src")
                if not upload_path.exists():
                    os.mkdir(upload_path.resolve().as_posix())

                for f in fileList:
                    filename = secure_filename(f.filename)
                    file_path = upload_path.joinpath(filename)
                    f.save(file_path.resolve().as_posix())
                    files.append(file_path.relative_to(relative_path).as_posix())

            params.update({ "fileList": files, "epk_path": upload_path, 'real_ip': request.headers.get('X-Forwarded-For', '127.0.0.1') })
            result, message = signalManager.actionPostApp.emit(params, jwt)
            if result:
                return response_result(ResponseCode.HTTP_SUCCESS, data=result)
            return response_result(message)
        except Exception as e:
            traceback.print_exc()
            current_app.logger.error(e)
            return response_result(ResponseCode.HTTP_SERVER_ERROR)


class AppResource(Resource):
    def __init__(self):
        pass
        # 特殊参数，即不是从json获取参数的接口，可以将这个注释打开
        # self.parser = RequestParser()

    @jwt_required(locations=["headers"])
    def get(self, uuid):
        # 特殊参数，即不是从json获取参数的接口，可以将这个注释打开
        # self.parser.add_argument("page", type=int, location="args", default=1)
        # self.parser.add_argument("pageSize", type=int, location="args", default=15)
        # args = self.parser.parse_args()

        try:
            jwt = get_jwt_identity()
            result, message = signalManager.actionGetApp.emit(uuid, jwt)
            if result:
                return response_result(ResponseCode.HTTP_SUCCESS, data=result)
            return response_result(message)
        except Exception as e:
            traceback.print_exc()
            current_app.logger.error(e)
            return response_result(ResponseCode.HTTP_SERVER_ERROR)


    @jwt_required(locations=["headers"])
    def put(self, uuid):
        try:
            jwt = get_jwt_identity()
            json_payload = request.json
            data = putAppSchema.load(json_payload)
            result, message = signalManager.actionPutApp.emit(uuid, data, jwt)
            if result:
                return response_result(ResponseCode.HTTP_SUCCESS, data=result)
            return response_result(message)
        except Exception as e:
            traceback.print_exc()
            current_app.logger.error(e)
            return response_result(ResponseCode.HTTP_SERVER_ERROR)


    @jwt_required(locations=["headers"])
    def delete(self, uuid):
        try:
            json_payload = request.json
            # data = deleteAppSchema.load(json_payload)
            print("========>", uuid, json_payload)
            result, message = signalManager.actionDeleteApp.emit(uuid)
            if result:
                return response_result(ResponseCode.HTTP_SUCCESS, data=result)
            return response_result(message)
        except Exception as e:
            traceback.print_exc()
            current_app.logger.error(e)
            return response_result(ResponseCode.HTTP_SERVER_ERROR)