mcaptcha-api-py/scripts/create_captcha.py

114 lines
2.7 KiB
Python

# SPDX-FileCopyrightText: 2023 Aravinth Manivannan <realaravinth@batsense.net>
#
# SPDX-License-Identifier: Apache-2.0
# SPDX-License-Identifier: MIT
import sys
from urllib.parse import urlparse, urlunparse, urlencode, parse_qsl
from requests import Session
def clean_url(url):
parsed = urlparse(url)
return urlunparse((parsed.scheme, parsed.netloc, "", "", "", ""))
def set_path(url: str, path: str) -> str:
parsed = urlparse(url)
return urlunparse((parsed.scheme, parsed.netloc, path, "", "", ""))
def help():
print("Usage: create_captcha.py <option>")
print("OPTIONS:\ncreate_captcha <mcaptcha_instance_url> <username> <password>")
print("get_secret <mcaptcha_instance_url> <username> <password>")
print("widget_url <mcaptcha_instance_url> <sitekey>")
c = Session()
if len(sys.argv) < 1:
help()
exit(1)
option = sys.argv[1]
def login():
if len(sys.argv) < 4:
help()
exit(1)
instance_url = clean_url(sys.argv[2])
username = sys.argv[3]
password = sys.argv[4]
url = set_path(instance_url, "/api/v1/signin")
payload = {"login": username, password: password}
resp = c.post(url, json=payload)
assert resp.status_code == 200
def create_captcha():
instance_url = clean_url(sys.argv[2])
username = sys.argv[3]
password = sys.argv[4]
levels = [
{"difficulty_factor": 50, "visitor_threshold": 50},
{"difficulty_factor": 500, "visitor_threshold": 500},
]
payload = {
"levels": levels,
"duration": 30,
"description": "create_captcha_test_script",
"publish_benchmarks": False,
}
url = set_path(instance_url, "/api/v1/mcaptcha/create")
resp = c.post(url, json=payload, cookies=c.cookies.get_dict())
assert resp.status_code == 200
return resp.json()["key"]
def get_secret():
if len(sys.argv) < 3:
help()
exit(1)
instance_url = clean_url(sys.argv[2])
username = sys.argv[3]
password = sys.argv[4]
instance_url = clean_url(sys.argv[2])
sitekey = sys.argv[3]
url = set_path(instance_url, "/api/v1/account/secret/get")
resp = c.get(url, cookies=c.cookies.get_dict())
assert resp.status_code == 200
return resp.json()["secret"]
def widget_url():
if len(sys.argv) < 4:
help()
exit(1)
instance_url = set_path(clean_url(sys.argv[2]), "/widget")
sitekey = sys.argv[3]
query = urlencode({"sitekey": sitekey})
url = f"{instance_url}?{query}"
return url
if __name__ == "__main__":
if option == "create_captcha":
login()
print(create_captcha())
elif option == "get_secret":
login()
print(get_secret())
elif option == "widget_url":
print(widget_url())
else:
help()