Source Code for Module helpers.DiskSetup

 1  #! /usr/bin/env python 
 2  #------------------------------------------------------------------------------ 
 3  #$Id: DiskSetup.py 9344 2012-08-03 11:28:27Z EckhardSutorius $ 
 4  """ 
 5     Sets up a new disk, creating subdirs ingest, products etc. Given the disk 
 6     name, eg 'disk05' and archive, e.g. 'wsa' will create all the essential 
 7     subdirectories in /disk05/wsa/. 
 8   
 9     @author: Eckhard Sutorius 
10     @org:    WFAU, IfA, University of Edinburgh 
11   
12     @newfield contributors: Contributors, Contributors (Alphabetical Order) 
13     @contributors: R.S. Collins 
14  """ 
15  #------------------------------------------------------------------------------ 
16  from   __future__      import division, print_function 
17  import os 
18   
19  from   wsatools.CLI         import CLI 
20  from   wsatools.SystemConstants import SystemConstants as sysc 
21  import wsatools.Utilities       as utils 
22  #------------------------------------------------------------------------------ 
23  # Entry point for script. 
24   
25  # Allow module to be imported as well as executed from the command line 
26  if __name__ == "__main__": 
27      # Define command-line interface options 
28      CLI.progArgs += [ 
29        CLI.Argument("diskName", "disk05", 
30          isValOK=lambda x: x.startswith("disk") and x[-2:].isdigit()), 
31        CLI.Argument("archive", "wsa")] 
32   
33      cli = CLI("DiskSetup", "$Revision: 9344 $", __doc__) 
34   
35      disk = os.path.join(' ', cli.getArg("diskName"), cli.getArg("archive")) 
36      disk = os.path.normpath(disk.strip()) 
37   
38      if not os.path.exists(disk): 
39          raise SystemExit(disk + " does not exist!") 
40   
41      # ingest/fits/ 
42      utils.ensureDirExist(os.path.join(disk, sysc.fitsDir)) 
43   
44      # ingest/deprecated/ 
45      utils.ensureDirExist(os.path.join(disk, sysc.deprecatedDir)) 
46   
47      # products/ 
48      utils.ensureDirExist(os.path.join(disk, sysc.productsDir)) 
49   
50      # products/logs/ 
51      utils.ensureDirExist(os.path.join(disk, sysc.productLogDir)) 
52   
53      # products/stacks/ 
54      utils.ensureDirExist(os.path.join(disk, sysc.stackDir)) 
55   
56      # products/mosaics/ 
57      utils.ensureDirExist(os.path.join(disk, sysc.mosaicDir)) 
58   
59      # products/diffs/ 
60      utils.ensureDirExist(os.path.join(disk, sysc.diffDir)) 
61   
62      # products/jpgs/ 
63      utils.ensureDirExist(os.path.join(disk, sysc.compressImDir)) 
64   
65  #------------------------------------------------------------------------------ 
66  # Change log: 
67  # 
68  #  7-Jul-2006, ETWS: First Version 
69