dream

1
#!/usr/bin/env python
2
import asyncio
3
import logging
4
from typing import Callable, Dict
5

6
import requests
7

8
logging.basicConfig(format="%(asctime)s - %(name)s - %(levelname)s - %(message)s", level=logging.INFO)
9
logger = logging.getLogger(__name__)
10

11
headers = {"Content-Type": "application/json;charset=utf-8"}
12

13

14
class BatchConnector:
15
    def __init__(self, url: str):
16
        self._url = url
17

18
    async def send(self, payload: Dict, callback: Callable):
19
        emotion_result = requests.request(url=self._url, headers=headers, json=payload["payload"], method="POST").json()
20
        asyncio.create_task(callback(task_id=payload["task_id"], response={"batch": emotion_result}))
21