Current File : //snap/core22/current/usr/share/subiquity/subiquitycore/ui/width.py
# Copyright 2018 Canonical, Ltd.
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, version 3.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
import urwid
size_neutral_decorations = (
urwid.AttrMap,
urwid.PopUpLauncher,
urwid.WidgetDisable,
)
size_neutral_widget_wraps = (
urwid.Button,
urwid.CheckBox,
)
def widget_width(w):
"""Return the natural width of the widget w."""
if (isinstance(w, urwid.WidgetDecoration)
and getattr(w, 'has_original_width', False)):
return widget_width(w.original_widget)
if hasattr(w, 'get_natural_width'):
return w.get_natural_width()
if isinstance(w, size_neutral_widget_wraps):
return widget_width(w._wrapped_widget)
if isinstance(w, size_neutral_decorations):
return widget_width(w.original_widget)
if isinstance(w, urwid.Padding):
if w.width == urwid.RELATIVE_100:
return w.left + w.right + widget_width(w.original_widget)
elif isinstance(w.width, int):
return w.width
if isinstance(w, urwid.Text):
return len(w.text)
if isinstance(w, urwid.Columns):
if len(w.contents) == 0:
return 0
r = 0
for w1, o in w.contents:
if o[0] == urwid.GIVEN:
r += o[1]
else:
r += widget_width(w1)
r += (len(w.contents) - 1) * w.dividechars
return r
raise Exception(f"don't know how to find width of {w!r}")
Mini Shell