From e2df0b4ab8078ef095e7f109f1cb15de44381535 Mon Sep 17 00:00:00 2001
From: Males Tomlinson <47947037+DMT07@users.noreply.github.com>
Date: Fri, 15 May 2020 13:28:30 +0200
Subject: [PATCH] Added row_factory argument to class __init__

Allow user to set a row factory handler in a more elegant way.
---
 sqlite3worker.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/sqlite3worker.py b/sqlite3worker.py
index cfb9ab2..3db6889 100644
--- a/sqlite3worker.py
+++ b/sqlite3worker.py
@@ -54,7 +54,7 @@ class Sqlite3Worker(threading.Thread):
         sql_worker.close()
     """
 
-    def __init__(self, file_name, max_queue_size=100):
+    def __init__(self, file_name, max_queue_size=100, row_factory=None):
         """Automatically starts the thread.
 
         Args:
@@ -67,6 +67,7 @@ def __init__(self, file_name, max_queue_size=100):
             file_name, check_same_thread=False, detect_types=sqlite3.PARSE_DECLTYPES
         )
         self._sqlite3_cursor = self._sqlite3_conn.cursor()
+        self._sqlite3_cursor.row_factory = row_factory
         self._sql_queue = Queue.Queue(maxsize=max_queue_size)
         self._results = {}
         self._max_queue_size = max_queue_size