Merge pull request #14 from suutari-ai/global-sanitizers-python-path

Add path of the config file to Python path

Author: suutari-ai

database_sanitizer/__main__.py

@@ -3,6 +3,7 @@
 from __future__ import unicode_literals
 
 import argparse
+import os
 import sys
 
 from .config import Configuration
@@ -19,15 +20,13 @@ def main(args=None):
     parser.add_argument(
         "--config",
         "-c",
-        nargs=1,
         type=str,
         dest="config",
         help="Path to the sanitizer configuration file.",
     )
     parser.add_argument(
         "--output",
         "-o",
-        nargs=1,
         type=str,
         dest="output",
         help=(
@@ -45,9 +44,11 @@ def main(args=None):
     config = None
 
     if args.config:
-        config = Configuration.from_file(args.config[0])
+        conf_dir = os.path.realpath(os.path.dirname(args.config))
+        sys.path.insert(0, conf_dir)
+        config = Configuration.from_file(args.config)
     if args.output:
-        output = open(args.output[0], "w")
+        output = open(args.output, "w")
 
     try:
         run(