lotterysim: create config.py.

move running_time, exchange_rate and vesting_file to config.py. update
README.

script/research/lotterysim/README.md

@@ -35,21 +35,19 @@ can be coupled with khonsu[^1] to achieve 100% accuracy and instant finality.
 
 # usage
 
-Replace `example.csv` with local distribution data. Modify the file name
-in core/constants.py as follows:
+Replace `example.csv` with local distribution data. Edit config.py as follows:
 
 ```python
-VESTING_FILE='example.csv'
+vesting_file = 'your_local_data.csv'
 ```
 
+Edit config.py to define the exchange rate and simulation running time,
+measured in slots.
+
 Then run the program:
 
 ```shell
 python vesting.py
 ```
 
-When prompted, enter the exchange rate, then enter the number of slots
-the simulation should run for. For testing purposes keep the slot time
-to 1-5k slots.
-
 [^1]: https://github.com/ertosns/thunderbolt

script/research/lotterysim/config.py

@@ -0,0 +1,9 @@
+# Running time measured in slots. Set to 0 to run for the whole vesting period.
+running_time = 1000
+
+# ERC20 DRK <-> native DRK exchange rate
+exchange_rate = 1
+
+# Path to data
+vesting_file = 'example.csv'
+

script/research/lotterysim/vesting.py

@@ -4,13 +4,14 @@ from core.constants import *
 from pid.pid_base import *
 from draw import draw
 import logging
+import config
 import os
 
 logging.basicConfig(filename='log/vesting.log', encoding='utf-8', level=logging.DEBUG)
 
-def vesting_instance(vesting, exchange_rate, running_time):
+def vesting_instance(vesting, running_time):
     os.system("rm log/*_feedback.hist; rm log/*_output.hist log/darkie* log/rewards.log")
-    native_drk = ERC20DRK * exchange_rate
+    native_drk = ERC20DRK * config.exchange_rate
     total_vesting = 0
     if __name__ == "__main__":
         darkies = []
@@ -31,12 +32,12 @@ def vesting_instance(vesting, exchange_rate, running_time):
         dt.write()
     return dt.avg_apr()
 
-if not os.path.exists(VESTING_FILE):
-    print('add vested distribution csv at path {} with vesting period {} (slots) aparts.'.format(VESTING_FILE, ONE_MONTH))
+if not os.path.exists(config.vesting_file):
+    print('add vested distribution csv at path {} with vesting period {} (slots) aparts.'.format(config.vesting_file, ONE_MONTH))
     exit()
 
 vesting = {}
-with open(VESTING_FILE) as f:
+with open(config.vesting_file) as f:
     for node  in f.readlines():
         keyval = node.split(',')
         key = keyval[0]
@@ -44,13 +45,12 @@ with open(VESTING_FILE) as f:
         vesting[keyval[0]] = eval(eval(val))
 
 nodes = len(vesting)
-exchange_rate = float(input("enter exchange rate:"))
-running_time = input("running time (leave it empty to run the whole vesting running time):")
-if running_time=='':
+if config.running_time== 0:
+    print("Running time is set to 0. Starting sim for whole vesting period...")
     running_time = len(next(iter(vesting.values())))*VESTING_PERIOD
 else:
-    running_time = int(running_time)
+    running_time = config.running_time
 
-apr = vesting_instance(vesting, exchange_rate, running_time)
+apr = vesting_instance(vesting, running_time)
 print('avg apr: {}%'.format(apr*100))
 draw()