diff -r 8a99ccb99361 -r 28c80ae695dc cronrec.py --- a/cronrec.py Tue Oct 05 15:38:38 2010 +0200 +++ b/cronrec.py Wed Oct 06 01:18:34 2010 +0200 @@ -16,7 +16,7 @@ WD = "working_path" CONFIG = {WD: str} CONFIG_SEP = "=" -DB_FILE = "cronrec.db" +DB_FILE = "tim.db" config = {} @@ -27,7 +27,7 @@ except ImportError: # Linux (hopefully) HOMEDIR = os.path.expanduser("~") -CONFIG_FILE = "%s/.cronrecrc" % HOMEDIR +CONFIG_FILE = "%s/.timrc" % HOMEDIR DATETIME_FORMAT = "%Y-%m-%d %H:%M:%S" @@ -122,6 +122,10 @@ db.pause(db_file(), begin) db.resume(db_file(), db.find_active_task(db_file(), begin), end, log) +def set_customer(args): + db.update_where(db_file(), "projects", "name", args.project, "customer", + args.customer) + def status(args): task = db.status(db_file()) if task: @@ -137,41 +141,49 @@ subs = parser.add_subparsers() sub_init = subs.add_parser("init") - sub_init.add_argument("working_path", type=str) + sub_init.add_argument("working_path") sub_init.set_defaults(func=init) sub_begin = subs.add_parser("begin") - sub_begin.add_argument("label", type=str) + sub_begin.add_argument("label") sub_begin.set_defaults(func=begin) sub_end = subs.add_parser("end") - sub_end.add_argument("label", type=str, nargs="?", default="all:all") - sub_end.add_argument("-m", type=str) + sub_end.add_argument("label", nargs="?", default="all:all") + sub_end.add_argument("-m") sub_end.set_defaults(func=end) sub_pause = subs.add_parser("pause") sub_pause.set_defaults(func=pause) sub_resume = subs.add_parser("resume") - sub_resume.add_argument("-m", type=str) + sub_resume.add_argument("-m") sub_resume.set_defaults(func=resume) sub_add = subs.add_parser("add") sub_add_subs = sub_add.add_subparsers() sub_add_task = sub_add_subs.add_parser("task") - sub_add_task.add_argument("label", type=str) - sub_add_task.add_argument("begin", type=str) - sub_add_task.add_argument("end", type=str) - sub_add_task.add_argument("-m", type=str) + sub_add_task.add_argument("label") + sub_add_task.add_argument("begin") + sub_add_task.add_argument("end") + sub_add_task.add_argument("-m") sub_add_task.set_defaults(func=add_task) sub_add_break = sub_add_subs.add_parser("break") - sub_add_break.add_argument("begin", type=str) - sub_add_break.add_argument("end", type=str) - sub_add_break.add_argument("-m", type=str) + sub_add_break.add_argument("begin") + sub_add_break.add_argument("end") + sub_add_break.add_argument("-m") sub_add_break.set_defaults(func=add_break) + sub_set = subs.add_parser("set") + sub_set_subs = sub_set.add_subparsers() + + sub_set_customer = sub_set_subs.add_parser("customer") + sub_set_customer.add_argument("project") + sub_set_customer.add_argument("customer") + sub_set_customer.set_defaults(func=set_customer) + sub_status = subs.add_parser("status") sub_status.set_defaults(func=status)