diff --git a/runner.py b/runner.py index f405849..9dfd170 100644 --- a/runner.py +++ b/runner.py @@ -1,5 +1,5 @@ import platform -from tray_icons.system_tray_icon_base import * +from modules.usage_manager import * if __name__ == "__main__": credential_manager = CredentialManager() @@ -25,6 +25,7 @@ main.run() else: print("We do not support '{}' OS yet. Let's try...".format(platform.system())) + from tray_icons.system_tray_icon_base import * utils = Utils() main = SystemTrayIcon(credential_manager, data_usage, utils) main.start_tray_icon() diff --git a/tray_icons/system_tray_icon_ubuntu.py b/tray_icons/system_tray_icon_ubuntu.py index cd15e02..d151116 100644 --- a/tray_icons/system_tray_icon_ubuntu.py +++ b/tray_icons/system_tray_icon_ubuntu.py @@ -1,7 +1,6 @@ from tray_icons.system_tray_icon_base import * from PIL import ImageFont - -import pystray._appindicator +import pystray._appindicator # This is required. Do not delete. class UbuntuUtils(Utils): diff --git a/tray_icons/system_tray_icon_windows.py b/tray_icons/system_tray_icon_windows.py index 59cdcbd..40bdda2 100644 --- a/tray_icons/system_tray_icon_windows.py +++ b/tray_icons/system_tray_icon_windows.py @@ -1,5 +1,6 @@ from tray_icons.system_tray_icon_base import * from PIL import ImageFont +import pystray._win32 # This is required. Do not delete. class WindowsUtils(Utils):