|
|
#!/usr/bin/env python3
|
|
|
#
|
|
|
# A small script to automatically reject idle Diffs
|
|
|
#
|
|
|
# you need to set the PHABBOT_USER and PHABBOT_TOKEN environment variable for authentication
|
|
|
|
|
|
import datetime
|
|
|
import os
|
|
|
import sys
|
|
|
|
|
|
import phabricator
|
|
|
|
|
|
MESSAGE = """There seems to have been no activities on this Diff for the past 3 Months.
|
|
|
|
|
|
By policy, we are automatically moving it out of the `need-review` state.
|
|
|
|
|
|
Please, move it back to `need-review` without hesitation if this diff should still be discussed.
|
|
|
|
|
|
:baymax:need-review-idle:
|
|
|
"""
|
|
|
|
|
|
|
|
|
PHAB_URL = "https://phab.mercurial-scm.org/api/"
|
|
|
USER = os.environ.get("PHABBOT_USER", "baymax")
|
|
|
TOKEN = os.environ.get("PHABBOT_TOKEN")
|
|
|
|
|
|
|
|
|
NOW = datetime.datetime.now()
|
|
|
|
|
|
# 3 months in seconds
|
|
|
DELAY = 60 * 60 * 24 * 30 * 3
|
|
|
|
|
|
|
|
|
def get_all_diff(phab):
|
|
|
"""Fetch all the diff that the need review"""
|
|
|
return phab.differential.query(
|
|
|
status="status-needs-review",
|
|
|
order="order-modified",
|
|
|
paths=[('HG', None)],
|
|
|
)
|
|
|
|
|
|
|
|
|
def filter_diffs(diffs, older_than):
|
|
|
"""filter diffs to only keep the one unmodified sin <older_than> seconds"""
|
|
|
olds = []
|
|
|
for d in diffs:
|
|
|
modified = int(d['dateModified'])
|
|
|
modified = datetime.datetime.fromtimestamp(modified)
|
|
|
d["idleFor"] = idle_for = NOW - modified
|
|
|
if idle_for.total_seconds() > older_than:
|
|
|
olds.append(d)
|
|
|
return olds
|
|
|
|
|
|
|
|
|
def nudge_diff(phab, diff):
|
|
|
"""Comment on the idle diff and reject it"""
|
|
|
diff_id = int(d['id'])
|
|
|
phab.differential.createcomment(
|
|
|
revision_id=diff_id, message=MESSAGE, action="reject"
|
|
|
)
|
|
|
|
|
|
|
|
|
if not USER:
|
|
|
print(
|
|
|
"not user specified please set PHABBOT_USER and PHABBOT_TOKEN",
|
|
|
file=sys.stderr,
|
|
|
)
|
|
|
elif not TOKEN:
|
|
|
print(
|
|
|
"not api-token specified please set PHABBOT_USER and PHABBOT_TOKEN",
|
|
|
file=sys.stderr,
|
|
|
)
|
|
|
sys.exit(1)
|
|
|
|
|
|
phab = phabricator.Phabricator(USER, host=PHAB_URL, token=TOKEN)
|
|
|
phab.connect()
|
|
|
phab.update_interfaces()
|
|
|
print('Hello "%s".' % phab.user.whoami()['realName'])
|
|
|
|
|
|
diffs = get_all_diff(phab)
|
|
|
print("Found %d Diffs" % len(diffs))
|
|
|
olds = filter_diffs(diffs, DELAY)
|
|
|
print("Found %d old Diffs" % len(olds))
|
|
|
for d in olds:
|
|
|
diff_id = d['id']
|
|
|
status = d['statusName']
|
|
|
modified = int(d['dateModified'])
|
|
|
idle_for = d["idleFor"]
|
|
|
msg = 'nudging D%s in "%s" state for %s'
|
|
|
print(msg % (diff_id, status, idle_for))
|
|
|
# uncomment to actually affect phab
|
|
|
nudge_diff(phab, d)
|
|
|
|
