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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
|
#!/usr/bin/env python3
#
# Copyright (c) 2017 Weitian LI <weitian@aaronly.me>
# MIT license
"""
Update the manifest.yaml with generated products.
"""
import os
import argparse
import logging
from glob import glob
from collections import OrderedDict
from _context import acispy
from acispy.manifest import get_manifest
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)
def get_path(pglob, pdir=None):
if pdir is not None:
pglob = os.path.join(pdir, pglob)
p = glob(pglob)
if len(p) == 0:
return None
elif len(p) == 1:
return p[0]
else:
return p
def add_repro(reprodir, manifest):
"""
Add the generated products by ``chandra_repro`` to the manifest.
"""
logging.info("Adding repro products from: {0}".format(reprodir))
keyglobs = OrderedDict([
("evt2", "acisf*_repro_evt2.fits"),
("bpix", "acisf*_repro_bpix1.fits"),
("fov", "acisf*_repro_fov1.fits"),
("asol", "pcadf*_asol?.fits"), # maybe multiple
("pbk", "acisf*_pbk0.fits"),
("msk", "acisf*_msk1.fits"),
])
for k, g in keyglobs.items():
p = get_path(g, pdir=reprodir)
manifest.setpath(k, p)
def main():
parser = argparse.ArgumentParser(
description="Update manifest.yaml with generated products")
parser.add_argument("-c", "--create", dest="create",
action="store_true",
help="create 'manifest.yaml' under current working " +
"directory if necessary")
parser.add_argument("-r", "--repro", dest="reprodir", default=None,
help="path to the repro directory; add the " +
"reprocessed products to manifest if specified")
args = parser.parse_args()
if args.create:
open("manifest.yaml", "a").close()
manifest = get_manifest()
if args.reprodir:
add_repro(reprodir=args.reprodir, manifest=manifest)
if __name__ == "__main__":
main()
|
