#!/usr/local/bin/python

 """

 A time recording tool.

 Author: Eugen Sawin (sawine@me73.com)

 Dependencies: libsqlite3-dev

 """

 import os

 import sys

 import argparse

 from datetime import datetime

 import db

 WD = "working_path"

 CONFIG = {WD: str}

 CONFIG_SEP = "="

 DB_FILE = "cronrec.db"

 config = {}

 # Trying to get the user's home directory path

 try: # Windows

 	from win32com.shell import shellcon, shell

 	HOMEDIR	= shell.SHGetFolderPath(0, shellcon.CSIDL_LOCAL_APPDATA, 0, 0)

 except ImportError: # Linux (hopefully)

 	HOMEDIR = os.path.expanduser("~")

 CONFIG_FILE = "%s/.cronrecrc" % HOMEDIR

 def read_config():

 	config = {}

 	with open(CONFIG_FILE, "r") as config_stream:

 		config_lines = [l.split(CONFIG_SEP) for l in config_stream.readlines() 

 						if CONFIG_SEP in l]

 		for key, value in config_lines:

 			key = key.strip().lower()

 			value = value.strip()

 			if key in CONFIG:

 				config[key] = CONFIG[key](value)

 	return config

 def write_config(config):

 	with open(CONFIG_FILE, "w") as config_input:

 		config_input.write("\n".join([CONFIG_SEP.join((k, v)) 

 							for k, v in config.iteritems() if k in CONFIG]))

 def db_file():

 	global config

 	if WD not in config:

 		print "Working directory path is not configured. \

 Please use the init command."

 		return None

 	return config[WD] + "/" + DB_FILE

 def init(args):

 	global config

 	last_wd = None

 	if WD in config:

 		last_wd = config[WD]

 	config[WD] = args.working_path

 	write_config(config)

 	path_exists = os.path.exists(config[WD]) and os.path.isdir(config[WD])

 	db_exists = path_exists and os.path.exists(db_file())\

 				and os.path.isfile(db_file())

 	if last_wd != config[WD]:		

 		print "Changed working directory from %s to %s." % (last_wd, config[WD])

 	else:

 		print "Working directory remains at %s." % config[WD]

 	if not path_exists:

 		print "Warning: working directory %s does not exist." % config[WD]

 		os.makedirs(config[WD])

 		print "Working directory %s created." % config[WD]

 	elif db_exists:

 		print "Database file %s already exists, please delete it before \

 initiating a new database at this location." % db_file()

 	if not db_exists:

 		db.init(db_file())	

 def begin(args):

 	project = None

 	activity = None

 	label = args.label.strip()

 	if ":" in label:

 		project, activity = label.split(":")

 	else:

 		project = label

 	db.begin(db_file(), project, activity, datetime.now())	

 	print "begins %s:%s" % (project, activity)

 def end(args):

 	project = None

 	activity = None

 	label = args.label.strip()

 	if ":" in label:

 		project, activity = label.split(":")

 	else:

 		project = label

 	db.end(db_file(), project, activity, datetime.now())	

 	print "ends %s:%s" % (project, activity)

 def main():

 	global config

 	config = read_config()

 	parser = argparse.ArgumentParser(description="Records time.")

 	subs = parser.add_subparsers()

 	sub_begin = subs.add_parser("init")

 	sub_begin.add_argument("working_path", type=str)

 	sub_begin.set_defaults(func=init)

 	sub_begin = subs.add_parser("begin")

 	sub_begin.add_argument("label", type=str)

 	sub_begin.set_defaults(func=begin)

 	sub_end = subs.add_parser("end")

 	sub_end.add_argument("label", type=str)

 	sub_end.set_defaults(func=end)

 	args = parser.parse_args()

 	args.func(args)

 if __name__ == "__main__":

 	main()