2015-07-21 15:55:02 +00:00
|
|
|
# Copyright 2015 Canonical, Ltd.
|
|
|
|
#
|
|
|
|
# This program is free software: you can redistribute it and/or modify
|
|
|
|
# it under the terms of the GNU Affero General Public License as
|
|
|
|
# published by the Free Software Foundation, either version 3 of the
|
|
|
|
# License, or (at your option) any later version.
|
|
|
|
#
|
|
|
|
# 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 Affero General Public License for more details.
|
|
|
|
#
|
|
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
|
2017-01-13 01:42:31 +00:00
|
|
|
from concurrent import futures
|
2017-09-08 03:59:30 +00:00
|
|
|
import fcntl
|
2015-07-21 15:55:02 +00:00
|
|
|
import logging
|
2018-02-07 21:37:22 +00:00
|
|
|
import os
|
|
|
|
import struct
|
2017-09-08 03:59:30 +00:00
|
|
|
import sys
|
2018-02-07 21:37:22 +00:00
|
|
|
import termios
|
|
|
|
import tty
|
2017-01-17 01:26:07 +00:00
|
|
|
|
2015-07-21 15:55:02 +00:00
|
|
|
import urwid
|
2017-11-16 00:29:22 +00:00
|
|
|
import yaml
|
2017-01-13 01:42:31 +00:00
|
|
|
|
2016-06-30 18:17:01 +00:00
|
|
|
from subiquitycore.signals import Signal
|
|
|
|
from subiquitycore.prober import Prober, ProberException
|
2015-07-21 15:55:02 +00:00
|
|
|
|
2016-06-30 18:17:01 +00:00
|
|
|
log = logging.getLogger('subiquitycore.core')
|
2015-07-21 15:55:02 +00:00
|
|
|
|
|
|
|
|
2016-07-25 00:38:19 +00:00
|
|
|
class ApplicationError(Exception):
|
2015-07-21 15:55:02 +00:00
|
|
|
""" Basecontroller exception """
|
|
|
|
pass
|
|
|
|
|
2017-09-08 03:59:30 +00:00
|
|
|
# From uapi/linux/kd.h:
|
|
|
|
KDGKBTYPE = 0x4B33 # get keyboard type
|
2018-02-07 21:37:22 +00:00
|
|
|
|
2018-05-22 17:46:00 +00:00
|
|
|
GIO_CMAP = 0x4B70 # gets colour palette on VGA+
|
|
|
|
PIO_CMAP = 0x4B71 # sets colour palette on VGA+
|
2017-09-12 13:28:08 +00:00
|
|
|
UO_R, UO_G, UO_B = 0xe9, 0x54, 0x20
|
2017-09-08 03:59:30 +00:00
|
|
|
|
2018-02-07 21:37:22 +00:00
|
|
|
# /usr/include/linux/kd.h
|
2018-05-22 17:46:00 +00:00
|
|
|
K_RAW = 0x00
|
|
|
|
K_XLATE = 0x01
|
2018-02-07 21:37:22 +00:00
|
|
|
K_MEDIUMRAW = 0x02
|
2018-05-22 17:46:00 +00:00
|
|
|
K_UNICODE = 0x03
|
|
|
|
K_OFF = 0x04
|
2018-02-07 21:37:22 +00:00
|
|
|
|
2018-05-22 17:46:00 +00:00
|
|
|
KDGKBMODE = 0x4B44 # gets current keyboard mode
|
|
|
|
KDSKBMODE = 0x4B45 # sets current keyboard mode
|
2018-02-07 21:37:22 +00:00
|
|
|
|
2017-09-08 03:59:30 +00:00
|
|
|
|
2017-11-21 22:26:47 +00:00
|
|
|
class ISO_8613_3_Screen(urwid.raw_display.Screen):
|
|
|
|
|
2017-11-21 22:37:49 +00:00
|
|
|
def __init__(self, _urwid_name_to_rgb):
|
|
|
|
self._fg_to_rgb = _urwid_name_to_rgb.copy()
|
|
|
|
self._fg_to_rgb['default'] = _urwid_name_to_rgb['light gray']
|
|
|
|
self._bg_to_rgb = _urwid_name_to_rgb.copy()
|
|
|
|
self._bg_to_rgb['default'] = _urwid_name_to_rgb['black']
|
2017-11-21 22:26:47 +00:00
|
|
|
super().__init__()
|
|
|
|
|
|
|
|
def _attrspec_to_escape(self, a):
|
|
|
|
f_r, f_g, f_b = self._fg_to_rgb[a.foreground]
|
|
|
|
b_r, b_g, b_b = self._bg_to_rgb[a.background]
|
2018-05-22 17:46:00 +00:00
|
|
|
return "\x1b[38;2;{};{};{};48;2;{};{};{}m".format(f_r, f_g, f_b,
|
|
|
|
b_r, b_g, b_b)
|
2017-09-12 13:28:08 +00:00
|
|
|
|
2017-09-08 03:59:30 +00:00
|
|
|
|
|
|
|
def is_linux_tty():
|
|
|
|
try:
|
|
|
|
r = fcntl.ioctl(sys.stdout.fileno(), KDGKBTYPE, ' ')
|
|
|
|
except IOError as e:
|
|
|
|
log.debug("KDGKBTYPE failed %r", e)
|
|
|
|
return False
|
|
|
|
log.debug("KDGKBTYPE returned %r", r)
|
|
|
|
return r == b'\x02'
|
|
|
|
|
2017-11-21 22:37:49 +00:00
|
|
|
|
|
|
|
def setup_screen(colors, styles):
|
2017-11-21 22:49:20 +00:00
|
|
|
"""Return a palette and screen to be passed to MainLoop.
|
|
|
|
|
|
|
|
colors is a list of exactly 8 tuples (name, (r, g, b))
|
|
|
|
|
|
|
|
styles is a list of tuples (stylename, fg_color, bg_color) where
|
|
|
|
fg_color and bg_color are defined in 'colors'
|
|
|
|
"""
|
|
|
|
# The part that makes this "fun" is that urwid insists on referring
|
|
|
|
# to the basic colors by their "standard" names but we overwrite
|
|
|
|
# these colors to mean different things. So we convert styles into
|
|
|
|
# an urwid palette by mapping the names in colors to the standard
|
|
|
|
# name, and then either overwrite the first 8 colors to be the
|
|
|
|
# colors from 'colors' (on the linux vt) or use a custom screen
|
|
|
|
# class that displays maps the standard color name to the value
|
|
|
|
# specified in colors using 24-bit control codes.
|
|
|
|
if len(colors) != 8:
|
2018-05-22 17:46:00 +00:00
|
|
|
raise Exception(
|
|
|
|
"setup_screen must be passed a list of exactly 8 colors")
|
2017-11-21 22:49:20 +00:00
|
|
|
urwid_8_names = (
|
|
|
|
'black',
|
|
|
|
'dark red',
|
|
|
|
'dark green',
|
|
|
|
'brown',
|
|
|
|
'dark blue',
|
|
|
|
'dark magenta',
|
|
|
|
'dark cyan',
|
|
|
|
'light gray',
|
|
|
|
)
|
2017-11-21 23:16:09 +00:00
|
|
|
urwid_name = dict(zip([c[0] for c in colors], urwid_8_names))
|
2017-11-21 22:49:20 +00:00
|
|
|
|
2017-11-21 23:16:09 +00:00
|
|
|
urwid_palette = []
|
2017-11-21 22:37:49 +00:00
|
|
|
for name, fg, bg in styles:
|
2017-11-21 23:16:09 +00:00
|
|
|
urwid_palette.append((name, urwid_name[fg], urwid_name[bg]))
|
2015-07-21 15:55:02 +00:00
|
|
|
|
2017-11-21 22:26:47 +00:00
|
|
|
if is_linux_tty():
|
|
|
|
curpal = bytearray(16*3)
|
|
|
|
fcntl.ioctl(sys.stdout.fileno(), GIO_CMAP, curpal)
|
|
|
|
for i in range(8):
|
|
|
|
for j in range(3):
|
2017-11-21 22:49:20 +00:00
|
|
|
curpal[i*3+j] = colors[i][1][j]
|
2017-11-21 22:26:47 +00:00
|
|
|
fcntl.ioctl(sys.stdout.fileno(), PIO_CMAP, curpal)
|
2017-11-21 23:16:09 +00:00
|
|
|
return urwid.raw_display.Screen(), urwid_palette
|
2017-11-21 22:26:47 +00:00
|
|
|
else:
|
2017-11-21 22:49:20 +00:00
|
|
|
_urwid_name_to_rgb = {}
|
|
|
|
for i, n in enumerate(urwid_8_names):
|
|
|
|
_urwid_name_to_rgb[n] = colors[i][1]
|
2017-11-21 23:16:09 +00:00
|
|
|
return ISO_8613_3_Screen(_urwid_name_to_rgb), urwid_palette
|
2017-11-21 22:26:47 +00:00
|
|
|
|
|
|
|
|
2018-02-07 21:37:22 +00:00
|
|
|
class KeyCodesFilter:
|
|
|
|
"""input_filter that can pass (medium) raw keycodes to the application
|
|
|
|
|
|
|
|
See http://lct.sourceforge.net/lct/x60.html for terminology.
|
|
|
|
|
|
|
|
Call enter_keycodes_mode()/exit_keycodes_mode() to switch into and
|
|
|
|
out of keycodes mode. In keycodes mode, the only events passed to
|
|
|
|
the application are "press $N" / "release $N" where $N is the
|
|
|
|
keycode the user pressed or released.
|
|
|
|
|
|
|
|
Much of this is cribbed from the source of the "showkeys" utility.
|
|
|
|
"""
|
|
|
|
|
|
|
|
def __init__(self):
|
|
|
|
self._fd = os.open("/proc/self/fd/0", os.O_RDWR)
|
|
|
|
self.filtering = False
|
|
|
|
|
|
|
|
def enter_keycodes_mode(self):
|
|
|
|
log.debug("enter_keycodes_mode")
|
|
|
|
self.filtering = True
|
2018-05-22 17:46:00 +00:00
|
|
|
# Read the old keyboard mode (it will proably always be
|
|
|
|
# K_UNICODE but well).
|
2018-02-07 21:37:22 +00:00
|
|
|
o = bytearray(4)
|
|
|
|
fcntl.ioctl(self._fd, KDGKBMODE, o)
|
|
|
|
self._old_mode = struct.unpack('i', o)[0]
|
|
|
|
# Make some changes to the terminal settings.
|
|
|
|
# If you don't do this, sometimes writes to the terminal hang (and no,
|
|
|
|
# I don't know exactly why).
|
|
|
|
self._old_settings = termios.tcgetattr(self._fd)
|
|
|
|
new_settings = termios.tcgetattr(self._fd)
|
|
|
|
new_settings[tty.IFLAG] = 0
|
2018-05-22 17:46:00 +00:00
|
|
|
new_settings[tty.LFLAG] = new_settings[tty.LFLAG] & ~(termios.ECHO |
|
|
|
|
termios.ICANON |
|
|
|
|
termios.ISIG)
|
2018-02-07 21:37:22 +00:00
|
|
|
new_settings[tty.CC][termios.VMIN] = 0
|
|
|
|
new_settings[tty.CC][termios.VTIME] = 0
|
|
|
|
termios.tcsetattr(self._fd, termios.TCSAFLUSH, new_settings)
|
|
|
|
# Finally, set the keyboard mode to K_MEDIUMRAW, which causes
|
|
|
|
# the keyboard driver in the kernel to pass us keycodes.
|
2018-05-22 17:46:00 +00:00
|
|
|
log.debug("old mode was %s, setting mode to %s",
|
|
|
|
self._old_mode, K_MEDIUMRAW)
|
2018-02-07 21:37:22 +00:00
|
|
|
fcntl.ioctl(self._fd, KDSKBMODE, K_MEDIUMRAW)
|
|
|
|
|
|
|
|
def exit_keycodes_mode(self):
|
|
|
|
log.debug("exit_keycodes_mode")
|
|
|
|
self.filtering = False
|
|
|
|
log.debug("setting mode back to %s", self._old_mode)
|
|
|
|
fcntl.ioctl(self._fd, KDSKBMODE, self._old_mode)
|
|
|
|
termios.tcsetattr(self._fd, termios.TCSANOW, self._old_settings)
|
|
|
|
|
|
|
|
def filter(self, keys, codes):
|
|
|
|
# Luckily urwid passes us the raw results from read() we can
|
|
|
|
# turn into keycodes.
|
|
|
|
if self.filtering:
|
|
|
|
i = 0
|
|
|
|
r = []
|
|
|
|
n = len(codes)
|
|
|
|
while i < len(codes):
|
|
|
|
# This is straight from showkeys.c.
|
|
|
|
if codes[i] & 0x80:
|
|
|
|
p = 'release '
|
|
|
|
else:
|
|
|
|
p = 'press '
|
2018-05-22 17:46:00 +00:00
|
|
|
if i + 2 < n and (codes[i] & 0x7f) == 0:
|
|
|
|
if (codes[i + 1] & 0x80) != 0:
|
|
|
|
if (codes[i + 2] & 0x80) != 0:
|
|
|
|
kc = (((codes[i + 1] & 0x7f) << 7) |
|
|
|
|
(codes[i + 2] & 0x7f))
|
|
|
|
i += 3
|
2018-02-07 21:37:22 +00:00
|
|
|
else:
|
|
|
|
kc = codes[i] & 0x7f
|
|
|
|
i += 1
|
|
|
|
r.append(p + str(kc))
|
|
|
|
return r
|
|
|
|
else:
|
|
|
|
return keys
|
|
|
|
|
|
|
|
|
|
|
|
class DummyKeycodesFilter:
|
|
|
|
# A dummy implementation of the same interface as KeyCodesFilter
|
|
|
|
# we can use when not running in a linux tty.
|
|
|
|
|
|
|
|
def enter_keycodes_mode(self):
|
|
|
|
pass
|
|
|
|
|
|
|
|
def exit_keycodes_mode(self):
|
|
|
|
pass
|
|
|
|
|
|
|
|
def filter(self, keys, codes):
|
|
|
|
return keys
|
|
|
|
|
|
|
|
|
2016-07-25 00:38:19 +00:00
|
|
|
class Application:
|
|
|
|
|
2016-07-25 00:51:39 +00:00
|
|
|
# A concrete subclass must set project and controllers attributes, e.g.:
|
|
|
|
#
|
|
|
|
# project = "subiquity"
|
2016-09-27 02:33:54 +00:00
|
|
|
# controllers = [
|
|
|
|
# "Welcome",
|
|
|
|
# "Installpath",
|
|
|
|
# "Network",
|
|
|
|
# "Filesystem",
|
|
|
|
# "Identity",
|
|
|
|
# "InstallProgress",
|
|
|
|
# "Login",
|
|
|
|
# ]
|
2016-09-29 01:06:09 +00:00
|
|
|
# The 'next-screen' and 'prev-screen' signals move through the list of
|
|
|
|
# controllers in order, calling the default method on the controller
|
|
|
|
# instance.
|
2016-06-30 18:50:21 +00:00
|
|
|
|
2015-07-21 15:55:02 +00:00
|
|
|
def __init__(self, ui, opts):
|
2015-10-23 15:03:04 +00:00
|
|
|
try:
|
|
|
|
prober = Prober(opts)
|
|
|
|
except ProberException as e:
|
|
|
|
err = "Prober init failed: {}".format(e)
|
|
|
|
log.exception(err)
|
2016-07-25 00:38:19 +00:00
|
|
|
raise ApplicationError(err)
|
2015-10-23 15:03:04 +00:00
|
|
|
|
2017-03-16 09:52:05 +00:00
|
|
|
opts.project = self.project
|
|
|
|
|
2017-11-16 00:29:22 +00:00
|
|
|
answers = {}
|
|
|
|
if opts.answers is not None:
|
|
|
|
answers = yaml.safe_load(open(opts.answers).read())
|
|
|
|
log.debug("Loaded answers %s", answers)
|
2017-12-13 03:10:36 +00:00
|
|
|
if not opts.dry_run:
|
|
|
|
open('/run/casper-no-prompt', 'w').close()
|
2017-11-16 00:29:22 +00:00
|
|
|
|
2018-02-07 21:37:22 +00:00
|
|
|
if is_linux_tty():
|
|
|
|
log.debug("is_linux_tty")
|
|
|
|
input_filter = KeyCodesFilter()
|
|
|
|
else:
|
|
|
|
input_filter = DummyKeycodesFilter()
|
|
|
|
|
2015-08-31 15:55:46 +00:00
|
|
|
self.common = {
|
|
|
|
"ui": ui,
|
|
|
|
"opts": opts,
|
|
|
|
"signal": Signal(),
|
2015-10-23 15:03:04 +00:00
|
|
|
"prober": prober,
|
2017-01-13 01:42:31 +00:00
|
|
|
"loop": None,
|
2018-05-18 00:33:25 +00:00
|
|
|
"pool": futures.ThreadPoolExecutor(4),
|
2017-11-16 00:29:22 +00:00
|
|
|
"answers": answers,
|
2018-02-07 21:37:22 +00:00
|
|
|
"input_filter": input_filter,
|
2015-08-31 15:55:46 +00:00
|
|
|
}
|
2016-11-22 23:08:36 +00:00
|
|
|
if opts.screens:
|
2018-05-22 17:46:00 +00:00
|
|
|
self.controllers = [c for c in self.controllers
|
|
|
|
if c in opts.screens]
|
2017-09-26 13:01:22 +00:00
|
|
|
ui.progress_completion = len(self.controllers)
|
2016-09-27 02:33:54 +00:00
|
|
|
self.common['controllers'] = dict.fromkeys(self.controllers)
|
|
|
|
self.controller_index = -1
|
2015-07-21 20:34:46 +00:00
|
|
|
|
2015-08-18 16:29:56 +00:00
|
|
|
def _connect_base_signals(self):
|
2015-07-21 20:34:46 +00:00
|
|
|
""" Connect signals used in the core controller
|
|
|
|
"""
|
|
|
|
signals = []
|
|
|
|
|
2015-07-23 02:31:21 +00:00
|
|
|
signals.append(('quit', self.exit))
|
2016-08-30 22:55:42 +00:00
|
|
|
if self.common['opts'].dry_run:
|
|
|
|
signals.append(('control-x-quit', self.exit))
|
2015-08-24 03:55:45 +00:00
|
|
|
signals.append(('refresh', self.redraw_screen))
|
2016-09-27 02:33:54 +00:00
|
|
|
signals.append(('next-screen', self.next_screen))
|
|
|
|
signals.append(('prev-screen', self.prev_screen))
|
2015-08-31 15:55:46 +00:00
|
|
|
self.common['signal'].connect_signals(signals)
|
2015-08-18 16:29:56 +00:00
|
|
|
|
|
|
|
# Registers signals from each controller
|
2016-09-27 02:33:54 +00:00
|
|
|
for controller, controller_class in self.common['controllers'].items():
|
2015-08-18 16:29:56 +00:00
|
|
|
controller_class.register_signals()
|
2015-08-31 15:55:46 +00:00
|
|
|
log.debug(self.common['signal'])
|
2015-07-21 20:34:46 +00:00
|
|
|
|
2016-09-27 02:33:54 +00:00
|
|
|
def next_screen(self, *args):
|
|
|
|
self.controller_index += 1
|
|
|
|
if self.controller_index >= len(self.controllers):
|
|
|
|
self.exit()
|
2017-09-26 13:01:22 +00:00
|
|
|
self.common['ui'].progress_current += 1
|
2016-09-27 02:33:54 +00:00
|
|
|
controller_name = self.controllers[self.controller_index]
|
2017-03-06 01:59:05 +00:00
|
|
|
log.debug("moving to screen %s", controller_name)
|
2016-09-27 02:33:54 +00:00
|
|
|
next_controller = self.common['controllers'][controller_name]
|
|
|
|
next_controller.default()
|
|
|
|
|
|
|
|
def prev_screen(self, *args):
|
|
|
|
if self.controller_index == 0:
|
|
|
|
return
|
|
|
|
self.controller_index -= 1
|
|
|
|
if self.controller_index >= len(self.controllers):
|
|
|
|
self.exit()
|
2017-09-26 13:01:22 +00:00
|
|
|
self.common['ui'].progress_current -= 1
|
2016-09-27 02:33:54 +00:00
|
|
|
controller_name = self.controllers[self.controller_index]
|
|
|
|
next_controller = self.common['controllers'][controller_name]
|
|
|
|
next_controller.default()
|
|
|
|
|
2015-07-21 15:55:02 +00:00
|
|
|
# EventLoop -------------------------------------------------------------------
|
|
|
|
def redraw_screen(self):
|
2017-01-13 01:48:25 +00:00
|
|
|
if self.common['loop'] is not None:
|
2015-07-21 15:55:02 +00:00
|
|
|
try:
|
2015-08-31 15:55:46 +00:00
|
|
|
self.common['loop'].draw_screen()
|
2015-07-21 15:55:02 +00:00
|
|
|
except AssertionError as e:
|
|
|
|
log.critical("Redraw screen error: {}".format(e))
|
|
|
|
|
|
|
|
def exit(self):
|
|
|
|
raise urwid.ExitMainLoop()
|
|
|
|
|
2018-05-01 00:49:39 +00:00
|
|
|
def run_scripts(self, scripts):
|
2018-05-01 01:00:27 +00:00
|
|
|
# run_scripts runs (or rather arranges to run, it's all async)
|
|
|
|
# a series of python snippets in a helpful namespace. This is
|
|
|
|
# all in aid of being able to test some part of the UI without
|
|
|
|
# having to click the same buttons over and over again to get
|
|
|
|
# the UI to the part you are working on.
|
|
|
|
#
|
|
|
|
# In the namespace are:
|
|
|
|
# * everything from view_helpers
|
|
|
|
# * wait, delay execution of subsequent scripts for a while
|
|
|
|
# * c, a function that finds a button and clicks it. uses
|
|
|
|
# wait, above to wait for the button to appear in case it
|
|
|
|
# takes a while.
|
2018-05-01 00:49:39 +00:00
|
|
|
from subiquitycore.testing import view_helpers
|
2018-05-01 00:57:03 +00:00
|
|
|
loop = self.common['loop']
|
|
|
|
|
|
|
|
class ScriptState:
|
|
|
|
def __init__(self):
|
|
|
|
self.ns = view_helpers.__dict__.copy()
|
|
|
|
self.waiting = False
|
|
|
|
self.wait_count = 0
|
|
|
|
self.scripts = scripts
|
|
|
|
|
|
|
|
ss = ScriptState()
|
|
|
|
|
2018-05-01 00:49:39 +00:00
|
|
|
def _run_script(*args):
|
2018-05-01 00:57:03 +00:00
|
|
|
log.debug("running %s", ss.scripts[0])
|
|
|
|
exec(ss.scripts[0], ss.ns)
|
|
|
|
if ss.waiting:
|
2018-05-01 00:49:39 +00:00
|
|
|
return
|
2018-05-01 00:57:03 +00:00
|
|
|
ss.scripts = ss.scripts[1:]
|
|
|
|
if ss.scripts:
|
|
|
|
loop.set_alarm_in(0.01, _run_script)
|
|
|
|
|
2018-05-01 00:49:39 +00:00
|
|
|
def c(pat):
|
2018-05-22 17:46:00 +00:00
|
|
|
but = view_helpers.find_button_matching(self.common['ui'],
|
|
|
|
'.*' + pat + '.*')
|
2018-05-01 00:49:39 +00:00
|
|
|
if not but:
|
2018-05-01 00:57:03 +00:00
|
|
|
ss.wait_count += 1
|
|
|
|
if ss.wait_count > 10:
|
2018-05-22 17:46:00 +00:00
|
|
|
raise Exception("no button found matching %r after"
|
|
|
|
"waiting for 10 secs" % pat)
|
|
|
|
wait(1, func=lambda: c(pat))
|
2018-05-01 00:49:39 +00:00
|
|
|
return
|
2018-05-01 00:57:03 +00:00
|
|
|
ss.wait_count = 0
|
2018-05-01 00:49:39 +00:00
|
|
|
view_helpers.click(but)
|
2018-05-01 00:57:03 +00:00
|
|
|
|
2018-05-01 00:49:39 +00:00
|
|
|
def wait(delay, func=None):
|
2018-05-01 00:57:03 +00:00
|
|
|
ss.waiting = True
|
2018-05-22 17:46:00 +00:00
|
|
|
|
2018-05-01 00:49:39 +00:00
|
|
|
def next(loop, user_data):
|
2018-05-01 00:57:03 +00:00
|
|
|
ss.waiting = False
|
2018-05-01 00:49:39 +00:00
|
|
|
if func is not None:
|
|
|
|
func()
|
2018-05-01 00:57:03 +00:00
|
|
|
if not ss.waiting:
|
|
|
|
ss.scripts = ss.scripts[1:]
|
|
|
|
if ss.scripts:
|
2018-05-01 00:49:39 +00:00
|
|
|
_run_script()
|
2018-05-01 00:57:03 +00:00
|
|
|
loop.set_alarm_in(delay, next)
|
|
|
|
|
|
|
|
ss.ns['c'] = c
|
|
|
|
ss.ns['wait'] = wait
|
|
|
|
ss.ns['ui'] = self.common['ui']
|
|
|
|
|
2018-05-01 00:49:39 +00:00
|
|
|
self.common['loop'].set_alarm_in(0.06, _run_script)
|
2018-05-01 00:29:29 +00:00
|
|
|
|
2015-07-21 15:55:02 +00:00
|
|
|
def run(self):
|
|
|
|
if not hasattr(self, 'loop'):
|
2017-11-21 22:26:47 +00:00
|
|
|
if self.common['opts'].run_on_serial:
|
|
|
|
palette = self.STYLES_MONO
|
|
|
|
screen = urwid.raw_display.Screen()
|
|
|
|
else:
|
2017-11-21 22:37:49 +00:00
|
|
|
screen, palette = setup_screen(self.COLORS, self.STYLES)
|
2015-07-21 15:55:02 +00:00
|
|
|
|
2015-08-31 15:55:46 +00:00
|
|
|
self.common['loop'] = urwid.MainLoop(
|
2017-11-21 23:05:30 +00:00
|
|
|
self.common['ui'], palette=palette, screen=screen,
|
2018-05-22 17:46:00 +00:00
|
|
|
handle_mouse=False, pop_ups=True,
|
|
|
|
input_filter=self.common['input_filter'].filter)
|
2018-02-07 21:37:22 +00:00
|
|
|
|
2015-08-31 15:55:46 +00:00
|
|
|
log.debug("Running event loop: {}".format(
|
|
|
|
self.common['loop'].event_loop))
|
2017-11-14 21:30:09 +00:00
|
|
|
self.common['base_model'] = self.model_class(self.common)
|
2015-07-21 15:55:02 +00:00
|
|
|
try:
|
2017-01-13 01:48:25 +00:00
|
|
|
self.common['loop'].set_alarm_in(0.05, self.next_screen)
|
2018-05-01 00:29:29 +00:00
|
|
|
if self.common['opts'].scripts:
|
2018-05-01 00:49:39 +00:00
|
|
|
self.run_scripts(self.common['opts'].scripts)
|
2018-05-22 17:46:00 +00:00
|
|
|
controllers_mod = __import__('%s.controllers' % self.project,
|
|
|
|
None, None, [''])
|
2017-09-26 13:01:22 +00:00
|
|
|
for k in self.controllers:
|
2015-08-31 15:55:46 +00:00
|
|
|
log.debug("Importing controller: {}".format(k))
|
2017-01-17 00:48:21 +00:00
|
|
|
klass = getattr(controllers_mod, k+"Controller")
|
2016-09-27 02:33:54 +00:00
|
|
|
self.common['controllers'][k] = klass(self.common)
|
|
|
|
log.debug("*** %s", self.common['controllers'])
|
2015-08-31 15:55:46 +00:00
|
|
|
self._connect_base_signals()
|
|
|
|
self.common['loop'].run()
|
2015-07-21 15:55:02 +00:00
|
|
|
except:
|
|
|
|
log.exception("Exception in controller.run():")
|
|
|
|
raise
|
2018-05-18 02:33:07 +00:00
|
|
|
finally:
|
|
|
|
# concurrent.futures.ThreadPoolExecutor tries to join all
|
|
|
|
# threads before exiting. We don't want that and this
|
|
|
|
# ghastly hack prevents it.
|
|
|
|
from concurrent.futures import thread
|
|
|
|
thread._threads_queues = {}
|