felicity-lims/backend/felicity_faker/analysis.py

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.choice([1,2]),
            "patientUid": random.randint(50, 2100),
            "priority":random.choice([0, 1]),
            "samples": [gen_sample() for _x in range(random.randint(1, 5))],
        } for i in range(10000)
    ] for x in range(2)
]

# 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)