h

csst_dfs_commons/convert/date_utils.py

@@ -4,6 +4,9 @@ import time
 def format_datetime(dt):
     return dt.strftime('%Y-%m-%d %H:%M:%S')
     
+def format_compact_datetime(dt):
+    return dt.strftime('%Y%m%d%H%M%S')
+
 def format_date(dt):
     return dt.strftime('%Y-%m-%d')
 

csst_dfs_commons/models/constants.py

-UPLOAD_CHUNK_SIZE = 1024*1000*4
+PI = 3.14159265358979323846
+
+UPLOAD_CHUNK_SIZE = 16*1024
 
 STATUS_SUCESS = 0
 

csst_dfs_commons/utils/fits.py

@@ -16,3 +16,8 @@ def get_healpix_id(ra, dec, nside=32):
     hp = HEALPix(nside=nside, order='nested', frame=ICRS())
     coord = SkyCoord(ra = ra * u.deg, dec = dec * u.deg, frame='icrs')
     return hp.skycoord_to_healpix(coord)
+
+def get_healpix_ids(ra, dec, radius, nside=32):
+    hp = HEALPix(nside=nside, order='nested', frame=ICRS())
+    coord = SkyCoord(ra = ra * u.deg, dec = dec * u.deg, frame='icrs')
+    return hp.cone_search_skycoord(coord, radius = radius * u.deg)    
\ No newline at end of file