chat2api/utils/Client.py

import random

from curl_cffi.requests import AsyncSession


class Client:
    def __init__(self, proxy=None, timeout=15, verify=True):
        self.proxies = {
            "http": proxy,
            "https": proxy,
        }
        self.timeout = timeout
        self.verify = verify
        self.impersonate = random.choice(["chrome", "safari", "safari_ios"])
        self.session = AsyncSession(proxies=self.proxies, timeout=self.timeout, verify=self.verify)

    async def post(self, *args, **kwargs):
        r = await self.session.post(*args, impersonate=self.impersonate, **kwargs)
        return r

    async def post_stream(self, *args, headers=None, cookies=None, **kwargs):
        if self.session:
            headers = headers or self.session.headers
            cookies = cookies or self.session.cookies
            await self.session.close()
            self.session = None
        self.session = AsyncSession(proxies=self.proxies, timeout=self.timeout, verify=self.verify)
        r = await self.session.post(*args, headers=headers, cookies=cookies, impersonate=self.impersonate, **kwargs)
        return r

    async def get(self, *args, **kwargs):
        r = await self.session.get(*args, impersonate=self.impersonate, **kwargs)
        return r

    async def request(self, *args, **kwargs):
        r = await self.session.request(*args, impersonate=self.impersonate, **kwargs)
        return r

    async def put(self, *args, **kwargs):
        r = await self.session.put(*args, impersonate=self.impersonate, **kwargs)
        return r

    async def close(self):
        try:
            await self.session.close()
            self.session = None
        except Exception:
            pass