From 403f5b52b116b998792ac1c6b0ad648ace251c97 Mon Sep 17 00:00:00 2001
From: Eduardo Cueto-Mendoza <cueto303@gmail.com>
Date: Sun, 17 Apr 2022 17:18:54 +0100
Subject: [PATCH] Update for 04-17-22

---
 run_service.py | 55 +++++++++++++++++++++++++++++++++++++++++++-------
 1 file changed, 48 insertions(+), 7 deletions(-)

diff --git a/run_service.py b/run_service.py
index 7cff94b..7e8cb1f 100755
--- a/run_service.py
+++ b/run_service.py
@@ -1,17 +1,58 @@
+import argparse
+from time import sleep
 import subprocess as sub
 
-cmd = ["python", "main_frequentist.py"]
-#cmd = ["python", "main_bayesian.py"]
-cmd2 = ["./cpu_watt.sh","freq_1_cpu_watts"]
-cmd3 = ["./mem_free.sh", "freq_1_ram_use"]
+# Construct an argument parser
+all_args = argparse.ArgumentParser()
+
+all_args.add_argument("-b", "--Value1", action="store", dest="b",
+   type=int, choices=range(1,6), help="Bayesian model of size x")
+all_args.add_argument("-f", "--Value2", action="store", dest="f",
+   type=int, choices=range(1,6), help="Frequentist model of size x")
+args = vars(all_args.parse_args())
+
+
+check = list(args.values())
+if all(v is None for v in check):
+    raise Exception("One argument required")
+elif None in check:
+    if args['f'] is not None:
+        cmd = ["python", "main_frequentist.py"]
+    elif args['b'] is not None:
+        cmd = ["python", "main_bayesian.py"]
+else:
+    raise Exception("Only one argument allowed")
+
+
+wide = args["f"] or args["b"]
+
+with open("tmp", "w") as file:
+    file.write(str(wide))
+
+sleep(3)
+
+
+if cmd[1] == "main_frequentist.py":
+    cmd2 = ["./cpu_watt.sh", "freq_{}_cpu_watts".format(wide)]
+    cmd3 = ["./mem_free.sh", "freq_{}_ram_use".format(wide)]
+    with open("frq", "w") as file:
+        file.write(str(1))
+    with open("bay", "w") as file:
+        file.write(str(0))
+elif cmd[1] == "main_bayesian.py":
+    cmd2 = ["./cpu_watt.sh", "bayes_{}_cpu_watts".format(wide)]
+    cmd3 = ["./mem_free.sh", "bayes_{}_ram_use".format(wide)]
+    with open("bay", "w") as file:
+        file.write(str(1))
+    with open("frw", "w") as file:
+        file.write(str(0))
 
-##from time import sleep
 
 path = sub.check_output(['pwd'])
 path = path.decode()
 path = path.replace('\n', '')
 
-startWattCounter = 'python ' + path + '/gpu_sample_draw.py'
+#startWattCounter = 'python ' + path + '/gpu_sample_draw.py'
 
 #test = startNODE.split()
 #test.append(pythonEnd)
@@ -28,7 +69,7 @@ p3 = sub.Popen(cmd2)
 p4 = sub.Popen(cmd3)
 
 retcode = p1.wait()
-print(retcode)
+print("Return code: {}".format(retcode))
 
 p1.kill()
 #p2.kill()