arrdem/source
1
0
Fork 0
Code Issues 11 Pull requests Projects Releases Packages Wiki Activity

More tuning of the unnotifier

Browse source
This commit is contained in:
Reid 'arrdem' McKenzie 2023-10-02 22:51:38 -06:00
parent a26d1d8b40
commit 8db16e8cb1
1 changed files with 78 additions and 50 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

68
projects/gh-unnotifier/src/python/ghunnotif/__main__.py
View file

@ -6,7 +6,7 @@ import tomllib
from typing import Optional
from datetime import datetime, timedelta, timezone
from time import sleep
from pprint import pformat
from pprint import pprint
import click
import requests
@ -149,8 +149,6 @@ def parse_seconds(text: str) -> timedelta:
"--config",
"config_path",
type=Path,
default=lambda: Path(os.getenv("BUILD_WORKSPACE_DIRECTORY", ""))
/ "projects/gh-unnotifier/config.toml",
)
@click.option("--schedule", "schedule", default="30 seconds", type=parse_seconds)
def maintain(config_path: Path, schedule: timedelta):
@ -162,27 +160,25 @@ def maintain(config_path: Path, schedule: timedelta):
team_shitlist = config["gh-unnotifier"].get("team_shitlist", [])
user = client.get_user()
user_teams = {it["slug"] for it in client.get_user_teams()}
mark = None
mark = savepoint = prev = None
def _resolve(notif, reason):
client.read(notif)
client.unsubscribe(notif)
click.echo(f"Resolved {notif['id']} {reason} ({notif['subject']})")
while True:
try:
prev = mark
mark = datetime.now(timezone.utc)
tick = mark + schedule
assert tick - schedule == mark
for notif in client.get_all_notifications(since=prev):
subject = notif["subject"]
pr = None
if subject["type"] == "PullRequest":
if notif["reason"] == "review_requested":
def _pr(subject, notif):
pr = client.get_pr(url=subject["url"])
if pr.get("merged", False):
_resolve(notif, "Merged")
return
if pr.get("state") == "closed":
_resolve(notif, "Closed")
return
if notif["reason"] == "review_requested":
reviewers = client.get_pr_reviewers(pr)
if (
any(org in subject["url"] for org in org_shitlist)
@ -197,9 +193,12 @@ def maintain(config_path: Path, schedule: timedelta):
)
):
_resolve(notif, "Reviewed")
continue
return
elif notif["reason"] == "team_mention":
print("Oustanding PR notification")
pprint({"subject": subject, "notif": notif, "pr": pr})
def _mention(subject, notif):
pr = client.get_pr(url=subject["url"])
reviewers = client.get_pr_reviewers(pr)
@ -216,9 +215,9 @@ def maintain(config_path: Path, schedule: timedelta):
)
):
_resolve(notif, "Ignoring team mention")
continue
return
elif subject["type"] == "Issue":
def _issue(subject, notif):
issue = client.get_issue(url=subject["url"])
if issue["state"] == "closed":
comments = client.get_comments(url=issue["comments_url"])
@ -227,6 +226,30 @@ def maintain(config_path: Path, schedule: timedelta):
):
_resolve(notif, "Unengaged issue closed")
while True:
try:
savepoint = prev
prev = mark
mark = datetime.now(timezone.utc)
tick = mark + schedule
assert tick - schedule == mark
for notif in client.get_all_notifications(since=prev):
notif_timestamp = datetime.fromisoformat(notif["updated_at"])
if (mark - notif_timestamp).days > 3:
_resolve(notif, "More than 3 days old")
subject = notif["subject"]
match subject["type"]:
case "PullRequest":
_pr(subject, notif)
case "mention":
_mention(subject, notif)
case "Issue":
_issue(subject, notif)
duration = (tick - datetime.now(timezone.utc)).total_seconds()
if duration > 0:
sleep(duration)
@ -234,6 +257,11 @@ def maintain(config_path: Path, schedule: timedelta):
except KeyboardInterrupt:
break
except requests.exceptions.HTTPError as e:
print("Encoutered exception", e, "backing off")
prev = savepoint
sleep(schedule.total_seconds())
if __name__ == "__main__":
cli()