[research/lotterysim] interactive runningtime option

script/research/lotterysim/vesting.py

@@ -41,8 +41,12 @@ with open(VESTING_FILE) as f:
         vesting[keyval[0]] = eval(eval(val))
 
 nodes = len(vesting)
-#running_time = len(next(iter(vesting.values())))*VESTING_PERIOD
-running_time = 1000000
+running_time = int(input("running time (leave it empty to run the whole vesting running time):"))
+if running_time=='':
+    running_time = len(next(iter(vesting.values())))*VESTING_PERIOD
+else:
+    running_time = int(running_time)
+
 apr = vesting_instance(vesting, running_time)
 print('avg apr: {}%'.format(apr*100))
 draw()