openpilot_comma/selfdrive/ui/layouts/network.py

import pyray as rl
from openpilot.system.ui.lib.application import Widget
from openpilot.system.ui.lib.wifi_manager import WifiManagerWrapper
from openpilot.system.ui.widgets.network import WifiManagerUI


class NetworkLayout(Widget):
  def __init__(self):
    self.wifi_manager = WifiManagerWrapper()
    self.wifi_ui = WifiManagerUI(self.wifi_manager)

  def render(self, rect: rl.Rectangle):
    self.wifi_ui.render(rect)

  def shutdown(self):
    self.wifi_manager.shutdown()