Mixonomer/music/db/part_generator.py

from music.model.user import User
from music.model.playlist import Playlist
import logging

logger = logging.getLogger(__name__)


class PartGenerator:

    def __init__(self, user: User = None, username: str = None):
        self.queried_playlists = []
        self.parts = []

        if user:
            self.user = user
        elif username:
            pulled_user = User.collection.filter('username', '==', username.strip().lower()).get()
            if pulled_user:
                self.user = pulled_user
            else:
                raise LookupError(f'{username} not found')
        else:
            raise NameError('no user info provided')

    def reset(self):
        self.queried_playlists = []
        self.parts = []

    def get_recursive_parts(self, name):
        logger.info(f'getting part from {name} for {self.user.username}')

        self.reset()
        self.process_reference_by_name(name)

        return [i for i in {i for i in self.parts}]

    def process_reference_by_name(self, name):

        playlist = Playlist.collection.parent(self.user.key).filter('name', '==', name).get()

        if playlist is not None:

            if playlist.id not in self.queried_playlists:

                self.parts += playlist.parts
                self.queried_playlists.append(playlist.id)

                for i in playlist.playlist_references:
                    if i.id not in self.queried_playlists:
                        self.process_reference_by_reference(i)

            else:
                logger.warning(f'playlist reference {name} already queried')

        else:
            logger.warning(f'playlist reference {name} not found')

    def process_reference_by_reference(self, ref):

        if ref.id not in self.queried_playlists:
            playlist_reference_object = ref.get().to_dict()
            self.parts += playlist_reference_object['parts']
            self.queried_playlists.append(ref.id)

            for i in playlist_reference_object['playlist_references']:
                self.process_reference_by_reference(i)

        else:
            logger.warning(f'playlist reference {ref.get().to_dict()["name"]} already queried')