add driver property for Database

Author: co0lc0der

simple_query_builder/database.py

@@ -17,19 +17,21 @@ def __call__(cls, *args, **kwargs):
 
 
 class DataBase(metaclass=MetaSingleton):
-    db_name = "db.db"
+    driver = "sqlite"
+    db_name = ":memory:"
     conn = None
     cursor = None
 
-    def connect(self, db_name=""):
-        if db_name != "":
     def connect(self, db_name: str = "", uri: bool = False):
         if db_name:
             self.db_name = db_name
 
         if self.conn is None:
-            self.conn = sqlite3.connect(self.db_name, uri=uri)
-            self.cursor = self.conn.cursor()
+            if self.driver == "sqlite":
+                self.conn = sqlite3.connect(self.db_name, uri=uri)
+                self.cursor = self.conn.cursor()
+            else:
+                print("Wrong DB driver. At present time it's supported 'sqlite' only")
 
         return self.conn
 

simple_query_builder/querybuilder.py

@@ -212,7 +212,7 @@ def column(self, column: Union[str, int] = 0):
             self.set_error(f"Incorrect type of column in {inspect.stack()[0][3]} method. Result dict is {self._result_dict}")
             return self
 
-        self.query("", (), self._FETCH_COLUMN, column)
+        self.query(fetch=self._FETCH_COLUMN, column=column)
         return self._result
 
     def pluck(self, key: Union[str, int] = 0, column: Union[str, int] = 1):
@@ -245,7 +245,7 @@ def get_last(self):
         return self._result[-1]
 
     def exists(self) -> bool:
-        result = self.one()
+        self.one()
         return self._count > 0
 
     def _prepare_aliases(self, items: Union[str, list, dict], as_list: bool = False) -> Union[str, list]:
@@ -688,6 +688,7 @@ def __str__(self):
         return self.get_sql(False)
 
     def create_view(self, view_name: str, add_exists: bool = True):
+        # this method will be moved to another class
         if not view_name:
             self.set_error(f"Empty view_name in {inspect.stack()[0][3]} method")
             return self
@@ -703,6 +704,7 @@ def create_view(self, view_name: str, add_exists: bool = True):
         return self
 
     def drop_view(self, view_name: str, add_exists: bool = True):
+        # this method will be moved to another class
         if not view_name:
             self.set_error(f"Empty view_name in {inspect.stack()[0][3]} method")
             return self