forked from cizra/pycat
-
Notifications
You must be signed in to change notification settings - Fork 2
/
cli.py
31 lines (24 loc) · 1.01 KB
/
cli.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
#!/usr/bin/env python3
import importlib
import os
import click
from pycat import Session
if not os.path.exists('data'):
os.mkdir('data')
os.chdir('data')
@click.command()
@click.option("--bind", default=os.environ.get('PYCAT_BIND', 'localhost'), help='Bind Address', show_default=True)
@click.option('--terminate-on-disconnect', is_flag=True, help='Quit pycat when the MUD disconnects')
@click.argument("world", default=lambda: os.environ.get('PYCAT_WORLD'))
@click.argument("port", default=lambda: os.environ.get('PYCAT_PORT', 7777))
@click.argument('arg', default=lambda: os.environ.get('PYCAT_ARG', ""))
def main(world: str, port: str | int, arg: str, bind: str, terminate_on_disconnect: bool, **kwargs: str) -> None:
if world:
world_module = importlib.import_module('worlds.' + world)
else:
world_module = importlib.import_module('portal')
port = int(port)
ses = Session(world_module, port, arg, bind, terminate_on_disconnect)
ses.run()
if __name__ == '__main__':
main.main()