Added opening of files via command line and added simple controls.
This commit is contained in:
parent
d089a57151
commit
ce254c8b54
6 changed files with 135 additions and 4 deletions
|
@ -4,7 +4,17 @@ from ui.main_window import MainWindow
|
||||||
|
|
||||||
|
|
||||||
class GUI:
|
class GUI:
|
||||||
def __init__(self):
|
def __init__(self, app):
|
||||||
|
self.app = app
|
||||||
|
|
||||||
self.window = MainWindow()
|
self.window = MainWindow()
|
||||||
|
|
||||||
self.window.show()
|
self.window.show()
|
||||||
|
|
||||||
|
self.connect()
|
||||||
|
|
||||||
|
def connect(self):
|
||||||
|
self.window.main_container.track_control.previous_button.triggered.connect(self.app.player.previous_track)
|
||||||
|
self.window.main_container.track_control.toggle_play_button.triggered.connect(self.app.player.toggle_playing)
|
||||||
|
self.window.main_container.track_control.stop_button.triggered.connect(self.app.player.stop)
|
||||||
|
self.window.main_container.track_control.next_button.triggered.connect(self.app.player.skip_current)
|
||||||
|
|
|
@ -3,13 +3,16 @@
|
||||||
import sys
|
import sys
|
||||||
from PyQt6.QtWidgets import QApplication
|
from PyQt6.QtWidgets import QApplication
|
||||||
from gui import GUI
|
from gui import GUI
|
||||||
|
from wobuzz.player.player import Player
|
||||||
|
|
||||||
|
|
||||||
class Wobuzz:
|
class Wobuzz:
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
self.qt_app = QApplication(sys.argv)
|
self.qt_app = QApplication([])
|
||||||
|
|
||||||
self.gui = GUI()
|
self.player = Player(sys.argv[1:])
|
||||||
|
|
||||||
|
self.gui = GUI(self)
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
|
|
3
wobuzz/player/__init__.py
Normal file
3
wobuzz/player/__init__.py
Normal file
|
@ -0,0 +1,3 @@
|
||||||
|
#!/usr/bin/python3
|
||||||
|
|
||||||
|
from .player import Player
|
86
wobuzz/player/player.py
Normal file
86
wobuzz/player/player.py
Normal file
|
@ -0,0 +1,86 @@
|
||||||
|
#!/usr/bin/python3
|
||||||
|
|
||||||
|
import os
|
||||||
|
|
||||||
|
import pygame.mixer
|
||||||
|
from pydub import AudioSegment
|
||||||
|
from pydub.effects import normalize
|
||||||
|
import io
|
||||||
|
from pygame import mixer
|
||||||
|
from .track import Track
|
||||||
|
|
||||||
|
|
||||||
|
class Player:
|
||||||
|
def __init__(self, file_paths: list=[]):
|
||||||
|
pygame.mixer.init()
|
||||||
|
self.mixer = mixer
|
||||||
|
self.mixer.music.set_endevent()
|
||||||
|
|
||||||
|
self.playing = False
|
||||||
|
self.paused = False
|
||||||
|
|
||||||
|
if not file_paths:
|
||||||
|
pass
|
||||||
|
# loading of last opened files will be implemented in the future
|
||||||
|
|
||||||
|
else:
|
||||||
|
self.current_playlist = self.load_tracks(file_paths)
|
||||||
|
|
||||||
|
if self.current_playlist:
|
||||||
|
self.playing_track = self.current_playlist[0]
|
||||||
|
self.current_playlist_index = 0
|
||||||
|
|
||||||
|
else:
|
||||||
|
self.playing_track = None
|
||||||
|
self.current_playlist_index = 0
|
||||||
|
|
||||||
|
def load_tracks(self, track_paths: list[str]):
|
||||||
|
"""
|
||||||
|
Load tracks from list of paths.
|
||||||
|
"""
|
||||||
|
|
||||||
|
tracks = []
|
||||||
|
|
||||||
|
for track_path in track_paths:
|
||||||
|
if os.path.isfile(track_path):
|
||||||
|
tracks.append(Track(track_path, True))
|
||||||
|
|
||||||
|
return tracks
|
||||||
|
|
||||||
|
def track_finished(self):
|
||||||
|
self.current_playlist_index += 1
|
||||||
|
self.playing_track = self.current_playlist[self.current_playlist_index]
|
||||||
|
|
||||||
|
self.playing_track.sound.play()
|
||||||
|
|
||||||
|
def skip_current(self):
|
||||||
|
self.mixer.stop()
|
||||||
|
self.track_finished()
|
||||||
|
|
||||||
|
def previous_track(self):
|
||||||
|
if self.current_playlist_index > 0:
|
||||||
|
self.mixer.stop()
|
||||||
|
|
||||||
|
self.current_playlist_index -= 1
|
||||||
|
self.playing_track = self.current_playlist[self.current_playlist_index]
|
||||||
|
|
||||||
|
self.playing_track.sound.play()
|
||||||
|
|
||||||
|
def toggle_playing(self):
|
||||||
|
if self.playing and self.paused:
|
||||||
|
self.mixer.pause()
|
||||||
|
self.paused = False
|
||||||
|
|
||||||
|
elif self.playing:
|
||||||
|
self.mixer.unpause()
|
||||||
|
self.paused = True
|
||||||
|
|
||||||
|
else:
|
||||||
|
self.playing_track.sound.play()
|
||||||
|
self.paused = False
|
||||||
|
self.playing = True
|
||||||
|
|
||||||
|
def stop(self):
|
||||||
|
self.mixer.stop()
|
||||||
|
self.playing = False
|
||||||
|
|
29
wobuzz/player/track.py
Normal file
29
wobuzz/player/track.py
Normal file
|
@ -0,0 +1,29 @@
|
||||||
|
#!/usr/bin/python3
|
||||||
|
|
||||||
|
from pydub import AudioSegment
|
||||||
|
from pydub.effects import normalize
|
||||||
|
from pygame.mixer import Sound
|
||||||
|
|
||||||
|
|
||||||
|
class Track:
|
||||||
|
"""
|
||||||
|
Class containing data for a track like file path, raw data...
|
||||||
|
"""
|
||||||
|
|
||||||
|
def __init__(self, path: str, cache: bool=False):
|
||||||
|
self.path = path
|
||||||
|
self.cached = cache
|
||||||
|
|
||||||
|
self.sound = self.cache() if self.cached else None
|
||||||
|
|
||||||
|
def cache(self):
|
||||||
|
audio = AudioSegment.from_mp3(self.path)
|
||||||
|
audio = normalize(audio)
|
||||||
|
|
||||||
|
wav = audio.export(format="wav")
|
||||||
|
|
||||||
|
sound = Sound(wav)
|
||||||
|
|
||||||
|
# audio_bytes = io.BytesIO(wav.read())
|
||||||
|
|
||||||
|
return sound
|
|
@ -27,6 +27,6 @@ class TrackControl(QToolBar):
|
||||||
self.play_position_slider = QSlider(Qt.Orientation.Horizontal, self)
|
self.play_position_slider = QSlider(Qt.Orientation.Horizontal, self)
|
||||||
self.addWidget(self.play_position_slider)
|
self.addWidget(self.play_position_slider)
|
||||||
|
|
||||||
self.track_length_indicator = QLabel("1:00")
|
self.track_length_indicator = QLabel("0:00")
|
||||||
self.addWidget(self.track_length_indicator)
|
self.addWidget(self.track_length_indicator)
|
||||||
|
|
||||||
|
|
Loading…
Add table
Reference in a new issue