mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-30 11:19:23 +00:00
39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
import urwid
|
|
|
|
class Tabs(urwid.WidgetWrap):
|
|
def __init__(self, tabs, tab_offset=0):
|
|
urwid.WidgetWrap.__init__(self, "")
|
|
self.tab_offset = tab_offset
|
|
self.tabs = tabs
|
|
self.show()
|
|
|
|
def _tab(self, content, attr):
|
|
p = urwid.Text(content, align="center")
|
|
p = urwid.Padding(p, align="center", width=("relative", 100))
|
|
p = urwid.AttrWrap(p, attr)
|
|
return p
|
|
|
|
def keypress(self, size, key):
|
|
if key in ["tab", "l"]:
|
|
self.tab_offset = (self.tab_offset + 1)%(len(self.tabs))
|
|
self.show()
|
|
elif key == "h":
|
|
self.tab_offset = (self.tab_offset - 1)%(len(self.tabs))
|
|
self.show()
|
|
return self._w.keypress(size, key)
|
|
|
|
def show(self):
|
|
headers = []
|
|
for i in range(len(self.tabs)):
|
|
txt = self.tabs[i][0]()
|
|
if i == self.tab_offset:
|
|
headers.append(self._tab(txt, "heading"))
|
|
else:
|
|
headers.append(self._tab(txt, "heading_inactive"))
|
|
headers = urwid.Columns(headers, dividechars=1)
|
|
self._w = urwid.Frame(
|
|
body = self.tabs[self.tab_offset][1](),
|
|
header = headers
|
|
)
|
|
self._w.set_focus("body")
|