various.py

import pygame
import time
import sys
from logger import logger as log

class TimerObj:
    def __init__(
            self,
            duration: int,
            starting: float = 0,
            paused: bool = False
        ) -> None:
        '''
        duration: duration of the timer
        starting: starting point of the timer
        paused: create paused timer when set to True
        '''

        self.duration = duration
        self.paused = paused

        self.time = starting
    
    def tick(self, delta_time: float) -> bool:
        if self.paused:
            return False
        
        self.time += delta_time

        if self.time > self.duration:
            self.time %= self.duration
            return True
    
    def pause(self): self.paused = True
    def resume(self): self.paused = False


class key_map:
    def __init__(self,up,down,left,right):
        log.debug(f'Initializing keybindings: {up}, {down}, {left}, {right}')

        self.up =    pygame.key.key_code(up)
        self.down =  pygame.key.key_code(down)
        self.left =  pygame.key.key_code(left)
        self.right = pygame.key.key_code(right)

        self.keys = [self.up,self.down,self.left,self.right]

    def __contains__(self, key) -> bool:
        return key in self.keys


def timer(func: callable) -> callable:

    def wrapper(*args, **kwargs) -> object:
        log.debug(f'Starting Benchmark on function: {func.__name__}')
        tick = time.perf_counter()
        output = func(*args, **kwargs)
        tock = time.perf_counter()
        log.debug(f'Benchmark ended on function: {func.__name__}')
        log.debug(f'Benchmark result: {tock-tick} (s)')

        return output
    
    return wrapper


def print_new_game(running: str = '') -> None:
    """Simple function to better separete the logs of each game"""
    log.debug('')
    log.debug('========== NEW GAME STARTING ==========')
    log.debug(f'Current time: {time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())}')
    log.debug(f'OS: {sys.platform}')
    log.debug(f'Python version: {sys.version}')
    log.debug(f'Pygame version: {pygame.version.ver}')
    if running:
        log.debug(f'Running: {running}')
    log.debug('')