view xena_restore.py @ 40:fd24e220f240

more edit on description

#!/usr/bin/env python

"""
  xena_restore.py: delete a dataset from Xena

  Back up the Xena data to a user-specified external directory.
"""

import argparse
import os
import re
import shutil
import subprocess
import sys
import traceback
import xena_utils as xena

def writeException(outFp, msg = None):
    exc_type, exc_value, exc_traceback = sys.exc_info()
    lines = traceback.format_exception(exc_type, exc_value, exc_traceback)
    allLines = ''.join('!! ' + line for line in lines)
    if msg is None:
        outFp.write("Unsuccessful: error %s\n" % allLines)
    else:
        outFp.write("%s\n%s" % (msg, allLines))
    
    

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("backupDir", type=str)
    parser.add_argument("outfile", type=str)
    args = parser.parse_args()

    outFp = open(args.outfile, "w")
    xenaFileDir = xena.fileDir()

    for thisFile in os.listdir(args.backupDir):
        # The files in the xena directory should be pairs of 
        # data and metadata files, with the metadata having the same
        # name as the data file plus the .json suffix.  When you find
        # a json file in the backup directory, then copy the json and data
        # file into the xena directory, starting with the json file, and 
        # execute the xena load command.
        if re.search(".json$", thisFile):
            jsonFile = thisFile
            dataFile = re.sub(".json$", "", jsonFile)
            assert(os.path.exists(args.backupDir + "/" + dataFile))
            try:
                shutil.copy(args.backupDir + "/" + jsonFile, xenaFileDir)
            except:
                writeException(outFp,
                               msg="Error: cannot restore %s from %s" \
                                   % (jsonFile, args.backupDir))
                outFp.close()
                sys.exit(-1)
            try:
                shutil.copy(args.backupDir + "/" + dataFile, xenaFileDir)
            except:
                writeException(outFp,
                               msg="Error: cannot restore %s from %s" \
                                   % (dataFile, args.backupDir))
                outFp.close()
                sys.exit(-1)
            # Now set up the xena load command and try to execute it.
            xenaLoadCmd = "java -jar %s -l --force %s/%s -p %s" % (xena.jarPath(),
                                                                   xenaFileDir,
                                                                   dataFile, 
                                                                   xena.port())
            try:
                subprocess.call(xenaLoadCmd, shell=True)
            except:
                writeException(outFp,
                               msg="Could not reload %s into Xena" % dataFile)
                outFp.close()
                sys.exit(-1)
    # At this point, the restore should've been successful
    outFp.write("Restore complete\n")
    outFp.close()
    sys.exit(0)


if __name__ == "__main__":
    main()