Skip to content

Commit

Permalink
Add /query/list/parents endpoint
Browse files Browse the repository at this point in the history
  • Loading branch information
albireox committed Aug 8, 2024
1 parent 89c4c78 commit 0edb732
Showing 1 changed file with 11 additions and 1 deletion.
12 changes: 11 additions & 1 deletion python/valis/routes/query.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
carton_program_list, carton_program_map,
get_targets_by_sdss_id, get_targets_by_catalog_id,
get_targets_obs, get_paged_target_list_by_mapper)
from sdssdb.peewee.sdss5db import database
from sdssdb.peewee.sdss5db import database, catalogdb

# convert string floats to proper floats
Float = Annotated[Union[float, str], BeforeValidator(lambda x: float(x) if x and isinstance(x, str) else x)]
Expand Down Expand Up @@ -172,6 +172,16 @@ async def program_map(self):

return carton_program_map()

@router.get('/list/parents', summary='Return a list of available parent catalog tables',
response_model=List[str])
async def parent_catalogs(self):
"""Return a list of available parent catalog tables."""

columns = catalogdb.SDSS_ID_To_Catalog._meta.fields.keys()
catalogs = [col.split('__')[0] for col in columns if '__' in col]

return sorted(catalogs)

@router.get('/carton-program', summary='Search for all SDSS targets within a carton or program',
response_model=List[SDSSModel], dependencies=[Depends(get_pw_db)])
async def carton_program(self,
Expand Down

0 comments on commit 0edb732

Please sign in to comment.