270 lines
10 KiB
Python
270 lines
10 KiB
Python
#!/usr/bin/env python
|
|
import os
|
|
import signal
|
|
import subprocess
|
|
import sys
|
|
import webbrowser
|
|
import time
|
|
from threading import Thread
|
|
|
|
import win32api
|
|
import win32con
|
|
import win32gui_struct
|
|
|
|
try:
|
|
import winxpgui as win32gui
|
|
except ImportError:
|
|
import win32gui
|
|
|
|
import install
|
|
|
|
class OMLTrayIcon(object):
|
|
QUIT = 'QUIT'
|
|
SPECIAL_ACTIONS = [QUIT]
|
|
|
|
FIRST_ID = 1023
|
|
|
|
def __init__(self):
|
|
self.thread_id = win32api.GetCurrentThreadId()
|
|
launch(self)
|
|
name = "Open Media Library"
|
|
self.icon = "ico/oml.ico"
|
|
self.hover_text = name
|
|
self.on_quit = quit
|
|
|
|
menu_options = (
|
|
('Launch', None, launch),
|
|
('Quit', None, self.QUIT)
|
|
)
|
|
self._next_action_id = self.FIRST_ID
|
|
self.menu_actions_by_id = set()
|
|
self.menu_options = self._add_ids_to_menu_options(list(menu_options))
|
|
self.menu_actions_by_id = dict(self.menu_actions_by_id)
|
|
del self._next_action_id
|
|
|
|
self.default_menu_index = 0
|
|
self.window_class_name = name
|
|
|
|
message_map = {win32gui.RegisterWindowMessage("TaskbarCreated"): self.restart,
|
|
win32con.WM_DESTROY: self.destroy,
|
|
win32con.WM_COMMAND: self.command,
|
|
win32con.WM_USER+20 : self.notify,}
|
|
# Register the Window class.
|
|
window_class = win32gui.WNDCLASS()
|
|
hinst = window_class.hInstance = win32gui.GetModuleHandle(None)
|
|
window_class.lpszClassName = self.window_class_name
|
|
window_class.style = win32con.CS_VREDRAW | win32con.CS_HREDRAW;
|
|
window_class.hCursor = win32gui.LoadCursor(0, win32con.IDC_ARROW)
|
|
window_class.hbrBackground = win32con.COLOR_WINDOW
|
|
window_class.lpfnWndProc = message_map # could also specify a wndproc.
|
|
classAtom = win32gui.RegisterClass(window_class)
|
|
# Create the Window.
|
|
style = win32con.WS_OVERLAPPED | win32con.WS_SYSMENU
|
|
self.hwnd = win32gui.CreateWindow(classAtom,
|
|
self.window_class_name,
|
|
style,
|
|
0,
|
|
0,
|
|
win32con.CW_USEDEFAULT,
|
|
win32con.CW_USEDEFAULT,
|
|
0,
|
|
0,
|
|
hinst,
|
|
None)
|
|
win32gui.UpdateWindow(self.hwnd)
|
|
self.notify_id = None
|
|
self.refresh_icon()
|
|
win32gui.PumpMessages()
|
|
|
|
def _add_ids_to_menu_options(self, menu_options):
|
|
result = []
|
|
for menu_option in menu_options:
|
|
option_text, option_icon, option_action = menu_option
|
|
if callable(option_action) or option_action in self.SPECIAL_ACTIONS:
|
|
self.menu_actions_by_id.add((self._next_action_id, option_action))
|
|
result.append(menu_option + (self._next_action_id,))
|
|
else:
|
|
print ('Unknown item', option_text, option_icon, option_action)
|
|
self._next_action_id += 1
|
|
return result
|
|
|
|
def refresh_icon(self):
|
|
# Try and find a custom icon
|
|
hinst = win32gui.GetModuleHandle(None)
|
|
if os.path.isfile(self.icon):
|
|
icon_flags = win32con.LR_LOADFROMFILE | win32con.LR_DEFAULTSIZE
|
|
hicon = win32gui.LoadImage(hinst,
|
|
self.icon,
|
|
win32con.IMAGE_ICON,
|
|
0,
|
|
0,
|
|
icon_flags)
|
|
else:
|
|
print ("Can't find icon file - using default.")
|
|
hicon = win32gui.LoadIcon(0, win32con.IDI_APPLICATION)
|
|
|
|
if self.notify_id: message = win32gui.NIM_MODIFY
|
|
else: message = win32gui.NIM_ADD
|
|
self.notify_id = (self.hwnd,
|
|
0,
|
|
win32gui.NIF_ICON | win32gui.NIF_MESSAGE | win32gui.NIF_TIP,
|
|
win32con.WM_USER+20,
|
|
hicon,
|
|
self.hover_text)
|
|
win32gui.Shell_NotifyIcon(message, self.notify_id)
|
|
|
|
def restart(self, hwnd, msg, wparam, lparam):
|
|
self.refresh_icon()
|
|
|
|
def destroy(self, hwnd, msg, wparam, lparam):
|
|
if self.on_quit: self.on_quit(self)
|
|
nid = (self.hwnd, 0)
|
|
win32gui.Shell_NotifyIcon(win32gui.NIM_DELETE, nid)
|
|
win32gui.PostQuitMessage(0) # Terminate the app.
|
|
|
|
def notify(self, hwnd, msg, wparam, lparam):
|
|
if lparam==win32con.WM_LBUTTONDBLCLK:
|
|
self.execute_menu_option(self.default_menu_index + self.FIRST_ID)
|
|
elif lparam==win32con.WM_RBUTTONUP:
|
|
self.show_menu()
|
|
elif lparam==win32con.WM_LBUTTONUP:
|
|
self.show_menu()
|
|
return True
|
|
|
|
def show_menu(self):
|
|
menu = win32gui.CreatePopupMenu()
|
|
self.create_menu(menu, self.menu_options)
|
|
#win32gui.SetMenuDefaultItem(menu, 1000, 0)
|
|
|
|
pos = win32gui.GetCursorPos()
|
|
# See http://msdn.microsoft.com/library/default.asp?url=/library/en-us/winui/menus_0hdi.asp
|
|
win32gui.SetForegroundWindow(self.hwnd)
|
|
win32gui.TrackPopupMenu(menu,
|
|
win32con.TPM_LEFTALIGN,
|
|
pos[0],
|
|
pos[1],
|
|
0,
|
|
self.hwnd,
|
|
None)
|
|
win32gui.PostMessage(self.hwnd, win32con.WM_NULL, 0, 0)
|
|
|
|
def create_menu(self, menu, menu_options):
|
|
for option_text, option_icon, option_action, option_id in menu_options[::-1]:
|
|
if option_icon:
|
|
option_icon = self.prep_menu_icon(option_icon)
|
|
|
|
if option_id in self.menu_actions_by_id:
|
|
item, extras = win32gui_struct.PackMENUITEMINFO(text=option_text,
|
|
hbmpItem=option_icon,
|
|
wID=option_id)
|
|
win32gui.InsertMenuItem(menu, 0, 1, item)
|
|
else:
|
|
submenu = win32gui.CreatePopupMenu()
|
|
self.create_menu(submenu, option_action)
|
|
item, extras = win32gui_struct.PackMENUITEMINFO(text=option_text,
|
|
hbmpItem=option_icon,
|
|
hSubMenu=submenu)
|
|
win32gui.InsertMenuItem(menu, 0, 1, item)
|
|
|
|
def prep_menu_icon(self, icon):
|
|
# First load the icon.
|
|
ico_x = win32api.GetSystemMetrics(win32con.SM_CXSMICON)
|
|
ico_y = win32api.GetSystemMetrics(win32con.SM_CYSMICON)
|
|
hicon = win32gui.LoadImage(0, icon, win32con.IMAGE_ICON, ico_x, ico_y, win32con.LR_LOADFROMFILE)
|
|
|
|
hdcBitmap = win32gui.CreateCompatibleDC(0)
|
|
hdcScreen = win32gui.GetDC(0)
|
|
hbm = win32gui.CreateCompatibleBitmap(hdcScreen, ico_x, ico_y)
|
|
hbmOld = win32gui.SelectObject(hdcBitmap, hbm)
|
|
# Fill the background.
|
|
brush = win32gui.GetSysColorBrush(win32con.COLOR_MENU)
|
|
win32gui.FillRect(hdcBitmap, (0, 0, 16, 16), brush)
|
|
# unclear if brush needs to be feed. Best clue I can find is:
|
|
# "GetSysColorBrush returns a cached brush instead of allocating a new
|
|
# one." - implies no DeleteObject
|
|
# draw the icon
|
|
win32gui.DrawIconEx(hdcBitmap, 0, 0, hicon, ico_x, ico_y, 0, 0, win32con.DI_NORMAL)
|
|
win32gui.SelectObject(hdcBitmap, hbmOld)
|
|
win32gui.DeleteDC(hdcBitmap)
|
|
|
|
return hbm
|
|
|
|
def command(self, hwnd, msg, wparam, lparam):
|
|
id = win32gui.LOWORD(wparam)
|
|
self.execute_menu_option(id)
|
|
|
|
def execute_menu_option(self, id):
|
|
menu_action = self.menu_actions_by_id[id]
|
|
if menu_action == self.QUIT:
|
|
win32gui.DestroyWindow(self.hwnd)
|
|
else:
|
|
menu_action(self)
|
|
|
|
def quit(self):
|
|
nid = (self.hwnd, 0)
|
|
win32gui.Shell_NotifyIcon(win32gui.NIM_DELETE, nid)
|
|
win32api.PostThreadMessage(self.thread_id, win32con.WM_QUIT, 0, 0)
|
|
|
|
class Check(Thread):
|
|
def __init__(self, parent, pid):
|
|
self.parent = parent
|
|
self.pid = pid
|
|
Thread.__init__(self)
|
|
self.daemon = True
|
|
self.start()
|
|
|
|
def run(self):
|
|
time.sleep(60)
|
|
while check_pid(self.pid):
|
|
time.sleep(60)
|
|
self.parent.quit()
|
|
|
|
def check_pid(pidfile):
|
|
if not os.path.exists(pidfile):
|
|
return False
|
|
try:
|
|
with open(pidfile) as fd:
|
|
pid = int(fd.read())
|
|
except:
|
|
return False
|
|
try:
|
|
os.kill(pid, 0)
|
|
except:
|
|
return False
|
|
else:
|
|
return True
|
|
|
|
def launch(sysTrayIcon=None):
|
|
base = os.path.join(os.getenv('LOCALAPPDATA'), 'Open Media Library')
|
|
pid = os.path.join(base, 'data', 'openmedialibrary.pid')
|
|
if check_pid(pid):
|
|
webbrowser.open_new_tab(os.path.join(base, 'openmedialibrary', 'static', 'html', 'load.html'))
|
|
sysTrayIcon._check = Check(sysTrayIcon, pid)
|
|
elif os.path.exists(base):
|
|
python = os.path.join(base, 'platform_win32', 'pythonw.exe')
|
|
oml = os.path.join(base, 'openmedialibrary')
|
|
subprocess.Popen([python, 'oml', 'server', pid], cwd=oml, start_new_session=True)
|
|
webbrowser.open_new_tab(os.path.join(base, 'openmedialibrary', 'static', 'html', 'load.html'))
|
|
sysTrayIcon._check = Check(sysTrayIcon, pid)
|
|
else:
|
|
install.run(base)
|
|
|
|
def quit(sysTrayIcon):
|
|
base = os.path.join(os.getenv('LOCALAPPDATA'), 'Open Media Library')
|
|
pid = os.path.join(base, 'data', 'openmedialibrary.pid')
|
|
if os.path.exists(pid):
|
|
with open(pid) as fd:
|
|
data = fd.read().strip()
|
|
try:
|
|
os.kill(int(data), signal.SIGTERM)
|
|
except:
|
|
pass
|
|
|
|
if __name__ == '__main__':
|
|
if getattr(sys, 'frozen', False):
|
|
base = os.path.dirname(sys.executable)
|
|
else:
|
|
base = os.path.dirname(os.path.realpath(__file__))
|
|
base = os.path.abspath(base)
|
|
os.chdir(base)
|
|
OMLTrayIcon()
|