1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
#!/usr/bin/python3
#
# Copyright (C) 2020-2024 Soleta Networks <info@soleta.eu>
#
# This program is free software: you can redistribute it and/or modify it under
# the terms of the GNU Affero General Public License as published by the
# Free Software Foundation; either version 3 of the License, or
# (at your option) any later version.
import time
import multiprocessing as mp
import psutil
from PIL import Image, ImageDraw
from pystray import Icon, Menu, MenuItem
SERVICE_POLL_INTERVAL = 5
def _create_default_image():
"""
Creates a default image for the tray icon. Use in case
no favicon.ico is found. The image will be a blue circle.
"""
width = height = 250
circle_color = (45, 158, 251)
# Create a new image with a transparent background
image = Image.new('RGBA', (width, height), (0, 0, 0, 0))
dc = ImageDraw.Draw(image)
# Draw circle
circle_radius = min(width, height) // 2 - 10
circle_center = (width // 2, height // 2)
dc.ellipse(
(circle_center[0] - circle_radius, circle_center[1] - circle_radius,
circle_center[0] + circle_radius, circle_center[1] + circle_radius),
fill=circle_color
)
return image
def create_icon_image():
try:
image = Image.open(r'./favicon.ico')
image = Image.composite(image, Image.new('RGB', image.size, 'white'), image)
except FileNotFoundError:
image = _create_default_image()
return image
def create_systray():
menu = Menu(MenuItem('ogclient service running',
lambda icon, item: None))
icon = Icon('ogClient', create_icon_image(), menu=menu)
icon.run()
def is_ogclient_service_active():
service_name = 'ogclient'
try:
service = psutil.win_service_get(service_name)
service = service.as_dict()
return service.get('status') == 'running'
except Exception:
return False
def _session_check_loop():
systray_process = None
try:
while True:
session_status = is_ogclient_service_active()
is_systray_active = systray_process and systray_process.is_alive()
if session_status and not is_systray_active:
systray_process = mp.Process(target=create_systray, daemon=True)
systray_process.start()
elif not session_status and is_systray_active:
systray_process.terminate()
systray_process.join()
systray_process = None
time.sleep(SERVICE_POLL_INTERVAL)
except KeyboardInterrupt:
if systray_process and systray_process.is_alive():
systray_process.terminate()
systray_process.join()
def main():
mp.freeze_support()
mp.set_start_method('spawn')
_session_check_loop()
if __name__ == "__main__":
main()
|
