Skip to content

Commit

Permalink
add union of apogee and boss queries
Browse files Browse the repository at this point in the history
  • Loading branch information
imedan committed Feb 2, 2024
1 parent 87167ef commit f7a72ee
Showing 1 changed file with 11 additions and 6 deletions.
17 changes: 11 additions & 6 deletions python/valis/db/queries.py
Original file line number Diff line number Diff line change
Expand Up @@ -300,20 +300,25 @@ def get_targets_obs(release: str, obs: str, obsWave: str) -> peewee.ModelSelect:
query_ap = apo.Star.select()\
.where(apo.Star.telescope == obs.lower() + '25m',
apo.Star.apred_vers == apred)
# temportary, just return some sdss_id
sdss_id_ap = [3350466]
query_ap = vizdb.SDSSidStacked.select()\
.where(vizdb.SDSSidStacked.sdss_id << sdss_id_ap)

# return union of the above
query_all = vizdb.SDSSidStacked.select()\
.where((vizdb.SDSSidStacked.sdss_id << query_boss) |
(vizdb.SDSSidStacked.sdss_id << query_ap))

if obsWave == 'boss':
return query_boss
elif obsWave == 'apogee':
# temportary, just return some sdss_id
query = vizdb.SDSSidStacked.select()\
.where(vizdb.SDSSidStacked.sdss_id << sdss_id_ap)
return query
return query_ap
elif obsWave == 'all':
return query_all
else:
raise ValueError('Did not pass "boss", "apogee" or "all" to obsWave')

return query


# test sdss ids
# 23326 - boss/astra
Expand Down

0 comments on commit f7a72ee

Please sign in to comment.