From 26214e3f7c5343fa30d2ff1ae71a6cf7197b6f3e Mon Sep 17 00:00:00 2001 From: Christian Pointner Date: Wed, 19 May 2021 16:52:36 +0200 Subject: initial version of syncoid autosuspender --- .../zfs/syncoid/templates/autosuspend.py.j2 | 92 ++++++++++++++++++++++ 1 file changed, 92 insertions(+) create mode 100644 roles/storage/zfs/syncoid/templates/autosuspend.py.j2 (limited to 'roles/storage/zfs/syncoid/templates/autosuspend.py.j2') diff --git a/roles/storage/zfs/syncoid/templates/autosuspend.py.j2 b/roles/storage/zfs/syncoid/templates/autosuspend.py.j2 new file mode 100644 index 00000000..e54ddd1b --- /dev/null +++ b/roles/storage/zfs/syncoid/templates/autosuspend.py.j2 @@ -0,0 +1,92 @@ +#!/usr/bin/env python3 + +import sys +import dbus +import re +import datetime +#import pprint + + +class AutoSuspender(object): + + def __init__(self): + self.__bus = dbus.SystemBus() + self.__timer_re = re.compile('^syncoid-run-.*\.timer$') + #self.__pp = pprint.PrettyPrinter(indent=2) + + def _get_interface(self, dest, object, interface): + try: + obj = self.__bus.get_object(dest, object) + return dbus.Interface(obj, interface) + except dbus.exceptions.DBusException as error: + print(error) + sys.exit(1) + + def _get_logind_manager_interface(self): + return self._get_interface("org.freedesktop.login1", "/org/freedesktop/login1", "org.freedesktop.login1.Manager") + + def _get_systemd_manager_interface(self): + return self._get_interface("org.freedesktop.systemd1", "/org/freedesktop/systemd1", "org.freedesktop.systemd1.Manager") + + def _get_systemd_timer_properties(self, path): + return self._get_interface("org.freedesktop.systemd1", path, "org.freedesktop.DBus.Properties").GetAll("org.freedesktop.systemd1.Timer") + + def _get_users_logged_in(self): + manager = self._get_logind_manager_interface() + users = [] + for user in manager.ListUsers(): + uid = int(user[0]) + name = str(user[1]) + users.append((name, uid)) + return users + + def _get_syncoid_timers(self): + manager = self._get_systemd_manager_interface() + timers = [] + for unit in manager.ListUnits(): + name = str(unit[0]) + path = str(unit[6]) + if not self.__timer_re.match(name): + continue + timers.append((name, path)) + return timers + + def _get_timer_next_elapse(self, path): + props = self._get_systemd_timer_properties(path) + try: + return datetime.datetime.fromtimestamp(int(props['NextElapseUSecRealtime']) / 1000000) + except ValueError: + return datetime.datetime.now() + + def check(self): + result = True + + users = self._get_users_logged_in() + if(len(users) > 0): + print("%d users logged in: %s" % (len(users), ", ".join([user[0] for user in users]))) + result = False + + timers = self._get_syncoid_timers() + for timer in timers: + next = self._get_timer_next_elapse(timer[1]) + until = next - datetime.datetime.now() + if(until < datetime.timedelta(minutes=10)): + print("Timer %s elapses in less then 10 Minutes -> %s (%s)" % (timer[0], next, until)) + result = False + + return result + + def suspend(self): + manager = self._get_logind_manager_interface() + if not manager.CanSuspend(): + print("suspending is not possible!") + return + + print("suspending system") + manager.Suspend(False) + + +if __name__ == "__main__": + s = AutoSuspender() + if s.check(): + s.suspend() -- cgit v1.2.3 From 131b103fa7c3659f84007bda2e60ada9649c996a Mon Sep 17 00:00:00 2001 From: Christian Pointner Date: Wed, 19 May 2021 23:45:39 +0200 Subject: syncoid/autosuspend: output state of checks in any case --- roles/storage/zfs/syncoid/templates/autosuspend.py.j2 | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) (limited to 'roles/storage/zfs/syncoid/templates/autosuspend.py.j2') diff --git a/roles/storage/zfs/syncoid/templates/autosuspend.py.j2 b/roles/storage/zfs/syncoid/templates/autosuspend.py.j2 index e54ddd1b..fdeada52 100644 --- a/roles/storage/zfs/syncoid/templates/autosuspend.py.j2 +++ b/roles/storage/zfs/syncoid/templates/autosuspend.py.j2 @@ -4,7 +4,7 @@ import sys import dbus import re import datetime -#import pprint +# import pprint class AutoSuspender(object): @@ -12,7 +12,7 @@ class AutoSuspender(object): def __init__(self): self.__bus = dbus.SystemBus() self.__timer_re = re.compile('^syncoid-run-.*\.timer$') - #self.__pp = pprint.PrettyPrinter(indent=2) + # self.__pp = pprint.PrettyPrinter(indent=2) def _get_interface(self, dest, object, interface): try: @@ -65,14 +65,18 @@ class AutoSuspender(object): if(len(users) > 0): print("%d users logged in: %s" % (len(users), ", ".join([user[0] for user in users]))) result = False + else: + print("no users logged in.") timers = self._get_syncoid_timers() for timer in timers: next = self._get_timer_next_elapse(timer[1]) until = next - datetime.datetime.now() if(until < datetime.timedelta(minutes=10)): - print("Timer %s elapses in less then 10 Minutes -> %s (%s)" % (timer[0], next, until)) + print("timer '%s' elapses in less then 10 Minutes -> %s (%s)" % (timer[0], next, until)) result = False + else: + print("timer '%s' elapses in %s (%s)" % (timer[0], until, next)) return result @@ -90,3 +94,5 @@ if __name__ == "__main__": s = AutoSuspender() if s.check(): s.suspend() + else: + print("not suspending system") -- cgit v1.2.3 From 7aa53137624142e666dd444c58994f1a2d387da9 Mon Sep 17 00:00:00 2001 From: Christian Pointner Date: Wed, 19 May 2021 23:57:12 +0200 Subject: syncoid/autosuspend: nicer output of autosuspender --- roles/storage/zfs/syncoid/templates/autosuspend.py.j2 | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) (limited to 'roles/storage/zfs/syncoid/templates/autosuspend.py.j2') diff --git a/roles/storage/zfs/syncoid/templates/autosuspend.py.j2 b/roles/storage/zfs/syncoid/templates/autosuspend.py.j2 index fdeada52..3dd893b2 100644 --- a/roles/storage/zfs/syncoid/templates/autosuspend.py.j2 +++ b/roles/storage/zfs/syncoid/templates/autosuspend.py.j2 @@ -11,7 +11,7 @@ class AutoSuspender(object): def __init__(self): self.__bus = dbus.SystemBus() - self.__timer_re = re.compile('^syncoid-run-.*\.timer$') + self.__timer_re = re.compile('^syncoid-pull-.*\.timer$') # self.__pp = pprint.PrettyPrinter(indent=2) def _get_interface(self, dest, object, interface): @@ -19,7 +19,7 @@ class AutoSuspender(object): obj = self.__bus.get_object(dest, object) return dbus.Interface(obj, interface) except dbus.exceptions.DBusException as error: - print(error) + print("ERR: %s" % error) sys.exit(1) def _get_logind_manager_interface(self): @@ -63,20 +63,20 @@ class AutoSuspender(object): users = self._get_users_logged_in() if(len(users) > 0): - print("%d users logged in: %s" % (len(users), ", ".join([user[0] for user in users]))) + print(" [NO] %d users logged in: %s" % (len(users), ", ".join([user[0] for user in users]))) result = False else: - print("no users logged in.") + print(" [ok] no users logged in.") timers = self._get_syncoid_timers() for timer in timers: next = self._get_timer_next_elapse(timer[1]) until = next - datetime.datetime.now() if(until < datetime.timedelta(minutes=10)): - print("timer '%s' elapses in less then 10 Minutes -> %s (%s)" % (timer[0], next, until)) + print(" [NO] timer '%s' elapses in less then 10 Minutes -> %s (%s)" % (timer[0], next, until)) result = False else: - print("timer '%s' elapses in %s (%s)" % (timer[0], until, next)) + print(" [ok] timer '%s' elapses in %s (%s)" % (timer[0], until, next)) return result @@ -85,14 +85,14 @@ class AutoSuspender(object): if not manager.CanSuspend(): print("suspending is not possible!") return - - print("suspending system") manager.Suspend(False) if __name__ == "__main__": s = AutoSuspender() + print("checking if system should be suspended:") if s.check(): + print("trying to suspend system") s.suspend() else: - print("not suspending system") + print("not suspending system because at least one check failed.") -- cgit v1.2.3