mew/core/repositories.py

import sqlite3
from abc import abstractmethod

from core import constants
from core.models import MusicFolder, Track, Tag, Profile


class Repository:
    TABLE_NAME = None
    MODEL_CLS = None

    @abstractmethod
    def __init__(self):
        self.cnn = sqlite3.connect(constants.DB_PATH)

    def execute(self, sql, *parameters):
        cur = self.cnn.cursor()
        cur.execute(sql, *parameters)
        return cur

    def get_by_id(self, id_):
        cur = self.execute(
            f"SELECT * FROM {self.TABLE_NAME} WHERE id=?;",
            id_
        )
        return self.MODEL_CLS(**cur.fetchone())

    def get_all(self):
        cur = self.execute(f"SELECT * FROM {self.TABLE_NAME};", )
        return [self.MODEL_CLS(**row) for row in cur.fetchall()]

    def get_by(self, field, val):
        cur = self.execute(f"SELECT * FROM {self.TABLE_NAME} WHERE {field}=?;", val)
        return [self.MODEL_CLS(**row) for row in cur.fetchall()]

    def get_by_raw_sql(self, sql, parameters=None):
        parameters = parameters if parameters is not None else []
        cur = self.execute(sql, parameters)
        return [self.MODEL_CLS(**row) for row in cur.fetchall()]

    def create(self, model, commit=False):
        fields, values = model.as_fields_and_values()
        self.execute(
            f"INSERT INTO {self.TABLE_NAME} ({', '.join(fields)}) VALUES (?);",
            *values
        )
        if commit:
            self.commit()

    def update(self, model, commit=False):
        fields, values = model.as_fields_and_values()
        self.execute(
            f"UPDATE {self.TABLE_NAME} SET ({'=?,'.join(fields)});",
            *values
        )
        if commit:
            self.commit()

    def delete(self, model, commit=False):
        self.execute(f"DELETE FROM {self.TABLE_NAME} WHERE id=?;", model.id)
        if commit:
            self.commit()

    def commit(self):
        self.cnn.commit()

    def rollback(self):
        self.cnn.rollback()

    def __del__(self):
        self.cnn.close()


class MusicFolderRepository(Repository):
    TABLE_NAME = "MusicFolders"
    MODEL_CLS = MusicFolder

    def __init__(self):
        super().__init__()


class TagRepository(Repository):
    TABLE_NAME = "Tags"
    MODEL_CLS = Tag

    def __init__(self):
        super().__init__()

    def get_by_track(self, track):
        cur = self.execute(
            f"""SELECT * 
                FROM Tags t
                INNER JOIN TracksTags tt 
                    ON tt.tag_id = t.id
                WHERE tt.track_id=?;""", track.id)
        return [self.MODEL_CLS(**row) for row in cur.fetchall()]


class TrackRepository(Repository):
    TABLE_NAME = "Tracks"
    MODEL_CLS = Track

    def __init__(self):
        super().__init__()

    def get_by_tag(self, tag):
        cur = self.execute(
            f"""SELECT * 
                FROM Tracks t
                INNER JOIN TracksTags tt 
                    ON tt.track_id = t.id
                WHERE tt.tag_id=?;""", tag.id)
        return [self.MODEL_CLS(**row) for row in cur.fetchall()]

    def get_by_tags(self, tags):
        cur = self.execute(
            f"""SELECT * 
                FROM Tracks t
                INNER JOIN TracksTags tt 
                    ON tt.track_id = t.id
                WHERE tt.tag_id in ({', '.join(['?' for _ in tags])};""", *[tag.id for tag in tags])
        return [self.MODEL_CLS(**row) for row in cur.fetchall()]

    def get_by_session(self, session):
        cur = self.execute(
            f"""SELECT * 
                FROM Tracks t
                INNER JOIN SessionsTracks st
                    ON st.track_id = t.id
                WHERE st.session_id=?;""", session.id)
        return [self.MODEL_CLS(**row) for row in cur.fetchall()]


class SessionRepository(Repository):
    TABLE_NAME = "Sessions"
    MODEL_CLS = Track

    def __init__(self):
        super().__init__()

    def get_by_track(self, track):
        cur = self.execute(
            f"""SELECT * 
                FROM Tracks t
                INNER JOIN SessionsTracks st
                    ON st.session_id = t.id
                WHERE st.track_id=?;""", track.id)
        return [self.MODEL_CLS(**row) for row in cur.fetchall()]