deemix-py/deemix/ui/MainFrame.py

68 lines
2.0 KiB
Python

#!/usr/bin/env python3
import wx
from deemix.ui.SettingsDialog import SettingsDialog
from deemix.app.functions import getIDFromLink, getTypeFromLink
from deemix.app.downloader import download_track, download_album, download_playlist
from deemix.app.settings import initSettings
menuIDs = {
"SETTINGS": 1
}
class MainFrame(wx.Frame):
def __init__(self, *args, **kwargs):
super().__init__(parent=None, title='deemix')
panel = wx.Panel(self)
self.settings = initSettings()
# Menubar
menubar = wx.MenuBar()
fileMenu = wx.Menu()
settingsItem = fileMenu.Append(menuIDs['SETTINGS'], 'Settings', 'Edit Settings')
fileMenu.AppendSeparator()
quitItem = fileMenu.Append(wx.ID_EXIT, 'Quit', 'Quit application')
menubar.Append(fileMenu, '&File')
self.SetMenuBar(menubar)
self.Bind(wx.EVT_MENU, self.close_app, quitItem)
self.Bind(wx.EVT_MENU, self.open_settings, settingsItem)
# Main app
main_sizer = wx.BoxSizer(wx.VERTICAL)
search_sizer = wx.BoxSizer(wx.HORIZONTAL)
main_sizer.Add(search_sizer, 0, wx.EXPAND, 5)
self.text_ctrl = wx.TextCtrl(panel)
search_sizer.Add(self.text_ctrl, 1, wx.ALL, 5)
my_btn = wx.Button(panel, label='Download')
my_btn.Bind(wx.EVT_BUTTON, self.download_track)
search_sizer.Add(my_btn, 0, wx.ALL, 5)
panel.SetSizer(main_sizer)
self.Show()
def download_track(self, event):
value = self.text_ctrl.GetValue()
if not value:
print("You didn't enter anything!")
return None
type = getTypeFromLink(value)
id = getIDFromLink(value, type)
print(type, id)
if type == "track":
download_track(id, self.settings)
elif type == "album":
download_album(id, self.settings)
elif type == "playlist":
download_playlist(id, self.settings)
self.text_ctrl.SetValue("")
def close_app(self, event):
self.Close()
def open_settings(self, event):
settings_dialog = SettingsDialog(self.settings, self)
res = settings_dialog.ShowModal()
if res == wx.ID_OK:
self.settings = settings_dialog.GetSettings()
settings_dialog.Destroy()