@@ -25,6 +25,9 @@ def load_class(script_name, className):
2525 return klass
2626
2727def load_types_of_results (response_type ):
28+ with open ("/beacon/conf/models/models_conf.yml" , 'r' ) as pfile :
29+ models_confile = yaml .safe_load (pfile )
30+ pfile .close ()
2831 list_of_results_classes_accepted = []
2932 version_catch = re .search (r"(v\d+(\.\d+)*)" , RequestAttributes .returned_schema [0 ]["schema" ])
3033 if version_catch :
@@ -34,6 +37,9 @@ def load_types_of_results(response_type):
3437 dirs = os .listdir ("/beacon/models" )
3538 for folder in dirs :
3639 subdirs = os .listdir ("/beacon/models/" + folder )
40+ if folder in models_confile :
41+ if models_confile [folder ]["model_enabled" ] == False :
42+ continue
3743 if "validator" in subdirs :
3844 validatordirs = os .listdir ("/beacon/models/" + folder + "/validator/" + response_type )
3945 for validatorfolder in validatordirs :
@@ -48,6 +54,9 @@ def load_types_of_results(response_type):
4854 list_of_results_classes_accepted .append (klass )
4955 for subfolder in subdirs :
5056 underdirs = os .listdir ("/beacon/models/" + folder + "/" + subfolder )
57+ if folder + '/' + subfolder in models_confile :
58+ if models_confile [folder + '/' + subfolder ]["model_enabled" ] == False :
59+ continue
5160 if "validator" in underdirs :
5261 try :
5362 validatordirs = os .listdir ("/beacon/models/" + folder + "/" + subfolder + "/validator/" + response_type )
@@ -67,10 +76,16 @@ def load_types_of_results(response_type):
6776 return union_type
6877
6978def load_routes ():
79+ with open ("/beacon/conf/models/models_conf.yml" , 'r' ) as pfile :
80+ models_confile = yaml .safe_load (pfile )
81+ pfile .close ()
7082 dirs = os .listdir ("/beacon/models" )
7183 routes_to_add = {}
7284 for folder in dirs :
7385 subdirs = os .listdir ("/beacon/models/" + folder )
86+ if folder in models_confile :
87+ if models_confile [folder ]["model_enabled" ] == False :
88+ continue
7489 if "conf" in subdirs :
7590 confiles = os .listdir ("/beacon/models/" + folder + "/conf/entry_types/" )
7691 for confile in confiles :
@@ -90,6 +105,9 @@ def load_routes():
90105 else :
91106 for subfolder in subdirs :
92107 underdirs = os .listdir ("/beacon/models/" + folder + "/" + subfolder )
108+ if folder + '/' + subfolder in models_confile :
109+ if models_confile [folder + '/' + subfolder ]["model_enabled" ] == False :
110+ continue
93111 if "conf" in underdirs :
94112 confiles = os .listdir ("/beacon/models/" + folder + "/" + subfolder + "/conf/entry_types/" )
95113 for confile in confiles :
@@ -191,12 +209,18 @@ def get_one_module_conf(entry_type):
191209 return entry_type_confile
192210
193211def get_modules_confiles ():
212+ with open ("/beacon/conf/models/models_conf.yml" , 'r' ) as pfile :
213+ models_confile = yaml .safe_load (pfile )
214+ pfile .close ()
194215 dirs = os .listdir ("/beacon/models" )
195216 list_of_confiles = []
196217 for folder in dirs :
197218 subdirs = os .listdir ("/beacon/models/" + folder )
198219 if "conf" in subdirs :
199220 confiles = os .listdir ("/beacon/models/" + folder + "/conf/entry_types/" )
221+ if folder in models_confile :
222+ if models_confile [folder ]["model_enabled" ] == False :
223+ continue
200224 for confile in confiles :
201225 if confile != '__pycache__' :
202226 with open ("/beacon/models/" + folder + "/conf/entry_types/" + confile , 'r' ) as pfile :
@@ -206,6 +230,9 @@ def get_modules_confiles():
206230 else :
207231 for subfolder in subdirs :
208232 underdirs = os .listdir ("/beacon/models/" + folder + "/" + subfolder )
233+ if folder + '/' + subfolder in models_confile :
234+ if models_confile [folder + '/' + subfolder ]["model_enabled" ] == False :
235+ continue
209236 if "conf" in underdirs :
210237 confiles = os .listdir ("/beacon/models/" + folder + "/" + subfolder + "/conf/entry_types/" )
211238 for confile in confiles :
0 commit comments