ti.py

#!/usr/bin/env python2
# encoding: utf-8

import os, sys, argparse, re
import datetime, time

import tidb
from tidb.sql import *

import stats

NAME = 'Technologia Incognita'

def pretty_print(m, form, human):
    print m.format(form, human_read=human).encode('utf-8')

parser = argparse.ArgumentParser(description='%s administration tool' % NAME,
        epilog='') # TODO funny epilog


"""

Examples:

List all members that as overdue in paying, in csv:

        python ti.py -f '%j, %p, %N' -s -n % -r 'overdue'

"""

"""
Member:
* Add
* List
* [Future] Set non-active
* [Future] Manage payments?


* Add member
* List members
"""


parser.add_argument('-n', '--nick', type=unicode)
parser.add_argument('-N', '--name', type=unicode)
parser.add_argument('-e', '--email', type=unicode)
parser.add_argument('-A', '--active-only', action='store_true', default=False)
parser.add_argument('-D', '--inactive-only', action='store_true', default=False)
parser.add_argument('--activate', action='store_true', default=False)
parser.add_argument('--deactivate', action='store_true', default=False)
parser.add_argument('-d', '--date', type=unicode,
    help='Either a valid format string or "now"')
parser.add_argument('--dateformat', type=unicode,
    default='%Y-%m-%d')

parser.add_argument('-H', '--human', action='store_true', default=False)
parser.add_argument('-s', '--search', action='store_true', default=False)
parser.add_argument('-f', '--format', type=str, default='%n %j %p',
    help="""Add percentage in front of the type. Allowed types:
id:        i
nick:      n
name:      N
mail:      m
join-date: j
paid:      p
keyid:     k
active:    A
""")
parser.add_argument('-r', '--restrict', default=None,
        help='Possible options: overdue,ontime,all')
parser.add_argument('-a', '--add', action='store_true', default=False)
parser.add_argument('-J', '--JSON', action='store_true', default=False)

parser.add_argument('--payment', action='store_true', default=False,
        help='Enter payment mode')
parser.add_argument('--payment-months', type=int,
        help='Amount of months')
parser.add_argument('--payment-amount', type=float,
        help='Payment Amount')
parser.add_argument('--payment-comment', type=str,
        help='Payment comment')
parser.add_argument('--payment-hash', type=int,
        help='Payment hash')

args = parser.parse_args()

if sum((args.search, args.add, args.deactivate)) > 1:
    print 'Searching, deleting and/or adding at the same time? Sense you no MAKE!'
    parser.print_help()
    sys.exit(1)

if args.JSON: # For LDAP
    import json
    q = Session.query(Member).filter(Member.active == True).all()

    members = []
    for x in q:
        members.append({"nickname":x.nick, "email":x.email})

    print json.dumps(members, indent=4)

activequery = None
if args.active_only:
    activequery = True
if args.inactive_only:
    activequery = False

if args.activate or args.deactivate:

    q = stats.members_query(args.nick, args.name, args.email, activequery)
    for m in q:
        m.active = args.activate and not args.deactivate
        Session.add(m)

    Session.commit()

elif args.search:
    if args.payment:
        q = stats.members_query(args.nick, args.name, args.email, activequery)
        r = q.all()

        for m in r:
            for p in m.payments:
                print m.nick, 'Date:', p.date, 'Amount:', p.amount, 'Months:', p.months

    else:
        q = stats.members_query(args.nick, args.name, args.email, activequery)
        r = q.all()

        for m in r:
            if args.restrict:
                try:
                    cmpfunc = {
                        'overdue': stats.member_overdue,
                        'ontime' : stats.member_ontime,
                        'all'    : None
                    }[args.restrict]
                except KeyError:
                    print 'Invalid restrict:', args.restrict
                    parser.print_help()
                    sys.exit(1)

                if cmpfunc(m):
                    pretty_print(m, args.format, args.human)

            else:
                pretty_print(m, args.format, args.human)

elif args.add:
    if args.payment:
        add_args = ['nick', 'payment_amount', 'payment_comment',
        'payment_months', 'payment_hash', 'date'] # XXX hash optional
        if not all(map(lambda x: getattr(args, x), add_args)):
            print 'Please provide: ' + ', '.join(add_args)
            parser.print_help()
            sys.exit(1)
        # TODO

    else:
        add_args = ['nick', 'name', 'email', 'date']
        if not all(map(lambda x: getattr(args, x), add_args)):
            print 'Please provide: ' + ', '.join(add_args)
            parser.print_help()
            sys.exit(1)

        d = args.date
        if d == 'now':
            d = datetime.date.today()
        else:
            d = datetime.datetime.strptime(args.date, args.dateformat)

        m = Member(nick=args.nick, name=args.name, email=args.email, member_date=d)
        Session.add(m)
        Session.commit()