diff --git a/serverseeker/__init__.py b/serverseeker/__init__.py index a2a0ecd..4b1b322 100644 --- a/serverseeker/__init__.py +++ b/serverseeker/__init__.py @@ -18,10 +18,10 @@ class BadRequest(Exception): class ServerSeeker: - def __init__(self, token: str, endpoint: str = "https://api.serverseeker.net"): + def __init__(self, api_key: str, endpoint: str = "https://api.serverseeker.net"): self.endpoint = endpoint self.session = requests.Session() - self.session.headers.update({"Authorization": "Bearer " + token}) + self.session.headers.update({"Authorization": "Bearer " + api_key}) self.session.headers.update({"User-Agent": f"ServerSeeker.py/{__version__} +https://git.magmaus3.eu.org/magmaus3/serverseeker.py"}) def server_info(self, ip: str, port: int = 25565) -> dict: @@ -44,7 +44,7 @@ class ServerSeeker: def servers(self, country_code: Optional[str] = None, max_players: Optional[int] = None, - online_players: Optional[tuple[Union[int, str], Union[int, str]]] = None, + online_players: Optional[Union[tuple[Union[int, str], Union[int, str]], int]] = None, protocol: Optional[int] = None, software: Optional[str] = None, cracked: Optional[bool] = None, @@ -52,13 +52,12 @@ class ServerSeeker: asn: Optional[int] = None, ignore_modded: Optional[bool] = None, only_bungeespoofable: Optional[bool] = None - ) -> list: """ Gives you a list of servers (with matching criteria). Limited to 100 results. """ rq = self.session.post( - urljoin(self.endpoint, "/server_info"), json={ + urljoin(self.endpoint, "/servers"), json={ "asn": asn, "country_code": country_code, "cracked": cracked, @@ -72,12 +71,15 @@ class ServerSeeker: } ) if not rq.ok: - if rq.status_code == 400: - raise BadRequest(rq.json()["error"]) - elif rq.status_code == 429: - raise LimitExceeded(rq.json()["error"]) - elif rq.status_code == 504: - raise QueryTimeout(rq.json()["error"]) + try: + if rq.status_code == 400: + raise BadRequest(rq.json()["error"]) + elif rq.status_code == 429: + raise LimitExceeded(rq.json()["error"]) + elif rq.status_code == 504: + raise QueryTimeout(rq.json()["error"]) + except: + raise Exception(rq.text) return rq.json() @@ -108,15 +110,22 @@ class ServerSeeker: return rq.json() - def whereis(self, name: Optional[str], uuid: Optional[str]) -> dict: + def whereis(self, name: Optional[str] = None, uuid: Optional[str] = None) -> dict: """ Gives you a list of servers where a player was. Limited to 1000 results. **Attention: You must use name OR uuid as your parameter, not both!** """ - + if name and not uuid: + data = {"name": name} + elif uuid and not name: + data = {"uuid": uuid} + else: + raise ValueError("You can only provide name or uuid, not both") + rq = self.session.post( - urljoin(self.endpoint, "/whereis"), json={"name": "name", "uuid": uuid} + urljoin(self.endpoint, "/whereis"), json=data ) + if not rq.ok: if rq.status_code == 400: raise BadRequest(rq.json()["error"])