player.py

import curses
import _curses

import pygame

from timetrack import Watch

from time import strftime, gmtime
import math
import sys
import os

pygame.mixer.init()

BORDER_CHARS = { # │ ─ ┌ ┬ ┐ ├ ┼ ┤ └ ┴ ┘
	"flat"	: {
		"vertical"   : "│",
		"horizontal" : "─",
	},

	"corners" : {
		"down_right" : "┌",
		"down_left"  : "┐",
		"up_right"   : "└",
		"up_left"	: "┘",
 	},

	"half_crosses" : {
		"vertical_right"  : "├",
		"vertical_left"   : "┤",
		"horizontal_down" : "┬",
		"horizontal_up"   : "┴",
	},

	"cross"   : "┼"
}

def main(stdscr):
	stdscr.nodelay(True)
	curses.curs_set(False)

	current_file = sys.argv[1]
	name = os.path.basename(current_file)

	stopped = True
	is_paused = True

	snd = pygame.mixer.Sound(current_file)
	chn = pygame.mixer.Channel(1)

	snd_length = snd.get_length()

	w = Watch()

	while True:
		lines, cols = stdscr.getmaxyx()
		
		# Creating a square border around the screen
		stdscr.addstr(0, 0, BORDER_CHARS["corners"]["down_right"] + (BORDER_CHARS["flat"]["horizontal"] * (cols - 2)) + BORDER_CHARS["corners"]["down_left"])
		
		for i in range(1, lines - 1):
			stdscr.addstr(i, 0, BORDER_CHARS["flat"]["vertical"].ljust(cols - 1) + BORDER_CHARS["flat"]["vertical"])

		# Using insstr because addstr fails when trying to place a character at the bottom right corner
		stdscr.insstr(lines - 1, 0, BORDER_CHARS["corners"]["up_right"] + (BORDER_CHARS["flat"]["horizontal"] * (cols - 2)) + BORDER_CHARS["corners"]["up_left"])



		# Adding sound metadata
		# Name
		stdscr.addstr(2, 2, name.center(cols - 4))
		stdscr.addstr(2, 2, "Currently playing:")

		# Duration
		stdscr.addstr(4, 2, (strftime("%H:%M:%S", gmtime(w.get_time())) + "/" + strftime("%H:%M:%S", gmtime(snd_length))).rjust(cols - 4))

		SPACE_TAKEN = 24 # Amount of space the time, borders and other things take around the progress bar

		if stopped:
			if w.start_time == 0: # Program has just been opened, show no progress bar
				chars = ""
			else:
				chars = "█" * (cols - SPACE_TAKEN)
		else:
			char_num = (w.get_time() / snd_length) * (cols - SPACE_TAKEN)

			chars = "█" * math.floor(char_num)

			if char_num - math.floor(char_num) > 0.5:
				chars += "▌"

		stdscr.addstr(4, 4, chars)

		# Adding status
		if is_paused:
			if stopped:
				stdscr.addstr(4, 2, u"■")
			else:
				stdscr.addstr(4, 2, u"‖")
		else:
			stdscr.addstr(4, 2, u"►")

		# Adding instructions
		stdscr.addstr(lines - 2, 2, "SPACE to pause/unpause, S to stop, Q to quit.".ljust(cols - 4), curses.A_REVERSE)

		stdscr.refresh()

		if w.get_time() > snd_length:
			stopped = True
			is_paused = True
			w.stop()
			chn.stop()

		try:
			key = stdscr.getch()
		except _curses.ERR:
			continue

		if key == ord(' '):
			if is_paused:
				if stopped:
					chn.play(snd)
					stopped = False
					w.start()
				else:
					chn.unpause()
					w.unpause()
			else:
				chn.pause()
				w.pause()

			is_paused = not is_paused

		if (key == ord('s')) and (stopped == False):
			is_paused = True
			stopped = True
			chn.stop()
			w.stop()

		if key == ord('q'):
			chn.stop()
			w.stop()
			break


curses.wrapper(main)