Commit d3678fb3 authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

add DFS utils

parent f263c7a0
Pipeline #2787 passed with stage
...@@ -9,8 +9,9 @@ Modified-History: ...@@ -9,8 +9,9 @@ Modified-History:
2023-07-11, Bo Zhang, updated DFS 2023-07-11, Bo Zhang, updated DFS
2023-12-08, Bo Zhang, tweaks 2023-12-08, Bo Zhang, tweaks
2023-12-20, Bo Zhang, do nothing in DFS.__init__() 2023-12-20, Bo Zhang, do nothing in DFS.__init__()
2023-12-28, Bo Zhang, add DFS.rec_to_table and DFS.rec_to_dlist
""" """
from astropy.table import Table
from csst_dfs_api.common.catalog import CatalogApi from csst_dfs_api.common.catalog import CatalogApi
from csst_dfs_api.facility import ( from csst_dfs_api.facility import (
Level0DataApi, Level0DataApi,
...@@ -48,6 +49,10 @@ class DFS: ...@@ -48,6 +49,10 @@ class DFS:
def L2DataApi(self): def L2DataApi(self):
return Level2DataApi() return Level2DataApi()
@property
def L2TypeApi(self):
return Level2TypeApi()
@property @property
def OtherDataApi(self): def OtherDataApi(self):
return OtherDataApi() return OtherDataApi()
...@@ -55,3 +60,16 @@ class DFS: ...@@ -55,3 +60,16 @@ class DFS:
@property @property
def CatApi(self): def CatApi(self):
return CatalogApi() return CatalogApi()
@staticmethod
def rec_to_table(rec):
assert rec["code"] == 0, rec
data = list(zip(*rec["data"]))
colnames = rec["columns"]
return Table(data=data, names=colnames)
@staticmethod
def rec_to_dlist(rec):
assert rec["code"] == 0, rec
colnames = rec["columns"]
return [dict(zip(colnames, _)) for _ in rec["data"]]
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment