Mercurial > getan
diff contrib/zeiterfassung @ 8:20414d892f04
Added script to generate lines for zeiterfassung.txt
author | Sascha L. Teichmann <teichmann@intevation.de> |
---|---|
date | Fri, 01 Aug 2008 18:13:31 +0200 |
parents | |
children | 4e8f5545256d |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/contrib/zeiterfassung Fri Aug 01 18:13:31 2008 +0200 @@ -0,0 +1,185 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- +# +# zeiterfassung +# ------------- +# (c) 2008 by Sascha L. Teichmann <sascha.teichmann@intevation.de> +# +# Simple script which generates lines for zeiterfassung.txt files. +# +# This is Free Software licensed under the terms of GPLv3 or later. +# For details see LICENSE coming with the source of 'getan'. +# +import os +import os.path +import sys +import getopt +import re + +from pysqlite2 import dbapi2 as db + +from datetime import date + +DEFAULT_DATABASE = "time.db" + +TYPE_OF_ENTRY = "-" + +WORKPACKAGE = re.compile("^\[(\w*)(\s|\])") + +USAGE = '''usage: %s <options> + with <options> + [--user=|-u <user>] : Name of user, default: $USER + [--database=|-d <database>]: getan database, default: time.db + [--project|-p] : Key of output project, default: all + [--list|-l] : list all projects + [--help|-h] : This text''' + +LIST_PROJECTS = ''' +SELECT key, description, active FROM projects +''' + +PROJECT_ID_BY_KEY = ''' +SELECT id FROM projects where key = :key +''' + +ALL_PROJECT_IDS = ''' +SELECT id, key FROM projects; +''' + +ENTRIES = ''' +SELECT + date(start_time), + sum(strftime('%s', stop_time) - strftime('%s', start_time)), + 'no description' +FROM entries +WHERE + project_id = :project_id AND + (description IS NULL or length(description) = 0) -- trim() function is missing +GROUP BY round(julianday(start_time)) +UNION +SELECT date(start_time), strftime('%s', stop_time) - strftime('%s', start_time), description +FROM entries +WHERE + project_id = :project_id AND + description IS NOT NULL AND length(description) > 0 +''' + +def human_time(s): + s_h = s % 3600 + h = s / 3600 + m = s_h / 60 + if (s_h % 60) >= 30: + m += 1 + if m == 60: + m = 0 + h += 1 + return "%2d:%02d" % (h, m) + +class TermError(Exception): + + def __init__(self, msg): + Exception.__init__(self) + self.msg = msg + + def __str__(self): + return repr(self.msg) + +def tolerantClose(c): + if c: + try: c.close() + except: pass + +def usage(exit_code = 0): + print USAGE % sys.argv[0] + sys.exit(exit_code) + +def main(): + + database = DEFAULT_DATABASE + user = None + list_projects = False + project = None + + opts, args = getopt.getopt( + sys.argv[1:], + 'd:u:p:hl', + ['database=', 'user=', 'project', 'help', 'list']) + + for opt, val in opts: + if opt in ("--database", "-d"): + database = val + elif opt in ("--user", "-u"): + user = val + elif opt in ("--project", "-p"): + project = val + elif opt in ("--help", "-h"): + usage() + elif opt in ("--list", "-l"): + list_projects = True + + if not user: + user = os.getenv("USER") + + if not os.path.isfile(database): + print >> sys.stderr, "'%s' does not exist or is not a file." % database + sys.exit(1) + + con, cur = None, None + try: + try: + con = db.connect(database) + cur = con.cursor() + + if list_projects: + cur.execute(LIST_PROJECTS) + while True: + row = cur.fetchone() + if not row: break + print "%s %s %s" % (row[0], row[2] and "*" or "-", row[1]) + else: + if project: + cur.execute(PROJECT_ID_BY_KEY, { 'key': project }) + row = cur.fetchone() + if row is None: + raise TermError("There is no project with key '%s'" % project) + project_ids = [[row[0], project]] + else: + cur.execute(ALL_PROJECT_IDS); + project_ids = cur.fetchall() + + for project_id, project in project_ids: + print "# project: %s" % project + cur.execute(ENTRIES, {'project_id': project_id}) + total = 0 + while True: + row = cur.fetchone() + if not row: break + d = date(*map(int, row[0].split('-'))) + t = max(60, row[1]) + c = row[2] + total += t + workpackage = "----" + if c: + m = WORKPACKAGE.match(c) + if m: + workpackage = m.group(1) + c = c[m.end():].strip() + print "%s %sh %s %3s [%s] %s" % ( + d.strftime("%d.%m.%Y"), + human_time(t), + TYPE_OF_ENTRY, + user, + workpackage, + c) + print "# total: %sh" % human_time(total) + finally: + tolerantClose(cur) + tolerantClose(con) + + except TermError, e: + print >> sys.stderr, "error: %s" % e.msg + +if __name__ == '__main__': + main() + +# vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8: