scripts: Initial commit.
This commit is contained in:
Brett Smith
parent
aea9fc0536
commit
50fd55482d
3 changed files with 333 additions and 0 deletions
156
scripts/rt-auto-remind
Executable file
156
scripts/rt-auto-remind
Executable file
|
|
@ -0,0 +1,156 @@
|
||||||
|
#!/usr/bin/env python3
|
||||||
|
|
||||||
|
import argparse
|
||||||
|
import datetime
|
||||||
|
import hashlib
|
||||||
|
import json
|
||||||
|
import pathlib
|
||||||
|
import os
|
||||||
|
import sqlite3
|
||||||
|
import subprocess
|
||||||
|
import sys
|
||||||
|
import tempfile
|
||||||
|
|
||||||
|
DATETIME_FMT = '%Y-%m-%d %H:%M:%S'
|
||||||
|
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 LastRunStorage:
|
||||||
|
DB_FILENAME = 'RunData.db'
|
||||||
|
V1_CREATE_SQL = '''
|
||||||
|
CREATE TABLE IF NOT EXISTS LastRunV1(
|
||||||
|
key text PRIMARY KEY,
|
||||||
|
max_date text
|
||||||
|
);
|
||||||
|
'''
|
||||||
|
V1_SELECT = 'SELECT max_date FROM LastRunV1 WHERE key = ?'
|
||||||
|
V1_INSERT = 'INSERT OR REPLACE INTO LastRunV1 VALUES(?, ?);'
|
||||||
|
|
||||||
|
def __init__(self, args):
|
||||||
|
self.db = sqlite3.connect(str(SHARE_DIR / self.DB_FILENAME))
|
||||||
|
self.db.executescript(self.V1_CREATE_SQL)
|
||||||
|
if args.key is None:
|
||||||
|
self.key = ';'.join(
|
||||||
|
'{}={!r}'.format(key, str(getattr(args, key)))
|
||||||
|
for key in ['body_file', 'date_field', 'max_days_diff', 'search']
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
self.key = args.key
|
||||||
|
|
||||||
|
def load(self):
|
||||||
|
cursor = self.db.execute(self.V1_SELECT, (self.key,))
|
||||||
|
result = cursor.fetchone()
|
||||||
|
if result is None:
|
||||||
|
return None
|
||||||
|
else:
|
||||||
|
return datetime.datetime.strptime(result[0], DATETIME_FMT)
|
||||||
|
|
||||||
|
def save(self, end_date):
|
||||||
|
end_date_s = end_date.strftime(DATETIME_FMT)
|
||||||
|
with self.db:
|
||||||
|
self.db.execute(self.V1_INSERT, (self.key, end_date_s))
|
||||||
|
|
||||||
|
def close(self):
|
||||||
|
self.db.close()
|
||||||
|
|
||||||
|
|
||||||
|
def parse_arguments(arglist):
|
||||||
|
parser = argparse.ArgumentParser()
|
||||||
|
|
||||||
|
action = parser.add_mutually_exclusive_group()
|
||||||
|
action.add_argument(
|
||||||
|
'--correspond',
|
||||||
|
dest='action', action='store_const', const='correspond',
|
||||||
|
default='correspond',
|
||||||
|
help="Send correspondence on found tickets (default)",
|
||||||
|
)
|
||||||
|
action.add_argument(
|
||||||
|
'--comment',
|
||||||
|
dest='action', action='store_const', const='comment',
|
||||||
|
help="Comment on found tickets",
|
||||||
|
)
|
||||||
|
|
||||||
|
parser.add_argument(
|
||||||
|
'--key', '-k',
|
||||||
|
help="Use this string to load and save run times in the database"
|
||||||
|
" (default is auto-generated)",
|
||||||
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
'--dry-run', '-n',
|
||||||
|
action='store_true',
|
||||||
|
help="Show the rt-bulk-send command that would be run; don't run it",
|
||||||
|
)
|
||||||
|
|
||||||
|
parser.add_argument(
|
||||||
|
'date_field',
|
||||||
|
help="RT date field to constrain in the search",
|
||||||
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
'min_days_diff',
|
||||||
|
type=int,
|
||||||
|
help="The earliest end of the date constraint, in days from today",
|
||||||
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
'max_days_diff',
|
||||||
|
type=int,
|
||||||
|
help="The latest end of the date constraint, in days from today",
|
||||||
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
'search',
|
||||||
|
help="TicketSQL search, like you would pass to `rt search`",
|
||||||
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
'body_file',
|
||||||
|
type=pathlib.Path,
|
||||||
|
help="Path to file that has the content of your correspondence/comment",
|
||||||
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
'rt_args', metavar='rt arguments',
|
||||||
|
nargs=argparse.REMAINDER,
|
||||||
|
help="Additional arguments to pass to `rt correspond/comment`",
|
||||||
|
)
|
||||||
|
args = parser.parse_args(arglist)
|
||||||
|
args.min_delta = datetime.timedelta(days=args.min_days_diff)
|
||||||
|
args.max_delta = datetime.timedelta(days=args.max_days_diff)
|
||||||
|
return args
|
||||||
|
|
||||||
|
def main(arglist=None, stdout=sys.stdout, stderr=sys.stderr):
|
||||||
|
args = parse_arguments(arglist)
|
||||||
|
last_run_db = LastRunStorage(args)
|
||||||
|
last_run_range_end = last_run_db.load()
|
||||||
|
start_datetime = datetime.datetime.utcnow()
|
||||||
|
date_range_start = start_datetime + args.min_delta
|
||||||
|
if (last_run_range_end is not None) and (last_run_range_end > date_range_start):
|
||||||
|
date_range_start = last_run_range_end
|
||||||
|
date_range_end = start_datetime + args.max_delta
|
||||||
|
search = '({search}) AND {field} > "{start}" AND {field} <= "{end}"'.format(
|
||||||
|
search=args.search,
|
||||||
|
field=args.date_field,
|
||||||
|
start=date_range_start.strftime(DATETIME_FMT),
|
||||||
|
end=date_range_end.strftime(DATETIME_FMT),
|
||||||
|
)
|
||||||
|
send_cmd = [
|
||||||
|
'rt-bulk-send', '--{}'.format(args.action), '--loglevel=debug',
|
||||||
|
search, str(args.body_file), *args.rt_args,
|
||||||
|
]
|
||||||
|
if args.dry_run:
|
||||||
|
import pprint
|
||||||
|
pprint.pprint(send_cmd)
|
||||||
|
returncode = 0
|
||||||
|
else:
|
||||||
|
try:
|
||||||
|
subprocess.run(send_cmd, check=True)
|
||||||
|
except subprocess.CalledProcessError as error:
|
||||||
|
returncode = error.returncode
|
||||||
|
else:
|
||||||
|
last_run_db.save(date_range_end)
|
||||||
|
returncode = 0
|
||||||
|
last_run_db.close()
|
||||||
|
return returncode
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
exit(main())
|
||||||
105
scripts/rt-bulk-send
Executable file
105
scripts/rt-bulk-send
Executable file
|
|
@ -0,0 +1,105 @@
|
||||||
|
#!/usr/bin/env python3
|
||||||
|
|
||||||
|
import argparse
|
||||||
|
import logging
|
||||||
|
import pathlib
|
||||||
|
import subprocess
|
||||||
|
import sys
|
||||||
|
|
||||||
|
logger = logging.getLogger('rt-bulk-send')
|
||||||
|
|
||||||
|
def parse_arguments(arglist):
|
||||||
|
parser = argparse.ArgumentParser()
|
||||||
|
|
||||||
|
action = parser.add_mutually_exclusive_group()
|
||||||
|
action.add_argument(
|
||||||
|
'--correspond',
|
||||||
|
dest='action', action='store_const', const='correspond',
|
||||||
|
default='correspond',
|
||||||
|
help="Send correspondence on found tickets (default)",
|
||||||
|
)
|
||||||
|
action.add_argument(
|
||||||
|
'--comment',
|
||||||
|
dest='action', action='store_const', const='comment',
|
||||||
|
help="Comment on found tickets",
|
||||||
|
)
|
||||||
|
|
||||||
|
parser.add_argument(
|
||||||
|
'--loglevel',
|
||||||
|
choices=['debug', 'info', 'warning', 'error', 'critical'],
|
||||||
|
default='warning',
|
||||||
|
help="Show log messages at this level (default %(default)s)",
|
||||||
|
)
|
||||||
|
|
||||||
|
parser.add_argument(
|
||||||
|
'search',
|
||||||
|
help="TicketSQL search, like you would pass to `rt search`",
|
||||||
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
'body_file',
|
||||||
|
type=pathlib.Path,
|
||||||
|
help="Path to file that has the content of your correspondence/comment",
|
||||||
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
'rt_args', metavar='rt arguments',
|
||||||
|
nargs=argparse.REMAINDER,
|
||||||
|
help="Additional arguments to pass to `rt correspond/comment`",
|
||||||
|
)
|
||||||
|
return parser.parse_args(arglist)
|
||||||
|
|
||||||
|
def setup_logger(logger, loglevel, stream):
|
||||||
|
formatter = logging.Formatter('%(name)s: %(levelname)s: %(message)s')
|
||||||
|
handler = logging.StreamHandler(stream)
|
||||||
|
handler.setFormatter(formatter)
|
||||||
|
logger.addHandler(handler)
|
||||||
|
logger.setLevel(loglevel)
|
||||||
|
|
||||||
|
def act_on_ticket(ticket_id, args, body_file):
|
||||||
|
body_file.seek(0)
|
||||||
|
return subprocess.run(
|
||||||
|
['rt', args.action, *args.rt_args, '-m', '-', ticket_id],
|
||||||
|
stdin=body_file,
|
||||||
|
check=True,
|
||||||
|
)
|
||||||
|
|
||||||
|
def main(arglist=None, stdout=sys.stdout, stderr=sys.stderr):
|
||||||
|
args = parse_arguments(arglist)
|
||||||
|
setup_logger(logger, getattr(logging, args.loglevel.upper()), stderr)
|
||||||
|
try:
|
||||||
|
body_file = args.body_file.open()
|
||||||
|
except OSError as error:
|
||||||
|
logger.critical("error opening {}: {}".format(args.body_file, error))
|
||||||
|
return 3
|
||||||
|
if not body_file.seekable():
|
||||||
|
logger.critical("file {} must be seekable".format(args.body_file))
|
||||||
|
with body_file:
|
||||||
|
return 3
|
||||||
|
|
||||||
|
failures = 0
|
||||||
|
with body_file, subprocess.Popen(
|
||||||
|
['rt', 'search', '-i', args.search],
|
||||||
|
stdout=subprocess.PIPE,
|
||||||
|
universal_newlines=True,
|
||||||
|
) as search_pipe:
|
||||||
|
for line in search_pipe.stdout:
|
||||||
|
ticket_id = line.rstrip('\n')
|
||||||
|
if not ticket_id:
|
||||||
|
continue
|
||||||
|
logger.info("Acting on %s", ticket_id)
|
||||||
|
try:
|
||||||
|
act_on_ticket(ticket_id, args, body_file)
|
||||||
|
except subprocess.CalledProcessError as error:
|
||||||
|
logger.error("Failed to %s on %s: rt returned exit code %s",
|
||||||
|
args.action, ticket_id, error.returncode)
|
||||||
|
failures += 1
|
||||||
|
|
||||||
|
if search_pipe.returncode != 0:
|
||||||
|
logger.critical("`rt search` returned exit code %s", search_pipe.returncode)
|
||||||
|
return 4
|
||||||
|
elif failures == 0:
|
||||||
|
return 0
|
||||||
|
else:
|
||||||
|
return min(10 + failures, 99)
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
exit(main())
|
||||||
72
scripts/rt-send-all-reminders
Executable file
72
scripts/rt-send-all-reminders
Executable file
|
|
@ -0,0 +1,72 @@
|
||||||
|
#!/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,
|
||||||
|
date_field='Due', body_file=None, action='correspond'):
|
||||||
|
if body_file is None:
|
||||||
|
body_file = pathlib.Path(SHARE_DIR, 'templates', key + '.txt')
|
||||||
|
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(
|
||||||
|
'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 in yaml_data:
|
||||||
|
reminder = Reminder(key, **yaml_data[key])
|
||||||
|
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())
|
||||||
Loading…
Reference in a new issue