RT-Client-Tools/scripts/rt-send-all-reminders

#!/usr/bin/env python3

import argparse
import os
import pathlib
import subprocess
import sys

import yaml

try:
    _data_home = os.environ['XDG_DATA_HOME']
except KeyError:
    _data_home = pathlib.Path(os.path.expanduser('~'), '.local', 'share')
SHARE_DIR = pathlib.Path(_data_home, 'rt-auto-remind')
SHARE_DIR.mkdir(parents=True, exist_ok=True)

class Reminder:
    def __init__(self, key, min_days, max_days, search, body_file,
                 date_field='Due', action='correspond'):
        self.key = key
        self.min_days_diff = int(min_days)
        self.max_days_diff = int(max_days)
        self.search = search
        self.date_field = date_field
        self.body_file = body_file
        self.action = action

    def remind_cmd(self):
        return [
            'rt-auto-remind', '--{}'.format(self.action),
            '--key', self.key,
            self.date_field,
            str(self.min_days_diff),
            str(self.max_days_diff),
            self.search,
            str(self.body_file),
        ]


def parse_arguments(arglist):
    parser = argparse.ArgumentParser()
    parser.add_argument(
        '--template-dir',
        type=pathlib.Path,
        default=SHARE_DIR / 'templates',
        help="Directory to search for templates that aren't specified in the"
        " YAML file (default `%(default)s`)",
    )
    parser.add_argument(
        'yaml_files', metavar='PATH',
        type=pathlib.Path,
        nargs='+',
        help="YAML file(s) with configuration of reminders to send out",
    )
    return parser.parse_args(arglist)

def main(arglist=None, stdout=sys.stdout, stderr=sys.stderr):
    args = parse_arguments(arglist)
    failures = 0
    for yaml_path in args.yaml_files:
        with yaml_path.open() as yaml_file:
            yaml_data = yaml.safe_load(yaml_file)
        for key, reminder_kwargs in yaml_data.items():
            if 'body_file' not in reminder_kwargs:
                reminder_kwargs['body_file'] = args.template_dir / (key + '.txt')
            reminder = Reminder(key, **reminder_kwargs)
            try:
                subprocess.run(reminder.remind_cmd(), check=True)
            except subprocess.CalledProcessError as error:
                print("warning: reminder {} exited {}".format(key, error.returncode))
                failures += 1
    if failures == 0:
        return 0
    else:
        return min(10 + failures, 99)

if __name__ == '__main__':
    exit(main())