Got some basic api testing done.
parent
fa37dccfff
commit
4cc4c5c99f
@ -0,0 +1,56 @@
|
||||
import requests
|
||||
from abc import ABC, abstractmethod
|
||||
from dataclasses import dataclass
|
||||
|
||||
BASE_URL = "http://LOCALHOST:4000/REST"
|
||||
FORMAT = '.json'
|
||||
|
||||
@dataclass
|
||||
class RxCui():
|
||||
id: str
|
||||
|
||||
def get_atc_class(self):
|
||||
pass
|
||||
def get_brandnames(self):
|
||||
pass
|
||||
|
||||
|
||||
|
||||
|
||||
def FindRxcuiByString(name: str, **kwargs) -> RxCui:
|
||||
'''
|
||||
Find a RxCUI by string based on a string
|
||||
Defaults to searching RxNorm (i.e. drugs) using a best match option
|
||||
'''
|
||||
|
||||
url = BASE_URL + "/rxcui" + FORMAT
|
||||
query = {'allsrc':0, 'srclist':'RXNORM', 'search':2} | kwargs | {'name':name}
|
||||
r = requests.get(url, params=query)
|
||||
|
||||
#extract RxCUIs
|
||||
return [RxCui(x) for x in r.json()['idGroup']['rxnormId']]
|
||||
|
||||
def get_all_properties(rxcui_list: RxCui,*prop_category):
|
||||
'''
|
||||
This is used to query for properties
|
||||
'''
|
||||
for rxcui in rxcui_list:
|
||||
url = BASE_URL + "/rxcui/" + rxcui.id + "/allProperties" + FORMAT
|
||||
r = requests.get(url, params={"prop": prop_category})
|
||||
j = r.json()
|
||||
yield j
|
||||
|
||||
def get_brands_from_ingredients(rxcui: RxCui):
|
||||
'''
|
||||
This is used to query for properties
|
||||
'''
|
||||
url = BASE_URL + "/brands" + FORMAT
|
||||
r = requests.get(url, params={"ingredientids": rxcui.id})
|
||||
j = r.json()
|
||||
|
||||
return [ AssociatedBrand(x,rxcui) for x in j['brandGroup']['conceptProperties']]
|
||||
|
||||
class AssociatedBrand():
|
||||
def __init__(self,brand,ingredient: RxCui):
|
||||
self.ingredient_rxcui = ingredient
|
||||
self.brand_rxcui = RxCui(brand['rxcui'])
|
||||
Loading…
Reference in New Issue