scikit-image

1
#!/usr/bin/env python
2
# Requires package 'editdistance'
3

4
# A mailmap file is used (by GitHub and other tools) to associate multiple
5
# commit emails with one user.  This helps to count number of commits,
6
# contributors, etc.
7

8
import subprocess
9
import shlex
10
import numpy as np
11
from collections import defaultdict
12

13
from editdistance import eval as dist
14

15
threshold = 5
16

17

18
def call(cmd):
19
    return subprocess.check_output(shlex.split(cmd), text=True).split('\n')
20

21

22
def _clean_email(email):
23
    if '@' not in email:
24
        return
25

26
    name, domain = email.split('@')
27
    name = name.split('+', 1)[0]
28

29
    return f'{name}@{domain}'.lower()
30

31

32
call("rm -f .mailmap")
33
authors = call("git log --format='%aN::%aE'")
34

35
names, emails = [], []
36

37
for name, email in (author.split('::') for author in authors if author.strip()):
38
    if email not in emails:
39
        names.append(name)
40
        emails.append(email)
41

42
N = len(names)
43
D = np.zeros((N, N)) + np.inf
44

45
for i in range(1, N):
46
    for j in range(i):
47
        D[i, j] = dist(names[i], names[j])
48

49
for i in range(N):
50
    (dupes,) = np.where(D[:, i] < threshold)
51
    for j in dupes:
52
        names[j] = names[i]
53

54
mailmap = defaultdict(set)
55
for name, email in zip(names, emails):
56
    email = _clean_email(email)
57
    if email:
58
        mailmap[name].add(email)
59

60
for key, value in list(mailmap.items()):
61
    if len(value) < 2 or (len(key.split()) < 2):
62
        mailmap.pop(key)
63

64
entries = []
65
for name, emails in mailmap.items():
66
    entries.append([name])
67
    entries[-1].extend([f'<{email}>' for email in emails])
68

69
entries = sorted(entries, key=lambda x: x[0].split()[-1])
70
for entry in entries:
71
    print(' '.join(entry))
72