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