mirror of
https://github.com/beak-insights/felicity-lims.git
synced 2025-02-24 08:53:00 +08:00
80 lines
2.3 KiB
Python
80 lines
2.3 KiB
Python
import concurrent.futures
|
|
from faker import Faker
|
|
import requests
|
|
import random
|
|
import time
|
|
from core import run_query, authenticate, do_work
|
|
|
|
import logging
|
|
|
|
logging.basicConfig(level=logging.INFO)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
engine = Faker()
|
|
|
|
add_ar_query = """
|
|
mutation AddAnalysisRequest ($clientRequestId: String!, $clientUid: Int!, $patientUid: Int!, $samples: [ARSampleInputType!]!) {
|
|
createAnalysisRequest(clientRequestId: $clientRequestId, clientUid: $clientUid, patientUid: $patientUid, samples: $samples) {
|
|
uid
|
|
}
|
|
}
|
|
"""
|
|
|
|
def gen_sample():
|
|
randoms = [
|
|
{
|
|
"sampletypes": [2,3],
|
|
"analyses": [None, 1,2,3],
|
|
"profiles": [None, 1,2],
|
|
},
|
|
{
|
|
"sampletypes": [4],
|
|
"analyses": [3],
|
|
"profiles": [None],
|
|
},
|
|
]
|
|
|
|
while True:
|
|
selected = random.choice(randoms)
|
|
s_typ = random.choice(selected.get("sampletypes"))
|
|
anal = random.choice(selected.get("analyses"))
|
|
prof = random.choice(selected.get("profiles"))
|
|
|
|
if anal or prof:
|
|
break
|
|
|
|
return {
|
|
"sampleType": s_typ,
|
|
"profiles": [prof] if prof else [],
|
|
"analyses": [anal] if anal else []
|
|
}
|
|
|
|
ar_variables = [ # list of lists - each list will be run in its own thread -> simulating multi user regs
|
|
[
|
|
{
|
|
"clientRequestId": engine.ssn(),
|
|
"clientUid": random.randint(1, 9),
|
|
"patientUid": random.randint(5, 1650),
|
|
"priority":random.choice([0, 1]),
|
|
"samples": [gen_sample() for _x in range(random.randint(1, 5))],
|
|
} for i in range(100)
|
|
] for x in range(10)
|
|
]
|
|
|
|
# def do_work1(var_list):
|
|
# auth_headers = authenticate()
|
|
#
|
|
# for variables in var_list:
|
|
# run_query(query=add_patient_query, variables=variables, headers=auth_headers)
|
|
#
|
|
|
|
def start_ar_reg():
|
|
with concurrent.futures.ThreadPoolExecutor(max_workers=1) as executor:
|
|
future_to_url = (executor.submit(do_work, add_ar_query, variables) for variables in ar_variables)
|
|
|
|
for future in concurrent.futures.as_completed(future_to_url):
|
|
try:
|
|
data = future.result()
|
|
logger.info("Done")
|
|
except Exception as exc:
|
|
logger.error(exc)
|