def test_clear_sessions(self):
file_name = '00010203-0405-0607-0809-0a0b0c0d0e0f'
try:
os.remove(os.path.dirname(os.path.realpath(__file__)) + "\\Sessions\\" + file_name)
except OSError:
pass
fake_uuid_generator = FakeUuidGenerator('{%s}' % file_name)
session_manager = SessionManager(fake_uuid_generator)
session_manager.create_session()
session_manager.clear_sessions()
self.assertEqual(session_manager.get_active_sessions(), [])
class MobTimerController(Tk):
def __init__(self, *args, **kwargs):
Tk.__init__(self, *args, **kwargs)
self.transparent_frame_monitor_index = 0
self.transparent_frame_position_index = 0
self.settings_manager = SettingsManager()
self.tips_manager = TipsManager()
self.time_options_manager = TimeSettingsManager()
self.mobber_manager = MobberManager(self.settings_manager.get_randomize_randomize_next_driver())
self.countdown_manager = CountdownManager(self)
self.session_manager = SessionManager(uuid)
self.timer_extension_count = self.settings_manager.get_timer_extension_count()
self.extensions_used = 0
atexit.register(self.session_manager.clear_sessions)
if self.session_manager.get_active_sessions().__len__() > 0:
self.quit_and_destroy_session()
self.session_manager.create_session()
self.iconbitmap(default='time-bomb.ico')
self.countdown_manager.subscribe_to_time_changes(self.show_screen_blocker_when_session_interupted)
self.theme_manager = ThemeManager()
theme = self.settings_manager.get_general_theme()
if not theme == 'none':
self.theme_manager.set_theme(theme)
monitors = ScreenUtility.get_monitors_or_default()
num_monitors = monitors.__len__()
self.parent_containers = [self]
self.containers = []
for monitor_index in range(1, num_monitors):
monitor_screen_blocker = Toplevel(self)
self.parent_containers.append(monitor_screen_blocker)
self.frame_types = (ScreenBlockerFrame, TransparentCountdownFrame, MinimalScreenBlockerFrame)
self.frames = {}
for frame_type in self.frame_types:
self.frames[frame_type] = []
for parent_container, monitor in zip(self.parent_containers, monitors):
parent_container.grid_rowconfigure(0, weight=1)
parent_container.grid_columnconfigure(0, weight=1)
container = OuterFrame(parent=parent_container, monitor=monitor)
self.containers.append(container)
container.grid(row=0, column=0, sticky=(N, S, E, W))
container.grid_rowconfigure(0, weight=1)
container.grid_columnconfigure(0, weight=1)
for frame_type in self.frame_types:
frame_instance = frame_type(container, self, self.time_options_manager, self.mobber_manager,
self.countdown_manager, self.settings_manager, self.tips_manager, self.theme_manager)
self.frames[frame_type].append(frame_instance)
frame_instance.grid(row=0, column=0, sticky=(N, S, E, W))
frame_instance.grid_rowconfigure(0, weight=1)
frame_instance.grid_columnconfigure(0, weight=1)
self.last_frame = None
self.show_screen_blocker_frame()
for frame_instance in self.frames[TransparentCountdownFrame]:
frame_instance.bind("<Enter>", self.toggle_transparent_frame_position)
self.transparent_frame_position = 0
self.title("Mob Timer")
self.bind_all("<Control-Return>", self.launch_transparent_countdown_if_blocking)
self.time_options_manager.set_countdown_time(self.settings_manager.get_timer_minutes(), self.settings_manager.get_timer_seconds())
def launch_transparent_countdown_if_blocking(self, event):
if self.frame_is_screen_blocking():
self.show_transparent_countdown_frame()
def frame_is_screen_blocking(self):
return self.last_frame == ScreenBlockerFrame or self.last_frame == MinimalScreenBlockerFrame
def show_minimal_screen_blocker_frame(self):
if self.last_frame != MinimalScreenBlockerFrame:
self.launch_blocking_Frame(MinimalScreenBlockerFrame)
self.mobber_manager.switch_next_driver()
def quit_and_destroy_session(self):
self.session_manager.clear_sessions()
self.quit()
sys.exit()
def show_screen_blocker_when_session_interupted(self, days, minutes, seconds):
if self.session_manager.get_active_sessions().__len__() == 0:
self.show_screen_blocker_frame()
self.session_manager.create_session()
def show_frame(self, frame_class):
switched_frames = False
if self.last_frame != frame_class:
for frame_instances in self.frames[frame_class]:
frame_instances.tkraise()
switched_frames = True
self.focus_force()
self.focus_set()
self.last_frame = frame_class
for container in self.parent_containers:
if isinstance(container, Toplevel):
if self.frame_is_screen_blocking():
container.deiconify()
else:
container.withdraw()
return switched_frames
def show_screen_blocker_frame(self):
if self.last_frame != ScreenBlockerFrame:
self.launch_blocking_Frame(ScreenBlockerFrame)
def launch_blocking_Frame(self, frame):
if self.show_frame(frame):
self.set_full_screen_always_on_top()
def show_transparent_countdown_frame(self, extend_minutes=None, extend_seconds=None):
if self.show_frame(TransparentCountdownFrame):
if extend_minutes is None and extend_seconds is None:
self.extensions_used = 0
self.countdown_manager.set_countdown_duration(self.time_options_manager.minutes,
self.time_options_manager.seconds)
for minimal_frame in self.frames[MinimalScreenBlockerFrame]:
minimal_frame.show_extend_time_button()
else:
self.countdown_manager.set_countdown_duration(extend_minutes, extend_seconds)
self.set_partial_screen_transparent()
def get_current_window_geometry(self):
return "{0}x{1}+0+0".format(
self.winfo_screenwidth(), self.winfo_screenheight())
def disable_resizing(self):
for container in self.containers:
container.master.resizable(0, 0)
def remove_title_bar(self):
if PlatformUtility.platform_is_mac():
return
for container in self.containers:
container.master.overrideredirect(1)
def set_always_on_top(self):
for container in self.containers:
container.master.wm_attributes("-topmost", True)
if PlatformUtility.platform_is_mac():
os.system('''/usr/bin/osascript -e 'tell app "Finder" to set frontmost of process "Python" to true' ''')
self.focus_force()
self.focus()
container.master.focus_force()
def set_full_screen_always_on_top(self):
self.set_always_on_top()
self.remove_title_bar()
self.disable_resizing()
for container in self.containers:
monitor = container.monitor
monitor_string = "{}x{}+{}+{}".format(monitor.width, monitor.height, monitor.x, monitor.y)
container.master.geometry(monitor_string)
if not PlatformUtility.platform_is_mac():
container.master.wait_visibility(container.master) # Mac removing this prevented the issue with the continue screen visibility
container.master.attributes("-alpha", 1)
def set_partial_screen_transparent(self):
self.set_always_on_top()
self.remove_title_bar()
self.disable_resizing()
for controller in self.containers:
screenwidth = self.winfo_screenwidth()
screenheight = self.winfo_screenheight()
size_percentage = self.settings_manager.get_transparent_window_screen_size_percent()
alpha = self.settings_manager.get_transparent_window_alpha_percent()
window_width = int(screenwidth * size_percentage)
window_height = int(screenheight * size_percentage)
window_size = "{0}x{1}+0+0".format(window_width, window_height)
controller.master.geometry(window_size)
controller.master.attributes("-alpha", alpha)
self.toggle_transparent_frame_position()
def fade_app(self):
for controller in self.containers:
controller.master.attributes("-alpha", self.settings_manager.get_continue_screen_blocker_window_alpha_percent())
def unfade_app(self):
for controller in self.containers:
controller.master.attributes("-alpha", 1)
def toggle_transparent_frame_position(self, e=None):
if self.state() == "withdrawn":
return
monitors = ScreenUtility.get_monitors_or_default(self)
monitor = monitors[self.transparent_frame_monitor_index]
screenwidth = monitor.width
screenheight = monitor.height
self.set_always_on_top()
self.remove_title_bar()
self.disable_resizing()
size_percentage = self.settings_manager.get_transparent_window_screen_size_percent()
window_width = int(screenwidth * size_percentage)
window_height = int(screenheight * size_percentage)
if self.transparent_frame_position_index == 0:
self.transparent_frame_position = monitor.x + screenwidth - window_width
self.transparent_frame_monitor_index = (self.transparent_frame_monitor_index + 1) % (monitors.__len__())
else:
self.transparent_frame_position = monitor.x + 0
self.transparent_frame_position_index = (self.transparent_frame_position_index + 1) % 2
bottom_left_screen = "{}x{}+{}+{}".format(window_width, window_height, self.transparent_frame_position, monitor.y +
screenheight - window_height)
self.geometry(bottom_left_screen)
def rewind_and_extend(self,minutes, seconds):
self.extensions_used += 1
self.mobber_manager.rewind_driver()
result = self.show_transparent_countdown_frame(minutes, seconds)
for minimal_frame in self.frames[MinimalScreenBlockerFrame]:
minimal_frame.show_extend_time_button()
return result
class MobTimerController(Tk):
def __init__(self, *args, **kwargs):
Tk.__init__(self, *args, **kwargs)
# self.iconbitmap(default='C:\\Users\\Chris\\OneDrive\\Git\\Pycharm\\MobTimer\\time-bomb.ico')
#TODO: iconbitmap needs to load the ico file as a string because of py2exe =/
self.time_options_manager = TimeOptionsManager()
self.mobber_manager = MobberManager()
self.countdown_manager = CountdownManager(self)
self.session_manager = SessionManager(uuid)
atexit.register(self.session_manager.clear_sessions)
if self.session_manager.get_active_sessions().__len__() > 0:
self.quit_and_destroy_session()
self.session_manager.create_session()
self.countdown_manager.subscribe_to_time_changes(
self.show_screen_blocker_when_session_interupted)
monitors = get_monitors()
num_monitors = monitors.__len__()
self.containers = [self]
for monitor_index in range(1, num_monitors):
monitor_screen_blocker = Toplevel(self)
self.containers.append(monitor_screen_blocker)
self.frame_types = (ScreenBlockerFrame, TransparentCountdownFrame)
self.frames = {}
for frame_type in self.frame_types:
self.frames[frame_type] = []
for container in self.containers:
container_frame = Frame(container)
container_frame.grid(row=0, column=0, sticky=N + S + E + W)
container_frame.grid_rowconfigure(0, weight=1)
container_frame.grid_columnconfigure(0, weight=1)
for frame_type in self.frame_types:
frame_instance = frame_type(container_frame, self,
self.time_options_manager,
self.mobber_manager,
self.countdown_manager)
self.frames[frame_type].append(frame_instance)
frame_instance.grid(row=0, column=0, sticky="nsew")
self.last_frame = None
self.show_screen_blocker_frame()
for frame_instance in self.frames[TransparentCountdownFrame]:
frame_instance.bind("<Enter>",
self.toggle_transparent_frame_position)
self.transparent_frame_position = 0
self.title("Mob Timer")
def quit_and_destroy_session(self):
self.session_manager.clear_sessions()
self.quit()
sys.exit()
def show_screen_blocker_when_session_interupted(self, days, minutes,
seconds):
if self.session_manager.get_active_sessions().__len__() == 0:
self.show_screen_blocker_frame()
self.session_manager.create_session()
def show_frame(self, frame_class):
switched_frames = False
if self.last_frame != frame_class:
for frame_instances in self.frames[frame_class]:
frame_instances.tkraise()
switched_frames = True
self.last_frame = frame_class
return switched_frames
def show_screen_blocker_frame(self):
if self.show_frame(ScreenBlockerFrame):
self.mobber_manager.switch_navigator_driver()
self.set_full_screen_always_on_top()
def show_transparent_countdown_frame(self):
if self.show_frame(TransparentCountdownFrame):
self.set_partial_screen_transparent()
def get_current_window_geometry(self):
return "{0}x{1}+0+0".format(self.winfo_screenwidth(),
self.winfo_screenheight())
def disable_resizing(self):
for container in self.containers:
container.resizable(0, 0)
def remove_title_bar(self):
for container in self.containers:
container.overrideredirect(1)
def set_always_on_top(self):
for container in self.containers:
container.wm_attributes("-topmost", 1)
def set_full_screen_always_on_top(self):
self.set_always_on_top()
self.remove_title_bar()
self.disable_resizing()
top_left_screen = "+0+0"
monitors = get_monitors()
for container, monitor in zip(self.containers, monitors):
monitor_string = "{}x{}+{}+{}".format(monitor.width,
monitor.height, monitor.x,
monitor.y)
container.geometry(monitor_string)
container.wait_visibility(container)
container.attributes("-alpha", 1)
def set_partial_screen_transparent(self):
self.set_always_on_top()
self.remove_title_bar()
self.disable_resizing()
for controller in self.containers:
screenwidth = self.winfo_screenwidth()
screenheight = self.winfo_screenheight()
window_width = int(screenwidth * 0.3)
window_height = int(screenheight * 0.3)
window_size = "{0}x{1}+0+0".format(window_width, window_height)
controller.geometry(window_size)
controller.attributes("-alpha", 0.3)
self.toggle_transparent_frame_position()
def toggle_transparent_frame_position(self, e=None):
screenwidth = self.winfo_screenwidth()
screenheight = self.winfo_screenheight()
self.set_always_on_top()
self.remove_title_bar()
self.disable_resizing()
window_width = int(screenwidth * 0.3)
window_height = int(screenheight * 0.3)
if self.transparent_frame_position == 0:
self.transparent_frame_position = screenwidth - window_width
else:
self.transparent_frame_position = 0
bottom_left_screen = "+{}+{}".format(self.transparent_frame_position,
screenheight - window_height)
for controller in self.containers:
controller.geometry(bottom_left_screen)
class MobTimerController(Tk):
def __init__(self, *args, **kwargs):
Tk.__init__(self, *args, **kwargs)
self.toggle_transparent_frame_position_function = self.toggle_transparent_frame_position_enabled
self.transparent_frame_monitor_index = 0
self.transparent_frame_position_index = 0
self.settings_manager = SettingsManager()
self.tips_manager = TipsManager()
self.time_options_manager = TimeSettingsManager()
self.date_time_utility = DateTimeUtility()
self.file_utilities = FileUtilities()
self.event_logging_manager = EventLoggingManager(self.file_utilities, self.date_time_utility)
self.mobber_manager = MobberManager(self.settings_manager.get_randomize_randomize_next_driver())
self.countdown_manager = CountdownManager(self)
self.session_manager = SessionManager(uuid)
self.timer_extension_count = self.settings_manager.get_timer_extension_count()
self.extensions_used = 0
atexit.register(self.session_manager.clear_sessions)
if self.session_manager.get_active_sessions().__len__() > 0:
self.quit_and_destroy_session()
self.session_manager.create_session()
if sys.platform != 'darwin':
self.iconbitmap(default='time-bomb.ico')
self.countdown_manager.subscribe_to_time_changes(self.show_screen_blocker_when_session_interupted)
self.theme_manager = ThemeManager()
theme = self.settings_manager.get_general_theme()
if not theme == 'none':
self.theme_manager.set_theme(theme)
monitors = ScreenUtility.get_monitors_or_default()
num_monitors = monitors.__len__()
self.parent_containers = [self]
self.containers = []
for monitor_index in range(1, num_monitors):
monitor_screen_blocker = Toplevel(self)
self.parent_containers.append(monitor_screen_blocker)
self.frame_types = (ScreenBlockerFrame, TransparentCountdownFrame, MinimalScreenBlockerFrame)
self.frames = {}
for frame_type in self.frame_types:
self.frames[frame_type] = []
for parent_container, monitor in zip(self.parent_containers, monitors):
parent_container.grid_rowconfigure(0, weight=1)
parent_container.grid_columnconfigure(0, weight=1)
container = OuterFrame(parent=parent_container, monitor=monitor)
self.containers.append(container)
container.grid(row=0, column=0, sticky=(N, S, E, W))
container.grid_rowconfigure(0, weight=1)
container.grid_columnconfigure(0, weight=1)
for frame_type in self.frame_types:
frame_instance = frame_type(container, self, self.time_options_manager, self.mobber_manager,
self.countdown_manager, self.settings_manager, self.tips_manager,
self.theme_manager)
self.frames[frame_type].append(frame_instance)
frame_instance.grid(row=0, column=0, sticky=(N, S, E, W))
frame_instance.grid_rowconfigure(0, weight=1)
frame_instance.grid_columnconfigure(0, weight=1)
self.last_frame = None
self.show_screen_blocker_frame()
for frame_instance in self.frames[TransparentCountdownFrame]:
frame_instance.bind("<Enter>", self.toggle_transparent_frame_position)
self.transparent_frame_position = 0
self.title("Mob Timer")
self.bind_all("<Control-Return>", self.launch_transparent_countdown_if_blocking)
self.time_options_manager.set_countdown_time(self.settings_manager.get_timer_minutes(),
self.settings_manager.get_timer_seconds())
self.dojo_manager = DojoManager(self)
if self.settings_manager.get_event_logging_enabled():
self.mobber_manager.subscribe_to_mobber_add(self.add_mobber_callback)
self.mobber_manager.subscribe_to_mobber_remove(self.remove_mobber_callback)
def remove_mobber_callback(self, mobber_name):
self.event_logging_manager.log(f'Removed: {mobber_name}')
def add_mobber_callback(self, mobber_name):
self.event_logging_manager.log(f'Added: {mobber_name}')
def launch_transparent_countdown_if_blocking(self, event=None):
if self.frame_is_screen_blocking():
self.show_transparent_countdown_frame()
def frame_is_screen_blocking(self):
return self.last_frame == ScreenBlockerFrame or self.last_frame == MinimalScreenBlockerFrame
def show_minimal_screen_blocker_frame(self):
self.toggle_transparent_frame_position_function = self.toggle_transparent_frame_position_enabled
self.theme_manager.reset_flashing_background_colors_to_normal()
if self.last_frame != MinimalScreenBlockerFrame:
self.launch_blocking_Frame(MinimalScreenBlockerFrame)
self.mobber_manager.switch_next_driver()
def quit_and_destroy_session(self):
self.session_manager.clear_sessions()
self.quit()
sys.exit()
def show_screen_blocker_when_session_interupted(self, days, minutes, seconds):
if self.session_manager.get_active_sessions().__len__() == 0:
self.show_screen_blocker_frame()
self.session_manager.create_session()
def show_frame(self, frame_class):
switched_frames = False
if self.last_frame != frame_class:
for frame_instances in self.frames[frame_class]:
frame_instances.tkraise()
switched_frames = True
self.focus_force()
self.focus_set()
self.last_frame = frame_class
for container in self.parent_containers:
if isinstance(container, Toplevel):
if self.frame_is_screen_blocking():
container.deiconify()
else:
container.withdraw()
container.focus_force()
container.focus_set()
return switched_frames
def show_screen_blocker_frame(self):
if self.last_frame != ScreenBlockerFrame:
self.launch_blocking_Frame(ScreenBlockerFrame)
def launch_blocking_Frame(self, frame):
if self.show_frame(frame):
self.set_full_screen_always_on_top()
def show_transparent_countdown_frame(self, extend_minutes=None, extend_seconds=None):
if self.show_frame(TransparentCountdownFrame):
if extend_minutes is None and extend_seconds is None:
self.extensions_used = 0
self.countdown_manager.set_countdown_duration(self.time_options_manager.minutes,
self.time_options_manager.seconds)
for minimal_frame in self.frames[MinimalScreenBlockerFrame]:
minimal_frame.show_extend_time_button()
else:
self.countdown_manager.set_countdown_duration(extend_minutes, extend_seconds)
self.set_partial_screen_transparent()
def get_current_window_geometry(self):
return "{0}x{1}+0+0".format(
self.winfo_screenwidth(), self.winfo_screenheight())
def disable_resizing(self):
for container in self.containers:
container.master.resizable(0, 0)
def remove_title_bar(self):
if PlatformUtility.platform_is_mac():
return
for container in self.containers:
container.master.overrideredirect(1)
def set_always_on_top(self):
for container in self.containers:
container.master.wm_attributes("-topmost", True)
if PlatformUtility.platform_is_mac():
os.system('''/usr/bin/osascript -e 'tell app "Finder" to set frontmost of process "Python" to true' ''')
self.focus_force()
self.focus()
container.master.focus_force()
def set_full_screen_always_on_top(self):
self.set_always_on_top()
self.remove_title_bar()
self.disable_resizing()
for container in self.containers:
monitor = container.monitor
monitor_string = "{}x{}+{}+{}".format(monitor.width, monitor.height, monitor.x, monitor.y)
container.master.geometry(monitor_string)
if not PlatformUtility.platform_is_mac():
container.master.wait_visibility(
container.master) # Mac removing this prevented the issue with the continue screen visibility
container.master.attributes("-alpha", 1)
def set_partial_screen_transparent(self):
self.set_always_on_top()
self.remove_title_bar()
self.disable_resizing()
for controller in self.containers:
screenwidth = self.winfo_screenwidth()
screenheight = self.winfo_screenheight()
size_percentage = self.settings_manager.get_transparent_window_screen_size_percent()
alpha = self.settings_manager.get_transparent_window_alpha_percent()
window_width = int(screenwidth * size_percentage)
window_height = int(screenheight * size_percentage)
window_size = "{0}x{1}+0+0".format(window_width, window_height)
controller.master.geometry(window_size)
controller.master.attributes("-alpha", alpha)
self.toggle_transparent_frame_position()
def fade_app(self):
for controller in self.containers:
controller.master.attributes("-alpha",
self.settings_manager.get_continue_screen_blocker_window_alpha_percent())
def unfade_app(self):
for controller in self.containers:
controller.master.attributes("-alpha", 1)
def flash_unobtrusive_transparent_countdown_frame(self):
self.toggle_transparent_frame_position_function = self.toggle_transparent_frame_position_disabled
for container in self.containers:
container.master.attributes("-alpha", 1)
container.focus_force()
container.focus_set()
def toggle_transparent_frame_position_disabled(self):
pass
def toggle_transparent_frame_position(self, e=None):
if self.state() == "withdrawn":
return
self.toggle_transparent_frame_position_function()
def toggle_transparent_frame_position_enabled(self):
monitors = ScreenUtility.get_monitors_or_default(self)
monitor = monitors[self.transparent_frame_monitor_index]
screenwidth = monitor.width
screenheight = monitor.height
self.set_always_on_top()
self.remove_title_bar()
self.disable_resizing()
size_percentage = self.settings_manager.get_transparent_window_screen_size_percent()
window_width = int(screenwidth * size_percentage)
window_height = int(screenheight * size_percentage)
if self.transparent_frame_position_index == 0:
self.transparent_frame_position = monitor.x + screenwidth - window_width
self.transparent_frame_monitor_index = (self.transparent_frame_monitor_index + 1) % (monitors.__len__())
else:
self.transparent_frame_position = monitor.x + 0
self.transparent_frame_position_index = (self.transparent_frame_position_index + 1) % 2
bottom_left_screen = "{}x{}+{}+{}".format(window_width, window_height, self.transparent_frame_position,
monitor.y +
screenheight - window_height)
self.geometry(bottom_left_screen)
def rewind_and_extend(self, minutes, seconds):
self.extensions_used += 1
self.mobber_manager.rewind_driver()
result = self.show_transparent_countdown_frame(minutes, seconds)
for minimal_frame in self.frames[MinimalScreenBlockerFrame]:
minimal_frame.show_extend_time_button()
return result