mirror of
https://github.com/RYGhub/royalnet.git
synced 2024-11-27 21:44:21 +00:00
64 lines
1.7 KiB
Python
64 lines
1.7 KiB
Python
|
# Special imports
|
||
|
|
from __future__ import annotations
|
||
|
|
import royalnet.royaltyping as t
|
||
|
|
|
||
|
|
# External imports
|
||
|
|
import logging
|
||
|
|
import importlib
|
||
|
|
import click
|
||
|
|
import asyncio
|
||
|
|
import royalnet.engineer as engi
|
||
|
|
|
||
|
|
# Internal imports
|
||
|
|
from .pda import ConsolePDA
|
||
|
|
|
||
|
|
# Special global objects
|
||
|
|
log = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
# Code
|
||
|
|
@click.command()
|
||
|
|
@click.option("-p", "--pack", "packs", multiple=True)
|
||
|
|
def main(packs: t.List[str]):
|
||
|
|
log.debug("Creating PDA...")
|
||
|
|
pda = ConsolePDA()
|
||
|
|
|
||
|
|
for pack in packs:
|
||
|
|
log.debug(f"Importing module: {pack!r}")
|
||
|
|
try:
|
||
|
|
pack = importlib.import_module(pack)
|
||
|
|
except ImportError as e:
|
||
|
|
log.error(f"Skipping {pack!r}: {e!r}")
|
||
|
|
continue
|
||
|
|
|
||
|
|
for attribute in dir(pack):
|
||
|
|
log.debug(f"Getting attribute: {attribute!r}")
|
||
|
|
value = pack.__getattribute__(attribute)
|
||
|
|
log.debug(f"Attribute is: {value!r}")
|
||
|
|
|
||
|
|
if isinstance(value, engi.PartialCommand):
|
||
|
|
log.debug(f"Attribute is a PartialCommand, registering it as: {value.f.__name__!r}")
|
||
|
|
pda.register_partial(part=value, names=[value.f.__name__])
|
||
|
|
|
||
|
|
elif isinstance(value, engi.Conversation):
|
||
|
|
log.debug(f"Attribute is a Conversation, registering it...")
|
||
|
|
pda.register_conversation(conv=value)
|
||
|
|
|
||
|
|
log.debug("Getting event loop...")
|
||
|
|
loop = asyncio.get_event_loop()
|
||
|
|
log.debug(f"Event loop is: {loop!r}")
|
||
|
|
|
||
|
|
log.debug("Running the PDA until interrupted...")
|
||
|
|
try:
|
||
|
|
loop.run_until_complete(pda.run())
|
||
|
|
except KeyboardInterrupt:
|
||
|
|
log.debug("Got an interrupt, shutting down...")
|
||
|
|
exit(0)
|
||
|
|
|
||
|
|
log.fatal("PDA stopped unexpectedly, shutting down...")
|
||
|
|
exit(1)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|