From b4a5dad35bfd0216ccc8bfd81ffee813d1876340 Mon Sep 17 00:00:00 2001 From: Josh Lay Date: Sat, 14 Oct 2023 21:45:56 -0500 Subject: [PATCH] init --- README.md | 26 +++++++ startup.py | 202 +++++++++++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 228 insertions(+) create mode 100644 README.md create mode 100755 startup.py diff --git a/README.md b/README.md new file mode 100644 index 0000000..52957c9 --- /dev/null +++ b/README.md @@ -0,0 +1,26 @@ +# autostart-i3ipc + +```bash +~ $ .config/sway/scripts/startup.py -h +usage: startup.py [-h] [-c CONFIG] + +A smart but also lazy login autostart manager for i3/Sway. + +Will conditionally exec other things defined in a YML dict. ie: every day, work days, or weekends + +Required i3/Sway config line: + exec .config/sway/scripts/startup.py + +Config sample: +--- +autostarts: + pre: [] # blocking tasks that run every day, before any other section. intended for backups/updates + common: [] # non-blocking tasks that run every day + weekend: [] # blocking tasks for weekends, after 'pre' but before 'common' + work: [] # non-blocking tasks run if Monday through Friday between 8AM - 4PM + +options: + -h, --help show this help message and exit + -c CONFIG, --config CONFIG + Path to the YML configuration file. (default: /home/user/.config/sway/autostart.yml) +``` diff --git a/startup.py b/startup.py new file mode 100755 index 0000000..827dd21 --- /dev/null +++ b/startup.py @@ -0,0 +1,202 @@ +#!/usr/bin/env python3 +""" +A smart but also lazy login autostart manager for i3/Sway. + +Will conditionally exec other things defined in a YML dict. ie: every day, work days, or weekends + +Required i3/Sway config line: + exec /home/jlay/.config/sway/scripts/startup.py + +Config sample: +--- +autostarts: + pre: [] # blocking tasks that run every day, before any other section. intended for backups/updates + common: [] # non-blocking tasks that run every day + weekend: [] # blocking tasks for weekends, after 'pre' but before 'common' + work: [] # non-blocking tasks run if Monday through Friday between 8AM - 4PM + +Dependencies: python3-i3ipc +""" +import os +import subprocess +from datetime import datetime +from time import sleep +import argparse +from textwrap import dedent +from systemd import journal +import yaml.loader +from i3ipc import Connection +from xdg import XDG_CONFIG_HOME # pylint: disable=no-name-in-module + + +def log_message( + message: str, level: str, syslog_identifier: str = "sway-autostart" +) -> None: + """Given `message`, send it to the journal and print + + Accepts 'journal' levels. ie: `journal.LOG_{ERR,INFO,CRIT,EMERG}' + """ + valid_levels = { + journal.LOG_EMERG, + journal.LOG_ALERT, + journal.LOG_CRIT, + journal.LOG_ERR, + journal.LOG_WARNING, + journal.LOG_NOTICE, + journal.LOG_INFO, + journal.LOG_DEBUG, + } + if level not in valid_levels: + raise ValueError(f"Invalid log level: {level}") + print(message) + journal.send(message, PRIORITY=level, SYSLOG_IDENTIFIER=syslog_identifier) + + +def parse_args(): + """If run interactively, this provides arg function to the user""" + description_text = dedent( + f"""\ + A smart but also lazy login autostart manager for i3/Sway. + + Will conditionally exec other things defined in a YML dict. ie: every day, work days, or weekends + + Required i3/Sway config line: + exec {os.path.abspath(__file__)} + + Config sample: + --- + autostarts: + pre: [] # blocking tasks that run every day, before any other section. intended for backups/updates + common: [] # non-blocking tasks that run every day + weekend: [] # blocking tasks for weekends, after 'pre' but before 'common' + work: [] # non-blocking tasks run if Monday through Friday between 8AM - 4PM + """ + ) + + class PlainDefaultFormatter( + argparse.ArgumentDefaultsHelpFormatter, argparse.RawDescriptionHelpFormatter + ): + """Combines two ArgParse formatter classes: + - argparse.ArgumentDefaultsHelpFormatter + - argparse.RawDescriptionHelpFormatter""" + + parser = argparse.ArgumentParser( + description=description_text, formatter_class=PlainDefaultFormatter + ) + + # Default path for the config + default_config = os.path.join(XDG_CONFIG_HOME, "sway", "autostart.yml") + + parser.add_argument( + "-c", + "--config", + default=default_config, + help="Path to the YML configuration file.", + ) + + return parser.parse_args() + + +# OOPy way to determine if it's a work day -- mon<->friday, 3AM<->5PM +class WorkTime(datetime): + """datetime but with work on the mind""" + + def is_workday(self): + """determine if it's a work day: monday-friday between 3AM and 5PM. + Use .vacation file to go on vacation""" + + # first check if ~/.vacation exists - if so, not a work day + if os.path.isfile(os.path.expanduser("~/.vacation")): + return False + + # note: last number in range isn't included + if not self.is_weekend() and self.hour in range(8, 16): + return True + return False + + def is_weekend(self): + """determine if it's the weekend or not, ISO week day outside 1-5""" + if self.isoweekday() not in range(1, 6): + return True + return False + + +if __name__ == "__main__": + args = parse_args() + config_path = args.config + + # get the current time + now = WorkTime.now() + # determine if it's a work day using WorkTime above + workday = now.is_workday() + weekend = now.is_weekend() + + # initialize empty lists for the different categories + wants = [] # non-blocking tasks from 'common' and 'workday' sections in config + pre_list = [] # blocking tasks before the rest + weekend_list = [] # non-blocking tasks for weekend days/logins + + # check the config file for existence/structure. if found, extend the lists + if os.path.exists(config_path): + print(f"found/loading config: '{config_path}'") + with open(config_path, "r", encoding="utf-8") as _config: + config_file = yaml.load(_config, Loader=yaml.FullLoader) + try: + loaded_starts = config_file["autostarts"] + wants.extend(loaded_starts["common"]) + if loaded_starts["pre"]: + pre_list.extend(loaded_starts["pre"]) + if workday: + wants.extend(loaded_starts["work"]) + if weekend: + weekend_list.extend(loaded_starts["weekend"]) + except KeyError as key_err: + log_message( + f"Key not found in {config_path}: {key_err.args[0]}", + journal.LOG_ERR, + ) + except NameError as name_err: + log_message(f"name error: {name_err}", journal.LOG_ERR) + + # get the party started, create a connection to the window manager + _wm = Connection(auto_reconnect=True) + + # start the blocking tasks - 'pre' and 'weekend' + # avoid sending them to the WM, would become backgrounded/async + for pre_item in pre_list: + try: + log_message( + f'running (blocking) "pre" task: "{pre_item}"', journal.LOG_INFO + ) + subprocess.run(pre_item, shell=True, check=False) + except subprocess.CalledProcessError as pre_ex: + log_message(f'failed "{pre_item}": {pre_ex.output}', journal.LOG_ERR) + + if weekend: + for weekend_item in weekend_list: + try: + log_message( + f'running (blocking) "weekend" task: "{weekend_item}"', + journal.LOG_INFO, + ) + subprocess.check_output(weekend_item, shell=True) + except subprocess.CalledProcessError as weekend_except: + log_message( + f'Exception during "{weekend_item}": {weekend_except.output}', + journal.LOG_ERR, + ) + + # launch 'common' and 'work' tasks; not expected to block, sent to window manager + for wanteditem in wants: + command = "exec " + wanteditem + log_message(f'sending to WM: "{command}"', journal.LOG_INFO) + reply = _wm.command(command) + sleep(0.1) + if reply[0].error: + # note: this doesn't check return codes + # serves to check if there was a parsing/comm. error with the WM + log_message( + f'autostart "{command}" failed, couldn\'t reach WM', journal.LOG_ERR + ) + + _wm.main_quit()