gone-phishing/main.py

42 lines
763 B
Python

#!/usr/bin/env python3
# TODO should have some systemd integration, at least start/stop service
from argparse import ArgumentParser
import sys
import os
import yaml
from rtmbot import RtmBot
sys.path.append(os.getcwd())
def parse_args():
parser = ArgumentParser()
parser.add_argument(
'-c',
'--config',
help='Full path to config file.',
metavar='path'
)
return parser.parse_args()
def main(args=None):
# load args with config path if not specified
if not args:
args = parse_args()
config = yaml.load(open(args.config or 'rtmbot.conf', 'r'))
bot = RtmBot(config)
try:
bot.start()
except KeyboardInterrupt:
sys.exit(0)
if __name__ == "__main__":
main()