libreremo_telegram_bot/database_lib.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import logging
import libreremo_res as res
import sqlite3 as sql3

logger = logging.getLogger(__name__)


class Database:

    def __init__(self, filename):
        self.filename = filename
        self.con = None
        self.cur = None

    def _execute(self, sql, factory):
        result_list = None
        try:
            self.con = sql3.connect(self.filename)
            self.con.row_factory = factory
            self.cur = self.con.cursor()
            logging.info("Connection to db open")
            self.cur.execute(sql)
            result_list = self.cur.fetchall()
        except sql3.Error as error:
            logging.info("an error occured", error)
        finally:
            if self.con:
                self.con.close()
                logging.info("Connection to db closed")
            return result_list

    def select_books_by_title(self, query):
        query = format_query(query)
        sql = f"""
        SELECT title, author_sort, path FROM books
            WHERE title LIKE "{query}";
        """
        return self._execute(sql, res.book_factory)

    def select_books_by_author(self, query):
        query = format_query(query)
        sql = f"""
        SELECT title, author_sort, path FROM books
        WHERE books.author_sort LIKE '%' || 
        (SELECT sort FROM authors 
            WHERE sort LIKE "{query}") || '%';
        """
        return self._execute(sql, res.book_factory)

    def select_authors(self, query):
        query = format_query(query)
        sql = f"""
        SELECT name, sort FROM authors
        WHERE name LIKE "{query}"; 
        """
        return self._execute(sql, res.author_factory)


def format_query(query):
    formatted_query = "%" + query.replace(" ", "%") + "%"
    return formatted_query