From 7d69bbf9d29c8089a9d22d71ffdb5f20b2261307 Mon Sep 17 00:00:00 2001
From: BO ZHANG <bozhang@nao.cas.cn>
Date: Mon, 9 Jan 2023 17:42:51 +0800
Subject: [PATCH] set default columns for RC query

---
 csst_common/data_manager.py | 37 ++++++++++++++++++++++++++++++++++---
 1 file changed, 34 insertions(+), 3 deletions(-)

diff --git a/csst_common/data_manager.py b/csst_common/data_manager.py
index d59f75c..7c1b3f0 100644
--- a/csst_common/data_manager.py
+++ b/csst_common/data_manager.py
@@ -627,12 +627,43 @@ class CsstMsDataManager:
         except:
             return False
 
-    def dfs_rc_query(self, ra=180, dec=0, radius=2, min_mag=0, max_mag=30, obstime=-1, limit=-1):
+    def dfs_rc_query(
+            self,
+            ra=180,
+            dec=0,
+            radius=2,
+            columns=(
+                    'ref_epoch',
+                    'ra',
+                    'ra_error',
+                    'dec',
+                    'dec_error',
+                    'parallax',
+                    'parallax_error',
+                    'pmra',
+                    'pmra_error',
+                    'pmdec',
+                    'pmdec_error',
+                    'phot_g_mean_mag',
+                    'source_id'
+            ),
+            min_mag=0,
+            max_mag=30,
+            obstime=-1,
+            limit=-1
+    ):
         """ Query Reference Catalog (RC) from DFS. """
         try:
             cat = self.dfs_CatApi.catalog_query(
-                catalog_name="gaia3", ra=ra, dec=dec, radius=radius, min_mag=min_mag, max_mag=max_mag,
-                obstime=obstime, limit=limit
+                catalog_name="gaia3",
+                ra=ra,
+                dec=dec,
+                columns=columns,
+                radius=radius,
+                min_mag=min_mag,
+                max_mag=max_mag,
+                obstime=obstime,
+                limit=limit
             )
             tbl = self.dfs_CatApi.to_table(cat)
             return tbl
-- 
GitLab