From 2dd581b902b8e500e60bc4bcd303a342fb28fac6 Mon Sep 17 00:00:00 2001 From: The MMGen Project Date: Tue, 12 Dec 2023 10:19:50 +0000 Subject: [PATCH] main.py: launch(): support launching of functions --- mmgen/main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mmgen/main.py b/mmgen/main.py index 5807f605..f09787a3 100755 --- a/mmgen/main.py +++ b/mmgen/main.py @@ -22,7 +22,7 @@ main: Script launcher for the MMGen Project import sys,os -def launch(*, mod=None, package='mmgen'): +def launch(*, mod=None, func=None, package='mmgen'): if sys.platform == 'linux' and sys.stdin.isatty(): import termios,atexit @@ -31,7 +31,7 @@ def launch(*, mod=None, package='mmgen'): atexit.register(lambda: termios.tcsetattr(fd,termios.TCSADRAIN,old)) try: - __import__(f'{package}.main_{mod}') + __import__(f'{package}.main_{mod}') if mod else func() except KeyboardInterrupt: sys.stderr.write('\nUser interrupt\n') sys.exit(1)