# controller/controller.py
from flask_restx import Namespace, Resource, Api, fields
from service.service import Service
# 创建命名空间
ns = Namespace('sdk', description='SDK API')

#控制器层

@ns.route('/initialize')
class handler(Resource):
    @ns.doc(description='半实物初始化')
    def get(self):
        """半实物初始化"""
        return Service.initialize_usrp()


@ns.route('/status')
class handler(Resource):
    @ns.doc(description='半实物状态获取')
    def get(self):
        """半实物状态获取"""
        return Service.get_sdr_status()

@ns.route('/data')
class handler(Resource):
    @ns.doc(description='基于干扰策略、抗干扰策略获取数据')
    @ns.expect(ns.model('/data-model', {
    'jamming_policy': fields.String(required=True, description='干扰策略'),
    'anti_jamming_policy': fields.String(required=True, description='抗干扰策略')
    }))
    def post(self):
        """基于干扰策略、抗干扰策略获取数据"""
        args = self.api.payload
        return Service.data(args)


# @ns.route('/demo')
# class handler(Resource):
#     @ns.doc(description='测试')
#     def get(self):
#         return service.demo()