A quick script to plot which Haskell libs I import

who-imports.py

#! /usr/bin/env python3

import os
import os.path as path
import re
import collections
from ascii_graph import Pyasciigraph
import sys

imports_re = re.compile('^import (qualified )?([a-z\.A-Z0-9]+)')

def flatten(iterable):
    for i in iterable:
        yield from i

def get_files(root):
    for dirpath, dirnames, filenames in os.walk(root):
        for f in filenames:
            filepath = path.join(dirpath, f)
            if path.splitext(filepath)[1] == '.hs':
                yield filepath

def grab_imports(name):
    with open(name) as f:
        for line in f:
            match = re.match(imports_re, line)
            if match:
                yield match.groups()[1]

def join_same_values(d):
    def trim_key(k):
        k.sort()
        pretty = " / ".join(k)
        return pretty[0:80] + ('...' if len(pretty) > 77 else '')

    out = collections.defaultdict(list)
    for k, v in d.items():
        out[v].append(k)

    return [(trim_key(k), v) for v, k in out.items()]

if __name__ == '__main__':
    imports = flatten(map(grab_imports, get_files(sys.argv[1])))
    counts = join_same_values(collections.Counter(imports))
    counts.sort(key=lambda x: -x[1])

    for line in Pyasciigraph().graph('Imports counts', counts):
        print(line)