1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
|
#!/usr/bin/env python3
#
# Copyright (c) 2017 Weitian LI <liweitianux@live.com>
# 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:
<name>_oi<obsid>
"""
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()
|
