Skip to content

Commit 6d924d7

Browse files
Update initModel.py
1 parent a9ec45d commit 6d924d7

1 file changed

Lines changed: 4 additions & 6 deletions

File tree

RFEM/initModel.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -241,7 +241,7 @@ def __delete__(self, index_or_name):
241241
Args:
242242
index_or_name (str or int): Name of the index of model
243243
'''
244-
if isinstance(index_or_name, str):
244+
if isinstance(index_or_name, str) and index_or_name in self.clientModelDct:
245245
assert index_or_name in list(self.clientModelDct)
246246
self.clientModelDct.pop(index_or_name)
247247
if len(self.clientModelDct) > 0:
@@ -252,8 +252,7 @@ def __delete__(self, index_or_name):
252252
if isinstance(index_or_name, int):
253253
assert index_or_name <= len(self.clientModelDct)
254254
modelLs = connectionGlobals.client.service.get_model_list()
255-
256-
if modelLs:
255+
if modelLs and (modelLs.name[index_or_name] in self.clientModelDct):
257256
self.clientModelDct.pop(modelLs.name[index_or_name])
258257
if len(self.clientModelDct) > 0:
259258
model_key = list(self.clientModelDct)[-1]
@@ -344,9 +343,8 @@ def closeModel(index_or_name, save_changes = False):
344343
'''
345344

346345
connectToServer()
347-
348346
if isinstance(index_or_name, int):
349-
# Model.__delete__(Model, index_or_name)
347+
Model.__delete__(Model, index_or_name)
350348
connectionGlobals.client.service.close_model(index_or_name, save_changes)
351349

352350
elif isinstance(index_or_name, str):
@@ -356,7 +354,7 @@ def closeModel(index_or_name, save_changes = False):
356354
modelLs = connectionGlobals.client.service.get_model_list().name
357355
if index_or_name in modelLs:
358356
try:
359-
# Model.__delete__(Model, index_or_name)
357+
Model.__delete__(Model, index_or_name)
360358
connectionGlobals.client.service.close_model(modelLs.index(index_or_name), save_changes)
361359
except:
362360
print('Model did NOT close properly.')

0 commit comments

Comments
 (0)