Source code for sofascore_wrapper.api

import aiohttp

BASE_URL = "https://www.sofascore.com/api/v1"



class SofascoreAPI:
    default_headers = {"User-Agent": "Mozilla/5.0"}
    
    def __init__(self, headers: dict = None):
        self.session = None
        self.headers = self.default_headers if not headers else headers


    async def _get(self, endpoint):
        if self.session is None:
            self.session = aiohttp.ClientSession(headers = self.headers)
        async with self.session.get(f"{BASE_URL}{endpoint}") as response:
            if response.status == 200:
                return await response.json()
            else:
                raise Exception(f"Failed to fetch {endpoint}: {response.status}")
            
    async def _raw_get(self, endpoint):
        if self.session is None:
            self.session = aiohttp.ClientSession(headers = self.headers)
        async with self.session.get(endpoint) as response:
            if response.status == 200:
                return await response.json()
            else:
                raise Exception(f"Failed to fetch {endpoint}: {response.status}")



    async def close(self):
        if self.session:
            await self.session.close()