main.py: launch(): support launching of functions
This commit is contained in:
parent
64d7434f59
commit
2dd581b902
1 changed files with 2 additions and 2 deletions
|
|
@ -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)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue