--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/data_manager/data_manager_genomic_super_signature_ravmodels.py	Wed Jan 12 19:07:19 2022 +0000
@@ -0,0 +1,64 @@
+#!/usr/bin/env python
+# Dan Blankenberg
+
+
+import argparse
+import json
+import os
+import re
+import shutil
+import urllib.request
+
+
+OUTPUT_BASE_NAME = 'RAVmodel.rds'
+
+BUCKET_NAME = "genomic_super_signature"
+RAVMODEL_NAME = "RAVmodel_{prior}.rds"
+PRIOR_URL = "https://storage.googleapis.com/{bucket_name}/{prior_filename}"
+
+
+def get_prior_url(prior):
+    return PRIOR_URL.format(bucket_name=BUCKET_NAME, prior_filename=RAVMODEL_NAME.format(prior=prior))
+
+
+def clean(text):
+    return re.sub(r'[^\w\-_\.]', '-', text)
+
+
+def main():
+    parser = argparse.ArgumentParser()
+    parser.add_argument('--output', dest='output', action='store', type=str, default=None)
+    parser.add_argument('--json', dest='json', action='store', type=str, default=None)
+    parser.add_argument('--dbkey', dest='dbkey', action='store', type=str, default=None, help='dbkey')
+    parser.add_argument('--name', dest='name', action='store', type=str, default=None)
+    parser.add_argument('--id', dest='id', action='store', type=str, default=None)
+    parser.add_argument('--prior', dest='prior', action='store', type=str, default=None)
+    parser.add_argument('--url', dest='url', action='store', type=str, default=None)
+    parser.add_argument('--input', dest='input', action='store', type=str, default=None)
+    parser.add_argument('--symlink', dest='symlink', action='store_true', default=False)
+    parser.add_argument('--version', dest='version', action='store', type=str, default="0")
+    parser.add_argument('--tablename', dest='tablename', action='store', type=str, default="genomic_super_signature_ravmodels")
+    args = parser.parse_args()
+
+    os.makedirs(args.output, exist_ok=True)
+    output_filename = os.path.join(args.output, OUTPUT_BASE_NAME)
+    if args.symlink:
+        assert args.input, "You must provide a filename when using symlink functionality."
+        os.symlink(args.input, output_filename)
+    else:
+        url = args.url
+        if args.prior:
+            url = get_prior_url(args.prior)
+        if url:
+            urllib.request.urlretrieve(url, output_filename)
+        else:
+            assert args.input, "You must provide a filename, prior, or URL."
+            shutil.copyfile(args.input, output_filename)
+    data_manager_dict = {'data_tables': {args.tablename: [dict(value=clean(args.id), dbkey=args.dbkey, version=args.version, name=args.name, path=OUTPUT_BASE_NAME)]}}
+
+    with open(args.json, 'w') as fh:
+        json.dump(data_manager_dict, fh, sort_keys=True)
+
+
+if __name__ == "__main__":
+    main()