From 64500411980790e9a411eb3ffb677f80cf2c0c0c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Kevin=20Midb=C3=B8e?= <kevin.midboe@gmail.com>
Date: Sun, 12 Aug 2018 19:27:57 +0200
Subject: [PATCH] Added varaible argument to main function and fixed import
 from utils.

---
 deluge_cli.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

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