From b659b95b4f86f8d890ab87d6a588646392d16f24 Mon Sep 17 00:00:00 2001 From: Alex Ryndin Date: Mon, 15 Mar 2021 19:13:42 +0300 Subject: [PATCH] add support for timeout parameter --- connector.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/connector.py b/connector.py index 5ae983d..b8a2afd 100644 --- a/connector.py +++ b/connector.py @@ -125,19 +125,20 @@ 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=int(timeout)) self.db_name = db_name self.db_url = db_url self.username = username self.password = password self.readonly = readonly + self.timeout = int(timeout) def close(self): pass