Find plugins in a Wordpress folder

header.py

#!/usr/bin/env python
import re
import glob
import os.path

GARBAGE = re.compile("^(\s|[*#])+")


def headers(lines, keys=["Plugin Name", "Plugin URI", "Version", "Author",
                         "License", "Author URI", "Description", "Network"]):
    state = None
    h = dict()
    for line in lines:
        if state != 'comment':
            if line.strip().startswith('/*'):
                state = 'comment'
                line = line[2:]
        elif state == 'comment':
            if line.strip().endswith('*/'):
                state = None
                line = line[:-2]
            line = GARBAGE.sub("", line).rstrip()
            if line != "":
                kv = line.split(': ', 1)
                if len(kv) == 2:
                    if kv[0] in keys:
                        h[kv[0]] = kv[1]
    if len(h) == 0:
        return None
    return h


def plugins(folder):
    for pattern in ['%s/*.php', '%s/*/*.php']:
        for p in glob.glob(pattern % folder):
            with open(p, 'r') as f:
                h = headers(f)
                if h is not None and 'Plugin Name' in h:
                    yield h


def all_plugins(web_root):
    for name in ['plugins', 'mu-plugins']:
        p = "%s/wp-content/%s" % (web_root, name)
        if os.path.exists(p):
            for plugin in plugins(p):
                yield plugin

if __name__ == '__main__':
    import sys

    for plugin in all_plugins(sys.argv[1]):
        print plugin