phi/src/phicli

#!/usr/bin/env python3
from pprint import pformat as pp
import sys
import argparse

import phi.ldap.client
from phi.config import get_config
from phi.logging import setup_logging, get_logger

from phi.ldap.user import get_user_by_uid

log = get_logger(__name__)


if __name__ == '__main__':
    parser = argparse.ArgumentParser()

    parser.add_argument(
        '--config', metavar='config.yml', help='custom configuration file')

    subparses = parser.add_subparsers(title='actions', dest='action')

    parser_showuser = subparses.add_parser(
        'showuser', help='display user fields')
    parser_showuser.add_argument(dest='uid', help='user identifier')

    namespace = parser.parse_args(sys.argv[1:])
    config_file = namespace.config

    config_file, config = get_config(config_file)
    setup_logging(config.get('logging', {}))
    log.info("Using configuration at '{}':\n{}"
             .format(config_file, pp(config)))

    # TODO: check fields in config
    client = phi.ldap.client.Client(**config['ldap'])

    log.info('Opening LDAP client')
    client.open()

    log.info('Arguments: {}'.format(pp(namespace.__dict__)))

    if namespace.action == 'showuser':
        uid = namespace.uid
        user = get_user_by_uid(client, uid)
        print('\nuid: {}\n{}\n\n'.format(uid, pp(user)))

    log.info('Closing LDAP client')
    client.close()