84 lines
2.6 KiB
Python
84 lines
2.6 KiB
Python
from pathlib import Path
|
|
import pandas as pd
|
|
import shlex
|
|
import subprocess
|
|
from zipfile import ZipFile
|
|
import json
|
|
|
|
def available_states():
|
|
#TODO: Make a setting for this path
|
|
location = Path('/opt/development/synthea_webservice/synthea/src/main/resources/geography/')
|
|
|
|
df = pd.read_csv(location / 'timezones.csv', index_col=False)
|
|
# The state information is expected in the first column
|
|
states = df[df.columns[0]].to_list()
|
|
states.sort()
|
|
return states
|
|
|
|
def available_modules():
|
|
#TODO: Make a setting for this path
|
|
location = Path('/opt/development/synthea_webservice/synthea/src/main/resources/modules/')
|
|
|
|
# Assumption here: A folder is a single module. And all .json in the main modules folder is a module.
|
|
modules = []
|
|
for module in location.iterdir():
|
|
if module.is_file() and module.suffix == '.json':
|
|
data = json.loads(module.read_text())
|
|
modules.append({'module' : module.name.replace('.json',''), 'name' : data['name']})
|
|
|
|
modules = sorted(modules, key=lambda k: k['name'].lower())
|
|
return modules
|
|
|
|
def run_synthea(state = None, population = None, gender = None, age = None, module = None):
|
|
# TODO: Make synthea setting(s)
|
|
location = '/opt/development/synthea_webservice/synthea/'
|
|
synthea_cmd = ['/opt/development/synthea_webservice/synthea/run_synthea']
|
|
zip_file = 'Synthea_'
|
|
zip_export = location
|
|
|
|
if population:
|
|
synthea_cmd.append('-p')
|
|
synthea_cmd.append(str(population))
|
|
zip_file += f'population_{population}_'
|
|
|
|
if gender:
|
|
synthea_cmd.append('-g')
|
|
synthea_cmd.append(gender.upper())
|
|
zip_file += f'gender_{gender}_'
|
|
|
|
if age:
|
|
synthea_cmd.append('-a')
|
|
synthea_cmd.append(age)
|
|
zip_file += f'age_{age}_'
|
|
|
|
if module:
|
|
synthea_cmd.append('-m')
|
|
synthea_cmd.append(module)
|
|
zip_file += f'module_{module}_'
|
|
|
|
if state:
|
|
synthea_cmd.append(state)
|
|
zip_file += f'state_{state}'
|
|
|
|
process_ok = False
|
|
log = ''
|
|
with subprocess.Popen(synthea_cmd,cwd=location, stdout=subprocess.PIPE,stderr=subprocess.PIPE) as process:
|
|
for line in process.stdout:
|
|
line = line.decode('utf8')
|
|
log += line
|
|
if not process_ok:
|
|
process_ok = line.find('BUILD SUCCESSFUL') >= 0
|
|
|
|
if process_ok:
|
|
with ZipFile(f'{zip_export}/{zip_file}.zip', 'w') as export:
|
|
for file in Path(location + 'output/fhir_stu3').iterdir():
|
|
export.write(file,file.name)
|
|
|
|
return Path(f'{zip_export}/{zip_file}.zip')
|
|
else:
|
|
raise Exception(log)
|
|
|
|
|
|
|
|
|