From f50d166f80adcfaaa30a62c1e0b25bfc9d76a2a4 Mon Sep 17 00:00:00 2001 From: Aaron LI Date: Tue, 21 Feb 2017 20:17:39 +0800 Subject: Move various Python tools from 'scripts/' to 'bin/' --- bin/analyze_path.py | 52 ++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 52 insertions(+) create mode 100755 bin/analyze_path.py (limited to 'bin/analyze_path.py') diff --git a/bin/analyze_path.py b/bin/analyze_path.py new file mode 100755 index 0000000..d2acd42 --- /dev/null +++ b/bin/analyze_path.py @@ -0,0 +1,52 @@ +#!/usr/bin/env python3 +# +# Copyright (c) 2017 Weitian LI +# MIT license +# +# Weitian LI +# 2017-02-06 + +""" +Extract the object name and observation ID from the directory path. + +The root directory of the object data has the format: + _oi +""" + +import os +import argparse + +from _context import acispy +from acispy.analyze_path import get_name, get_obsid + + +def main(): + parser = argparse.ArgumentParser( + description="Extract object name and ObsID from data directory") + parser.add_argument("-b", "--brief", dest="brief", + action="store_true", help="Be brief") + parser.add_argument("-n", "--name", dest="name", + action="store_true", help="Only get object name") + parser.add_argument("-i", "--obsid", dest="obsid", + action="store_true", help="Only get observation ID") + parser.add_argument("path", nargs="?", default=os.getcwd(), + help="Path to the data directory " + + "(default: current working directory)") + args = parser.parse_args() + path = os.path.abspath(args.path) + + b_get_name = False if args.obsid else True + b_get_obsid = False if args.name else True + + if b_get_name: + if not args.brief: + print("Name:", end=" ") + print(get_name(path)) + if b_get_obsid: + if not args.brief: + print("ObsID:", end=" ") + print(get_obsid(path)) + + +if __name__ == "__main__": + main() -- cgit v1.2.2