customer.py

"""
Manage customers (credit-unions).
"""

import shellish
import subprocess
import re


class Customer(shellish.Command):
    """ Manage customers (credit-unions). """

    name = 'customer'

    def setup_args(self, parser):
        self.add_subcommand(Create)


class Create(shellish.Command):
    """ Create a new customer.

    This will create DB tables, records and filesystem assets for a new
    credit-union.

    NOTE: This is a wrapper for the wellknown `cuset.pl` tool.
    """

    name = 'create'
    vendors = (
        'AMI',
        'COMPUSOURCE',
        'CORRELATION',
        'CRUISE',
        'CU-CENTRIC',
        'CUC',
        'CUPRODIGY',
        'CUSA',
        'ESP',
        'FEDCOMP',
        'GALAXY',
        'HARLAND',
        'INFOWARE',
        'SHARETEC',
        'SOS',
        'SYMITAR',
        'ULTRADATA',
        'WEBONLY',
        'XP-SYSTEMS',
    )

    def setup_args(self, parser):
        self.add_argument('ident', help='Credit Union Identifier',
                          type=self.ident)
        self.add_argument('--vendor', choices=self.vendors, help='Core vendor')
        self.add_argument('--batch', action='store_true', help='Use `Batch` '
                          'insteade of `Live` mode.')
        self.add_argument('--nodb', action='store_true', help='No Database')
        self.add_argument('--nofs', action='store_true', help='No Filesystem')

    def ident(self, value):
        if re.match('[a-zA-Z0-9]*', value).group() != value:
            raise ValueError('Invalid characters')
        return value

    def run(self, args):
        cmd = [
            '/usr/bin/env', 'perl', '/cli/cuset.pl',
            '-f',
            '-u', args.ident,
            '-v', args.vendor if args.vendor else '',
            '-t', 'B' if args.batch else 'L',
            '-x', 'Y' if args.nofs else 'N',
            '-w', 'Y' if args.nodb else 'N',
        ]
        raise SystemExit(subprocess.call(cmd))