import json, subprocess, time
rows = [r for r in json.load(open('land_rows.json')) if r['k'] == 'PR' and r['status'] != 'draft']
nums = [r['n'] for r in rows]; out = {}; t0 = time.time()
F = '''number title createdAt updatedAt isDraft mergeable reviewDecision author{login} authorAssociation baseRefName
 files(first:40){nodes{path}}
 commits(last:1){nodes{commit{committedDate statusCheckRollup{state}}}}
 reviews(last:20){nodes{author{login} authorAssociation state submittedAt}}
 reviewRequests(first:5){nodes{requestedReviewer{... on User{login} ... on Team{name}}}}
 comments(last:6){nodes{author{login} authorAssociation createdAt}}
 timelineItems(last:15,itemTypes:[REVIEW_REQUESTED_EVENT,READY_FOR_REVIEW_EVENT]){nodes{__typename ... on ReviewRequestedEvent{createdAt requestedReviewer{... on User{login}}} ... on ReadyForReviewEvent{createdAt}}}'''
for i in range(0, len(nums), 15):
    part = nums[i:i+15]
    q = '{repository(owner:"huggingface",name:"transformers"){' + ' '.join(f'n{n}: pullRequest(number:{n}){{{F}}}' for n in part) + '}}'
    for _ in range(5):
        r = subprocess.run(['gh', 'api', 'graphql', '-f', 'query=' + q], capture_output=True, text=True, timeout=120)
        if r.returncode == 0: break
        time.sleep(10)
    d = json.loads(r.stdout)['data']['repository']
    for n in part: out[n] = d[f'n{n}']
    print(f'[{len(out)}/{len(nums)}] | elapsed {time.time()-t0:.0f}s', flush=True)
json.dump(out, open('rq.json', 'w')); print('done')