gitosis/group.py


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

import logging
from ConfigParser import NoSectionError, NoOptionError
 
def _getMembership(config, user, seen):
    log = logging.getLogger('gitosis.group.getMembership')
 
    for section in config.sections():
        GROUP_PREFIX = 'group '
        if not section.startswith(GROUP_PREFIX):
            continue
        group = section[len(GROUP_PREFIX):]
        if group in seen:
            continue
 
        try:
            members = config.get(section, 'members')
        except (NoSectionError, NoOptionError):
            members = []
        else:
            members = members.split()
 
        if user in members:
            log.debug('found %(user)r in %(group)r' % dict(
                user=user,
                group=group,
                ))
            seen.add(group)
            yield group
 
            for member_of in _getMembership(
                config, '@%s' % group, seen,
                ):
                yield member_of
 
 
def getMembership(config, user):
    """
    Generate groups ``user`` is member of, according to ``config``
 
    :type config: RawConfigParser
    :type user: str
    :param _seen: internal use only
    """
 
    seen = set()
    for member_of in _getMembership(config, user, seen):
        yield member_of
 
    # everyone is always a member of group "all" 
    yield 'all'