diff --git a/sqlreduce/main.py b/sqlreduce/main.py index 62911f9..f83c005 100755 --- a/sqlreduce/main.py +++ b/sqlreduce/main.py @@ -28,7 +28,7 @@ def sqlreduce_main(): query = args.file.read().rstrip() # check database connection - if not '=' in args.database: + if not '=' in args.database and not "postgres://" in args.database: args.database = f"dbname={args.database}" sqlreduce.check_connection(args.database)