pep/poezio
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fix #3424 (fix receipts)

Browse source
This commit is contained in:
mathieui 2018-08-11 22:26:24 +02:00
parent 3c8fb156ad
commit b6c84cba5b
No known key found for this signature in database
GPG key ID: C59F84CEEFD616E3
1 changed files with 16 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
poezio/core/tabs.py
View file

@ -145,7 +145,8 @@ class Tabs:
self._tab_types = defaultdict(list)
self._tab_names = dict()
for tab in self._tabs:
self._tab_types[type(tab)].append(tab)
for cls in _get_tab_types(tab):
self._tab_types[cls].append(tab)
self._tab_names[tab.name] = tab
self._update_numbers()
@ -203,7 +204,8 @@ class Tabs:
else:
tab.nb = self._tabs[-1].nb + 1
self._tabs.append(tab)
self._tab_types[type(tab)].append(tab)
for cls in _get_tab_types(tab):
self._tab_types[cls].append(tab)
self._tab_names[tab.name] = tab
def delete(self, tab: tabs.Tab, gap=False):
@ -218,7 +220,8 @@ class Tabs:
is_current = tab is self.current_tab
self._tab_types[type(tab)].remove(tab)
for cls in _get_tab_types(tab):
self._tab_types[cls].remove(tab)
del self._tab_names[tab.name]
if gap:
@ -338,3 +341,13 @@ class Tabs:
result = self._insert_nogaps(old_pos, new_pos)
self._update_numbers()
return result
def _get_tab_types(tab: tabs.Tab) -> List[Type[tabs.Tab]]:
"""Return all parent classes of a tab type"""
types = []
current_cls = tab.__class__
while current_cls != tabs.Tab:
types.append(current_cls)
current_cls = current_cls.__bases__[0]
return types