db.close()
if not os.path.exists("ps1dr2.db") or args.db_only:
print("building temporary database")
build_db(args.warp_meta, args.warp_files)
print("completed")
if args.db_only:
sys.exit(0)
test_time_agreement = False
# connect to catch database
config = Config.from_args(args)
with Catch.with_config(config) as catch:
# setup WCS object to calculate image corners
w = WCS(naxis=2)
w.wcs.ctype = "RA---TAN", "DEC--TAN"
w.wcs.cdelt = -6.94444461e-05, 6.94444461e-05
catch.db.drop_spatial_index()
observations = []
tri = ProgressTriangle(1, catch.logger, base=2)
bad_dt = []
# iterate over rows in temporary database
for row in get_rows(args.start_offset):
# PS1DR2 object inherits sbsearch.model.Observation columns
# observation_id, source, mjd_start, mjd_stop, fov, spatial_terms,
def catch_cli(*args):
"""CATCH command-line script."""
import sys
import argparse
import uuid
from astropy.time import Time
from astropy.table import Table
from catch import Catch, Config
from catch.config import _config_example
parser = argparse.ArgumentParser(
"catch",
epilog=f"Configuration files are JSON-formatted:\n{_config_example}")
parser.add_argument("--config", help="CATCH configuration file")
parser.add_argument("--database", help="use this database URI")
parser.add_argument("--log", help="save log messages to this file")
parser.add_argument("--arc-limit",
type=float,
help="maximal arc length to search, radians")
parser.add_argument("--time-limit",
type=float,
help="maximal time length to search, days")
parser.add_argument("--debug", action="store_true", help="debug mode")
subparsers = parser.add_subparsers(help="sub-command help")
verify = subparsers.add_parser(
"verify", help="connect to database and verify and create tables")
verify.set_defaults(command="verify")
list_sources = subparsers.add_parser("sources",
help="show available data sources")
list_sources.set_defaults(command="sources")
search = subparsers.add_parser("search", help="search for an object")
search.set_defaults(command="search")
search.add_argument("desg", help="object designation")
search.add_argument(
"--source",
dest="sources",
action="append",
help="search this observation source (may be used multiple times)",
)
search.add_argument("--force",
dest="cached",
action="store_false",
help="do not use cached results")
search.add_argument("-o", help="write table to this file")
args = parser.parse_args()
try:
getattr(args, "command")
except AttributeError:
parser.print_help()
sys.exit()
if args.command == "verify":
print("Verify databases and create as needed.\n")
rows = []
config = Config.from_args(args)
with Catch.with_config(config) as catch:
if args.command == "verify":
pass
elif args.command == "sources":
print("Available sources:\n *",
"\n * ".join(catch.sources.keys()))
elif args.command == "search":
job_id = uuid.uuid4()
catch.query(args.desg,
job_id,
sources=args.sources,
cached=args.cached)
columns = set()
# catch.caught returns a list of rows.
for row in catch.caught(job_id):
r = {}
# Each row consists of a Found and an Observation object. The
# Observation object will be a subclass, e.g.,
# NeatPalomarTricam, or SkyMapper.
for data_object in row:
# Aggregate fields and values from each data object
for k, v in _serialize_object(data_object):
r[k] = v
columns = columns.union(set(r.keys()))
r["cutout_url"] = row.Observation.cutout_url(
row.Found.ra, row.Found.dec)
r["date"] = Time(row.Found.mjd, format="mjd").iso
rows.append(r)
if args.command == "search":
if rows == []:
print("# none found")
else:
# make sure all rows have all columns
for i in range(len(rows)):
for col in columns:
rows[i][col] = rows[i].get(col)
tab = Table(rows=rows)
if args.o:
tab.write(args.o,
format="ascii.fixed_width_two_line",
overwrite=True)
else:
tab.pprint(-1, -1)