Skip to content

Commit a28dc47

Browse files
authored
Use $resource-types over Entity for db
1 parent fac4ff7 commit a28dc47

1 file changed

Lines changed: 9 additions & 8 deletions

File tree

aidbox_python_sdk/db.py

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -138,18 +138,19 @@ async def alchemy(self, statement, *, execute=False):
138138
async def _get_all_entities_name(self):
139139
result = None
140140

141-
#Support legacy instalations with entity attribute data structure
142-
query_url = f"{self._settings.APP_INIT_URL}/Entity?type=resource&_elements=id&_count=999"
141+
query_url = f"{self._settings.APP_INIT_URL}/$resource-types"
143142
async with self._client.get(query_url) as resp:
144-
json_resp = await resp.json()
145-
result = [entry["resource"]["id"] for entry in json_resp.get("entry", [])]
143+
if resp.status == 200:
144+
json_resp = await resp.json()
145+
result = list(json_resp.keys())
146146

147-
if result is None or len(result) == 0:
148-
# Load resource type from fhir schema
149-
query_url = f"{self._settings.APP_INIT_URL}/$resource-types"
147+
if not result:
148+
# Support legacy instalations with entity attribute data structure
149+
query_url = f"{self._settings.APP_INIT_URL}/Entity?type=resource&_elements=id&_count=999"
150150
async with self._client.get(query_url) as resp:
151151
json_resp = await resp.json()
152-
result = list(json_resp.keys())
152+
result = [entry["resource"]["id"] for entry in json_resp.get("entry", [])]
153+
153154
return result or []
154155

155156
async def _init_table_cache(self):

0 commit comments

Comments
 (0)