partitioncloud-server/partitioncloud/modules/classes/groupe.py

from ..db import get_db
from .album import Album

class Groupe():
    def __init__(self, uuid):
        db=get_db()

        self.uuid = uuid
        data = db.execute(
            """
            SELECT * FROM groupe
            WHERE uuid = ?
            """,
            (self.uuid,)
        ).fetchone()
        if data is None:
            raise LookupError
        self.name = data["name"]
        self.id = data["id"]
        self.users = None
        self.albums = None
        self.admins = None

    def delete(self):
        """
        Supprime le groupe, et les albums laissés orphelins (sans utilisateur)
        """
        db = get_db()
        db.execute(
            """
            DELETE FROM groupe
            WHERE id = ?
            """,
            (self.id,)
        )
        db.execute(
            """
            DELETE FROM groupe_contient_user
            WHERE groupe_id = ?
            """,
            (self.id,)
        )
        db.execute(
            """
            DELETE FROM groupe_contient_album
            WHERE groupe_id = ?
            """,
            (self.id,)
        )
        db.commit()

        # Supprime tous les albums laissés orphelins (maintenant ou plus tôt)
        data = db.execute(
            """
            SELECT id FROM album
            LEFT JOIN groupe_contient_album
                ON groupe_contient_album.album_id=album.id
            LEFT JOIN contient_user
                ON contient_user.album_id=album.id
            WHERE user_id IS NULL AND groupe_id IS NULL
            """
        ).fetchall()

        for i in data:
            album = Album(id=i["id"])
            album.delete()


    def get_users(self):
        """
        Renvoie les data["id"] des utilisateurs liés au groupe
        TODO: uniformiser le tout
        """
        db = get_db()
        return db.execute(
            """
            SELECT * FROM user
            JOIN groupe_contient_user ON user_id = user.id
            JOIN groupe ON groupe.id = groupe_id
            WHERE groupe.id = ?
            """,
            (self.id,)
        ).fetchall()

    def get_albums(self, force_reload=False):
        """
        Renvoie les uuids des albums liés au groupe
        """
        if self.albums is None or force_reload:
            db = get_db()
            data = db.execute(
                """
                SELECT * FROM album
                JOIN groupe_contient_album ON album_id = album.id
                JOIN groupe ON groupe.id = groupe_id
                WHERE groupe.id = ?
                """,
                (self.id,)
            ).fetchall()
            self.albums = [Album(uuid=i["uuid"]) for i in data]

        return self.albums

    def get_admins(self):
        """
        Renvoie les ids utilisateurs administrateurs liés au groupe
        """
        db = get_db()
        data = db.execute(
            """
            SELECT user.id FROM user
            JOIN groupe_contient_user ON user_id = user.id
            JOIN groupe ON groupe.id = groupe_id
            WHERE is_admin=1 AND groupe.id = ?
            """,
            (self.id,)
        ).fetchall()
        return [i["id"] for i in data]