LibraryMigration/ghutil.py at master · Hellozaq/LibraryMigration · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
import os
import re
import logging
from typing import List, Tuple


def get_tokens(token_file: str) -> List[str]:
    """
    Get a list of GitHub tokens from the supplied file. Assumes one token each line, and ignores blank line.
    :param token_file: path to the token file
    :return: A list of GitHub tokens
    """
    if not os.path.exists(token_file):
        logging.error(
            "Please put GitHub Tokens in {} for this script to work".format(token_file))
        return []
    with open(token_file, "r") as f:
        return list(x.strip() for x in f.readlines() if x.strip() != "")


def get_commit_link(repo_name: str, commit_sha: str) -> str:
    """
    Build a commit URL for manual browser access using full repository name and commit SHA1
    :param repo_name: full repository name (i.e. `{username}/{repoanme}`)
    :param commit_sha: 40 byte SHA1 for a commit
    :return: A commit URL
    """
    return "https://github.com/{}/commit/{}".format(repo_name, commit_sha)


def get_issue_link(repo_name: str, issue_number: int) -> str:
    """
    Build an issue URL for manual browser access using full repository name and issue number
    :param repo_name: full repository name (i.e. `{username}/{repoanme}`)
    :param issue_number: issue number used on GitHub
    :return: An issue URL
    """
    return "https://github.com/{}/issues/{}".format(repo_name, issue_number)


def get_pr_link(repo_name: str, pr_number: int) -> str:
    """
    Build a pull request URL for manual browser access using full repository name and PR number
    :param repo_name: full repository name (i.e. `{username}/{repoanme}`)
    :param pr_number: PR number used on GitHub
    :return: A pull request URL
    """
    return "https://github.com/{}/pull/{}".format(repo_name, pr_number)


def get_issues_in_text(repo_name: str, text: str) -> List[Tuple[str, int]]:
    """
    See https://docs.github.com/en/free-pro-team@latest/github/writing-on-github/
            autolinked-references-and-urls#issues-and-pull-requests
    :param repo_name: full repo name like "username/repository_name"
    :param text: the text to extract linked issues (e.g. commits)
    :return: Tuples of (repo_name, issue_number)
    """
    pattern = r"([\w\.\-_]+/[\w\.\-_]+)#(\d+)|#(\d+)|GH-(\d+)"
    result = []
    for match in re.finditer(pattern, text):
        # print(match.string)
        # print(text[match.start():match.end()])
        # print(match.groups())
        groups = match.groups()
        if groups[0] is not None and groups[1] is not None:
            result.append((groups[0], int(groups[1])))
        elif groups[2] is not None:
            result.append((repo_name, int(groups[2])))
        elif groups[3] is not None:
            result.append((repo_name, int(groups[3])))
    url_pattern = r"https://github\.com/([\w\.\-_]+/[\w\.\-_]+)/(issues|pull)/(\d+)"
    for match in re.finditer(url_pattern, text):
        result.append((match.groups()[0], int(match.groups()[2])))
    return result


if __name__ == "__main__":
    logging.basicConfig(level=logging.DEBUG)
    res = get_issues_in_text("a/a", "cc#23 is abc GH-32 where abc/def#222 and a-c/D.f#333" +
                             " https://github.com/abc/def/issues/4444 sadwec " +
                             "https://github.com/abc./def-_/pull/5555")
    print(res)