view xenaGetDataset.py @ 50:3167c1a26101

fix

#!/usr/bin/env python

import argparse
import re
import urllib2


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("dataHub", type=str)
    parser.add_argument("datasetId", type=str)
    parser.add_argument("datafile", type=str)
    args = parser.parse_args()

    datasetUrlHost = re.sub("/proj/", "/download/", args.dataHub)
    datasetIdTokens = re.split("/", args.datasetId)
    datasetUrl = datasetUrlHost + "/" + "/".join(datasetIdTokens[1:])

    data = open(args.datafile, "w")

    try:
        dd = urllib2.urlopen(datasetUrl)
        data.write(dd.read())
        dd.close()
        data.close()
    except:
        data.wriet("You entered dataset id: %s\n" % (args.datasetId))
        data.write("Dataset does not exist. check typo.")
        data.close()
        sys.exit(1)

if __name__ == "__main__":
    main()