diff --git a/shakestation.py b/shakestation.py index eac22e1..09fd20c 100644 --- a/shakestation.py +++ b/shakestation.py @@ -43,6 +43,27 @@ functions = { def main(params, authentication): return {"status": "Success"} + +def checkStaked(domain): + """ + Check if a domain is staked on Shakestation. + """ + response = requests.post(f"https://shakestation.io/api", + data={"action":"getStakedTLD","tld":domain}) + + if response.status_code != 200: + return False + + data = response.json() + + # Check if the domain is staked + if "data" not in data: + return False + + return data["data"] + + return False + def domain_info(params, authentication): domain = params.get("domain") if not domain: @@ -62,25 +83,27 @@ def domain_info(params, authentication): return {"info": "No type found in the data"} domainType = data["data"]["type"] - skippedTypes = ["auction"] score = data["data"].get("score", "No score available") - if domainType in skippedTypes: - return {"info": f"