Soundboard.py

import yaml
import os
import logging
import sys
import pygame
from EventHandler import EventHandler


class Soundboard(object):
    __profiles = list()
    __sounds = dict()
    __current_profile = None

    def __init__(self, profiles_folder: str):
        self.__load_profiles(profiles_folder)
        self.__load_soundboard(self.__profiles)

        if self.__current_profile is None:
            raise FileNotFoundError("The soundboard cannot work without profiles")

    def __load_profiles(self, profiles_folder: str):
        """
        Loads the profile paths into memory.
        """
        self.__profiles = [os.path.join(profiles_folder, x) for x in next(os.walk(profiles_folder))[1]]
        if len(self.__profiles) == 0:
            logging.debug("No profiles available")

    def __load_soundboard(self, profile_list: list):
        """
        Loads all the sounds inside a 2D list:

        - C:/path/profiles/1
          - Loaded sound number 1
          - Loaded sound number 2
        - C:/path/profiles/2
          - Loaded sound number 5

        The number of the sound is the key that has to be pressed to play it.
        """
        self.__sounds = dict()
        pygame.mixer.pre_init(frequency=44100, size=-16, channels=2, buffer=1024)
        pygame.init()
        pygame.display.set_mode((100, 100))

        for folder in profile_list:
            profile_number = int(os.path.split(folder)[1])
            directory_tree = next(os.walk(folder))
            sound_list = dict()
            for sound_file in directory_tree[2]:
                sound_key = int(sound_file.split(".")[0])
                sound_list[sound_key] = pygame.mixer.Sound(os.path.join(folder, sound_file))

            self.__sounds[profile_number] = sound_list

        # Just get the "first" element of the dict
        for key, value in self.__sounds.items():
            self.__current_profile = key
            break

    def use_profile(self, profile_number: int):
        """
        Starts using the profile number passed.
        """
        if profile_number not in self.__sounds.keys():
            logging.debug("The profile number %d is not available" % profile_number)
            return

        self.__current_profile = profile_number

        logging.debug("Using profile number %d" % profile_number)

    def play(self, sound_key: int, loop: bool):
        """
        Plays the given sound key located in the given profile.
        """
        profile = self.__current_profile
        if sound_key in self.__sounds[profile]:
            sound = self.__sounds[profile][sound_key]
            sound.stop()
            should_loop = -1 if loop else 0
            sound.play(loops=should_loop)
            logging.debug("Playing sound %d in profile %d with loop = %d" % (sound_key, profile, should_loop))
        else:
            logging.debug("Either the profile number %d key or the sound key %d do not exist" % (profile, sound_key))

    def stop_all_sounds(self):
        """
        Stops all sounds
        """
        pygame.mixer.stop()

    def close(self):
        """
        Close the soundboard application
        """
        logging.debug("Closing application")
        pygame.quit()
        sys.exit(0)


def main():
    logging.basicConfig(stream=sys.stderr, level=logging.DEBUG)
    current_dir = os.path.dirname(__file__)
    config = dict()
    config_path = os.path.join(current_dir, "config.yml")

    with open(config_path, "r") as stream:
        config = yaml.safe_load(stream)

    profiles_dir = os.path.join(current_dir, config['profiles_folder'])

    file_map = {
        pygame.K_KP0: 0,
        pygame.K_KP1: 1,
        pygame.K_KP2: 2,
        pygame.K_KP3: 3,
        pygame.K_KP4: 4,
        pygame.K_KP5: 5,
        pygame.K_KP6: 6,
        pygame.K_KP7: 7,
        pygame.K_KP8: 8,
        pygame.K_KP9: 9,
    }

    soundboard = Soundboard(profiles_dir)
    event_handler = EventHandler(soundboard, file_map)

    while True:
        event_handler.handle(pygame.event.get())


if __name__ == "__main__":
    main()