Do not crash on damage/target profiles named like [this]

This commit is contained in:
DarkPhoenix
2019-08-20 00:47:49 +03:00
parent e488497a42
commit d266aa796e
2 changed files with 10 additions and 2 deletions

View File

@@ -40,8 +40,12 @@ class ChangeDamagePattern(ContextMenuUnconditional):
start, end = pattern.name.find('['), pattern.name.find(']')
if start is not -1 and end is not -1:
currBase = pattern.name[start + 1:end]
name = pattern.name[end + 1:].strip()
if not name:
self.singles.append(pattern)
continue
# set helper attr
setattr(pattern, "_name", pattern.name[end + 1:].strip())
setattr(pattern, "_name", name)
if currBase not in self.subMenus:
self.subMenus[currBase] = []
self.subMenus[currBase].append(pattern)

View File

@@ -68,8 +68,12 @@ class TargetProfileSwitcher(ContextMenuUnconditional):
start, end = profile.name.find('['), profile.name.find(']')
if start is not -1 and end is not -1:
currBase = profile.name[start + 1:end]
name = profile.name[end + 1:].strip()
if not name:
self.singles.append(profile)
continue
# set helper attr
setattr(profile, '_name', profile.name[end + 1:].strip())
setattr(profile, '_name', name)
if currBase not in self.subMenus:
self.subMenus[currBase] = []
self.subMenus[currBase].append(profile)