Phabricator-maintenance-bot/patchforreview_remover.py

import re
import time
from collections import defaultdict

from lib import Client


class Checker():
    def __init__(self, gerrit_bot_phid, project_patch_for_review_phid, client):
        self.gerrit_bot_phid = gerrit_bot_phid
        self.project_patch_for_review_phid = project_patch_for_review_phid
        self.client = client

    def check(self, t_id):
        phid = self.client.lookupPhid(t_id)
        return self.phid_check(phid)

    def phid_check(self, phid):
        gerrit_bot_actions = []
        for transaction in self.client.getTransactions(phid):
            if re.findall(re.escape('https://github.com/') + r'.+?/pull', str(transaction)):
                return False
            if transaction['authorPHID'] == self.gerrit_bot_phid:
                gerrit_bot_actions.append(transaction)
            else:
                if transaction['type'] == 'projects':
                    check = self.project_patch_for_review_phid in str(
                        transaction['fields'])
                    add_check = "'add'" in str(transaction['fields'])
                    if check and add_check:
                        return False

        gerrit_patch_status = defaultdict(list)
        for case in gerrit_bot_actions:
            if case['type'] != 'comment':
                continue

            if len(case['comments']) != 1:
                return False
            raw_comment = case['comments'][0]['content']['raw']
            gerrit_patch_id = re.findall(
                r'https://gerrit(?:-test|)\.wikimedia\.org/r/.*(\d+)(?:$|\]\])', raw_comment)[0]
            merged = re.findall(
                r'Change \d+ (?:\*\*merged\*\*|abandoned) by ',
                raw_comment)

            gerrit_patch_status[gerrit_patch_id].append(not(bool(merged)))

        for patch in gerrit_patch_status:
            if gerrit_patch_status[patch] != [False, True]:
                return False
        return True


client = Client.newFromCreds()

project_patch_for_review_phid = 'PHID-PROJ-onnxucoedheq3jevknyr'
checker = Checker(
    'PHID-USER-idceizaw6elwiwm5xshb',
    project_patch_for_review_phid,
    client)
gen = client.getTasksWithProject(project_patch_for_review_phid)
for phid in gen:
    if checker.phid_check(phid):
        print(client.taskDetails(phid)['id'])
        try:
            client.removeProjectByPhid(project_patch_for_review_phid, phid)
        except BaseException:
            continue
        time.sleep(10)