568215cf70
* count responses * perf(backend): ⚡ remove validation on recipe summary response use the construct() method from pydantic to reduce get time as well as optimize the SQL query for recipes * update UI to support new categories/tags * fix(backend): 🐛 restrict recipes by group Co-authored-by: Hayden <hay-kot@pm.me>
62 lines
1.3 KiB
Python
62 lines
1.3 KiB
Python
import json
|
|
import random
|
|
import string
|
|
import time
|
|
|
|
import requests
|
|
|
|
|
|
def random_string(length: int) -> str:
|
|
return "".join(random.choice(string.ascii_lowercase) for _ in range(length))
|
|
|
|
|
|
def payload_factory() -> dict:
|
|
return {"name": random_string(15)}
|
|
|
|
|
|
def login(username="changeme@email.com", password="MyPassword"):
|
|
|
|
payload = {"username": username, "password": password}
|
|
r = requests.post("http://localhost:9000/api/auth/token", payload)
|
|
|
|
# Bearer
|
|
token = json.loads(r.text).get("access_token")
|
|
return {"Authorization": f"Bearer {token}"}
|
|
|
|
|
|
def populate_data(token):
|
|
for _ in range(300):
|
|
payload = payload_factory()
|
|
r = requests.post("http://localhost:9000/api/recipes", json=payload, headers=token)
|
|
|
|
if r.status_code != 201:
|
|
print(f"Error: {r.status_code}")
|
|
print(r.text)
|
|
exit()
|
|
|
|
else:
|
|
print(f"Created recipe: {payload}")
|
|
|
|
|
|
def time_request(url, headers):
|
|
start = time.time()
|
|
r = requests.get(url, headers=headers)
|
|
|
|
print(f"Total Recipes {len(r.json())}")
|
|
end = time.time()
|
|
print(end - start)
|
|
|
|
|
|
def main():
|
|
print("Starting...")
|
|
token = login()
|
|
# populate_data(token)
|
|
|
|
for _ in range(10):
|
|
time_request("http://localhost:9000/api/recipes", token)
|
|
|
|
print("Finished...")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|