From 055f79e23b6905cc68815676e3393c39222aec51 Mon Sep 17 00:00:00 2001 From: powerfj Date: Wed, 15 Dec 2021 20:46:05 +0800 Subject: [PATCH] Add timeout parameter for connect_args --- connector.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/connector.py b/connector.py index 5ae983d..86e9cf5 100644 --- a/connector.py +++ b/connector.py @@ -125,14 +125,14 @@ class Connection(Database): """ These objects are small stateless factories for cursors, which do all the real work. """ - def __init__(self, db_name, db_url='http://localhost:8123/', username=None, password=None, readonly=False, ssl="False"): + def __init__(self, db_name, db_url='http://localhost:8123/', username=None, password=None, readonly=False, ssl="False", timeout=60): if ssl.upper() == "TRUE": db_url = db_url.replace("http", "https") elif ssl.upper() == "FALSE": pass else: raise ValueError("Not Supported value of ssl parameter, only True/False values are accepted") - super(Connection, self).__init__(db_name, db_url, username, password, readonly) + super(Connection, self).__init__(db_name, db_url, username, password, readonly, timeout=timeout) self.db_name = db_name self.db_url = db_url self.username = username