Added varaible argument to main function and fixed import from utils.

This commit is contained in:
2018-08-12 19:27:57 +02:00
parent 50fe8adaf1
commit 6450041198

View File

@@ -42,7 +42,7 @@ from pprint import pprint
from deluge_client import DelugeRPCClient from deluge_client import DelugeRPCClient
from sshtunnel import SSHTunnelForwarder from sshtunnel import SSHTunnelForwarder
from docopt import docopt from docopt import docopt
from .utils import ColorizeFilter, convert from utils import ColorizeFilter, convert
BASE_DIR = os.path.dirname(os.path.abspath(__file__)) BASE_DIR = os.path.dirname(os.path.abspath(__file__))
@@ -239,13 +239,13 @@ def signal_handler(signal, frame):
logger.info('\nGood bye!') logger.info('\nGood bye!')
sys.exit(0) sys.exit(0)
def main(): def main(arg):
""" """
Main function, parse the input Main function, parse the input
""" """
signal.signal(signal.SIGINT, signal_handler) signal.signal(signal.SIGINT, signal_handler)
arguments = docopt(__doc__, version='1') arguments = docopt(__doc__, argv=arg, version='1')
# Set logging level for streamHandler # Set logging level for streamHandler
if arguments['--debug']: if arguments['--debug']: