41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
import mariadb
|
|
|
|
import settings
|
|
|
|
|
|
class MariadbConnector:
|
|
def __init__(self):
|
|
# Settings
|
|
self.user = settings.MY_LOGIN
|
|
self.password = settings.MY_PASSWORD
|
|
self.host = settings.MY_HOST
|
|
self.port = int(settings.MY_PORT)
|
|
self.database = settings.MY_DATABASE
|
|
|
|
self.conn = self._get_connection()
|
|
self.cursor = self._get_cursor()
|
|
|
|
def _get_connection(self) -> mariadb.connections.Connection:
|
|
return mariadb.connect(user=self.user,
|
|
password=self.password,
|
|
host=self.host,
|
|
port=self.port,
|
|
database=self.database)
|
|
|
|
def _get_cursor(self) -> mariadb.cursors.Cursor:
|
|
return self.conn.cursor()
|
|
|
|
def select(self, query_string: str, as_list=True):
|
|
cursor = self.cursor
|
|
|
|
cursor.execute(query_string)
|
|
if not as_list:
|
|
return cursor.fetchone()
|
|
return cursor.fetchall()
|
|
|
|
def insert(self, query_string) -> int:
|
|
cursor = self.cursor
|
|
|
|
cursor.execute(query_string)
|
|
return cursor.insert_id
|