[VIRTS-2000] Obfuscators v2 API

app/api/v2/__init__.py

@@ -33,4 +33,7 @@ def make_app(services):
     from .handlers.agent_api import AgentApi
     AgentApi(services).add_routes(app)
 
+    from .handlers.obfuscator_api import ObfuscatorApi
+    ObfuscatorApi(services).add_routes(app)
+
     return app

app/api/v2/handlers/obfuscator_api.py

@@ -0,0 +1,33 @@
+import aiohttp_apispec
+from aiohttp import web
+
+from app.api.v2.handlers.base_object_api import BaseObjectApi
+from app.api.v2.managers.base_api_manager import BaseApiManager
+from app.api.v2.schemas.base_schemas import BaseGetAllQuerySchema, BaseGetOneQuerySchema
+from app.objects.c_obfuscator import Obfuscator, ObfuscatorSchema
+
+
+class ObfuscatorApi(BaseObjectApi):
+    def __init__(self, services):
+        super().__init__(description='obfuscator', obj_class=Obfuscator, schema=ObfuscatorSchema, ram_key='obfuscators',
+                         id_property='name', auth_svc=services['auth_svc'])
+        self._api_manager = BaseApiManager(data_svc=services['data_svc'], file_svc=services['file_svc'])
+
+    def add_routes(self, app: web.Application):
+        router = app.router
+        router.add_get('/obfuscators', self.get_obfuscators)
+        router.add_get('/obfuscators/{name}', self.get_obfuscator_by_name)
+
+    @aiohttp_apispec.docs(tags=['obfuscators'])
+    @aiohttp_apispec.querystring_schema(BaseGetAllQuerySchema)
+    @aiohttp_apispec.response_schema(ObfuscatorSchema(many=True, partial=True))
+    async def get_obfuscators(self, request: web.Request):
+        sources = await self.get_all_objects(request)
+        return web.json_response(sources)
+
+    @aiohttp_apispec.docs(tags=['obfuscators'])
+    @aiohttp_apispec.querystring_schema(BaseGetOneQuerySchema)
+    @aiohttp_apispec.response_schema(ObfuscatorSchema(partial=True))
+    async def get_obfuscator_by_name(self, request: web.Request):
+        source = await self.get_object(request)
+        return web.json_response(source)

app/objects/c_obfuscator.py

@@ -13,6 +13,10 @@ class ObfuscatorSchema(ma.Schema):
     description = ma.fields.String()
     module = ma.fields.String()
 
+    @ma.post_load
+    def build_obfuscator(self, data, **kwargs):
+        return None if kwargs.get('partial') is True else Obfuscator(**data)
+
 
 class Obfuscator(FirstClassObjectInterface, BaseObject):
     schema = ObfuscatorSchema()