#! /usr/bin/python

# $Id: roundup.py,v 1.4 2001-07-19 06:27:07 anthonybaxter Exp $ 

import sys
if int(sys.version[0]) < 2:
    print 'Roundup requires python 2.0 or later.'
    sys.exit(1)

import string, os, getpass
import config, date, roundupdb

def determineLogin(argv):
    n = 2
    name = password = ''
    if sys.argv[2] == '-user':
        l = sys.argv[3].split(':')
        name = l[0]
        if len(l) > 1:
            password = l[1]
        n = 4
    elif os.environ.has_key('ROUNDUP_LOGIN'):
        l = os.environ['ROUNDUP_LOGIN'].split(':')
        name = l[0]
        if len(l) > 1:
            password = l[1]
    while not name:
        name = raw_input('Login name: ')
    while not password:
        password = getpass.getpass('  password: ')
    return n, roundupdb.openDB(config.DATABASE, name, password)

def usage():
    print '''Usage:

 roundup init
   -- initialise the database
 roundup spec classname
   -- show the properties for a classname
 roundup create [-user login] classname propname=value ...
   -- create a new entry of a given class
 roundup list [-list] classname
   -- list the instances of a class
 roundup history [-list] designator
   -- show the history entries of a designator
 roundup get [-list] designator[,designator,...] propname
   -- get the given property of one or more designator(s)
 roundup set [-user login] designator[,designator,...] propname=value ...
   -- set the given property of one or more designator(s)
 roundup find [-list] classname propname=value ...
   -- find the class instances with a given property
 roundup retire designator[,designator,...]
   -- "retire" a designator
 roundup help    
   -- this help
 roundup morehelp
   -- even more detailed help
'''

def moreusage():
    usage()
    print '''
A designator is a classname and a nodeid concatenated, eg. bug1, user10, ...

Property values are represented as strings in command arguments and in the
printed results:
 . Strings are, well, strings.
 . Date values are printed in the full date format in the local time zone, and
   accepted in the full format or any of the partial formats explained below.
 . Link values are printed as node designators. When given as an argument,
   node designators and key strings are both accepted.
 . Multilink values are printed as lists of node designators joined by commas.
   When given as an argument, node designators and key strings are both
   accepted; an empty string, a single node, or a list of nodes joined by
   commas is accepted.

When multiple nodes are specified to the roundup get or roundup set
commands, the specified properties are retrieved or set on all the listed
nodes. 

When multiple results are returned by the roundup get or roundup find
commands, they are printed one per line (default) or joined by commas (with
the -list) option. 

Where the command changes data, a login name/password is required. The
login may be specified as either "name" or "name:password".
 . ROUNDUP_LOGIN environment variable
 . the -user command-line option
If either the name or password is not supplied, they are obtained from the
command-line. 

Date format examples:
  "2000-04-17.03:45" means <Date 2000-04-17.08:45:00>
  "2000-04-17" means <Date 2000-04-17.00:00:00>
  "01-25" means <Date yyyy-01-25.00:00:00>
  "08-13.22:13" means <Date yyyy-08-14.03:13:00>
  "11-07.09:32:43" means <Date yyyy-11-07.14:32:43>
  "14:25" means <Date yyyy-mm-dd.19:25:00>
  "8:47:11" means <Date yyyy-mm-dd.13:47:11>
  "." means "right now"
'''

def main():

    if len(sys.argv) == 1:
        usage()
        return 1

    command = sys.argv[1]
    if command == 'init':
        password = ''
        confirm = 'x'
        while password != confirm:
            password = getpass.getpass('Admin Password:')
            confirm = getpass.getpass('       Confirm:')
        roundupdb.initDB(config.DATABASE, password)
        return 0

    if command == 'get':
        db = roundupdb.openDB(config.DATABASE)
        designators = string.split(sys.argv[2], ',')
        propname = sys.argv[3]
        for designator in designators:
            classname, nodeid = roundupdb.splitDesignator(designator)
            print db.getclass(classname).get(nodeid, propname)

    elif command == 'set':
        n, db = determineLogin(sys.argv)
        designators = string.split(sys.argv[n], ',')
        props = {}
        for prop in sys.argv[n+1:]:
            key, value = prop.split('=')
            props[key] = value
        for designator in designators:
            classname, nodeid = roundupdb.splitDesignator(designator)
            cl = db.getclass(classname)
            properties = cl.getprops()
            for key, value in props.items():
                type =  properties[key]
                if type.isStringType:
                    continue
                elif type.isDateType:
                    props[key] = date.Date(value)
                elif type.isIntervalType:
                    props[key] = date.Interval(value)
                elif type.isLinkType:
                    props[key] = value
                elif type.isMultilinkType:
                    props[key] = value.split(',')
            apply(cl.set, (nodeid, ), props)

    elif command == 'find':
        db = roundupdb.openDB(config.DATABASE)
        classname = sys.argv[2]
        cl = db.getclass(classname)

        # look up the linked-to class and get the nodeid that has the value
        propname, value = sys.argv[3:].split('=')
        propcl = cl[propname].classname
        nodeid = propcl.lookup(value)

        # now do the find
        print cl.find(propname, nodeid)

    elif command == 'spec':
        db = roundupdb.openDB(config.DATABASE)
        classname = sys.argv[2]
        cl = db.getclass(classname)
        for key, value in cl.properties.items():
            print '%s: %s'%(key, value)

    elif command == 'create':
        n, db = determineLogin(sys.argv)
        classname = sys.argv[n]
        cl = db.getclass(classname)
        props = {}
        properties = cl.getprops()
        for prop in sys.argv[n+1:]:
            key, value = prop.split('=')
            type =  properties[key]
            if type.isStringType:
                props[key] = value 
            elif type.isDateType:
                props[key] = date.Date(value)
            elif type.isIntervalType:
                props[key] = date.Interval(value)
            elif type.isLinkType:
                props[key] = value
            elif type.isMultilinkType:
                props[key] = value.split(',')
        print apply(cl.create, (), props)

    elif command == 'list':
        db = roundupdb.openDB(config.DATABASE)
        classname = sys.argv[2]
        cl = db.getclass(classname)
        key = cl.getkey() or cl.properties.keys()[0]
        for nodeid in cl.list():
            value = cl.get(nodeid, key)
            print "%4s: %s"%(nodeid, value)

    elif command == 'history':
        db = roundupdb.openDB(config.DATABASE)
        classname, nodeid = roundupdb.splitDesignator(sys.argv[2])
        print db.getclass(classname).history(nodeid)

    elif command == 'retire':
        n, db = determineLogin(sys.argv)
        designators = string.split(sys.argv[2], ',')
        for designator in designators:
            classname, nodeid = roundupdb.splitDesignator(designator)
            db.getclass(classname).retire(nodeid)

    elif command == 'help':
	usage()
	return 0

    elif command == 'morehelp':
	moreusage()
	return 0

    else:
	print "Unknown command '%s'"%command
        usage()
        return 1

    db.close()
    return 0

if __name__ == '__main__':
    sys.exit(main())

#
# $Log: not supported by cvs2svn $
# Revision 1.3  2001/07/19 06:08:24  anthonybaxter
# fixed typo in usage string because it was bugging me each time I saw it.
#
# Revision 1.2  2001/07/19 05:52:22  anthonybaxter
# Added CVS keywords Id and Log
#