getan/getan: getan/backend.py annotate

annotate getan/backend.py @ 362:90c09cca49c3

Fix loading entries of a project Projects.entries is converted into a read only property. Therefore it is not allowed to set the entries directly. Instead the load_entries method must be used.

author	Björn Ricks <bjoern.ricks@intevation.de>
date	Mon, 03 Mar 2014 15:08:07 +0100
parents	5ded6192b85b
children	a487535e7844

rev	line source
23 9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	1 # -- coding: utf-8 --
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	2 #
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	3 # (c) 2008, 2009, 2010 by
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	4 # Sascha L. Teichmann <sascha.teichmann@intevation.de>
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	5 # Ingo Weinzierl <ingo.weinzierl@intevation.de>
97 99639833968d Add me as author in files that I did touch Björn Ricks <bjoern.ricks@intevation.de> parents: 68 diff changeset	6 # (c) 2011 Björn Ricks <bjoern.ricks@intevation.de>
23 9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	7 #
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	8 # This is Free Software licensed unter the terms of GPLv3 or later.
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	9 # For details see LICENSE coming with the source of 'getan'.
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	10 #
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	11
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	12 import logging
40 e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	13 import os
e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	14
33 f96a18c10836 Made getan python-2.4 compatible. Ingo Weinzierl <ingo.weinzierl@intevation.de> parents: 23 diff changeset	15 try:
f96a18c10836 Made getan python-2.4 compatible. Ingo Weinzierl <ingo.weinzierl@intevation.de> parents: 23 diff changeset	16 import sqlite3 as db
f96a18c10836 Made getan python-2.4 compatible. Ingo Weinzierl <ingo.weinzierl@intevation.de> parents: 23 diff changeset	17 except ImportError:
f96a18c10836 Made getan python-2.4 compatible. Ingo Weinzierl <ingo.weinzierl@intevation.de> parents: 23 diff changeset	18 from pysqlite2 import dbapi2 as db
23 9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	19
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	20 from getan.project import Project, Entry
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	21
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	22 DEFAULT_DATABASE = "time.db"
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	23
40 e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	24 CREATE_TABLES = [
348 328ec66517b0 Fix coding style for pep8 Björn Ricks <bjoern.ricks@intevation.de> parents: 284 diff changeset	25 """
40 e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	26 CREATE TABLE projects (
e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	27 id INTEGER PRIMARY KEY AUTOINCREMENT,
284 561441fde2ac Set unique constraint on project key Björn Ricks <bjoern.ricks@intevation.de> parents: 174 diff changeset	28 key VARCHAR(16) NOT NULL CONSTRAINT unique_key UNIQUE,
40 e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	29 description VARCHAR(256),
e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	30 active BOOLEAN DEFAULT 1
e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	31 )
e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	32 """,
348 328ec66517b0 Fix coding style for pep8 Björn Ricks <bjoern.ricks@intevation.de> parents: 284 diff changeset	33 """
40 e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	34 CREATE TABLE entries (
e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	35 id INTEGER PRIMARY KEY AUTOINCREMENT,
e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	36 project_id INTEGER REFERENCES projects(id),
e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	37 start_time TIMESTAMP NOT NULL,
e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	38 stop_time TIMESTAMP NOT NULL,
e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	39 description VARCHAR(256),
e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	40
e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	41 CHECK (strftime('%s', start_time) <= strftime('%s', stop_time))
e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	42 )
e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	43 """
e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	44 ]
e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	45
23 9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	46 LOAD_ACTIVE_PROJECTS = '''
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	47 SELECT id, key, description, total
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	48 FROM projects LEFT JOIN
144 9e853f9e63ad Remove whitespace Björn Ricks <bjoern.ricks@intevation.de> parents: 143 diff changeset	49 (SELECT
9e853f9e63ad Remove whitespace Björn Ricks <bjoern.ricks@intevation.de> parents: 143 diff changeset	50 project_id,
23 9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	51 sum(strftime('%s', stop_time) - strftime('%s', start_time)) AS total
144 9e853f9e63ad Remove whitespace Björn Ricks <bjoern.ricks@intevation.de> parents: 143 diff changeset	52 FROM entries
23 9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	53 GROUP BY project_id) ON project_id = id
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	54 WHERE active
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	55 '''
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	56
351 b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	57 LOAD_ACTIVE_PROJECT = '''
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	58 SELECT id, key, description, total
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	59 FROM projects LEFT JOIN
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	60 (SELECT
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	61 project_id,
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	62 sum(strftime('%s', stop_time) - strftime('%s', start_time)) AS total
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	63 FROM entries
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	64 GROUP BY project_id) ON project_id = id
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	65 WHERE active and key = :project_id
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	66 '''
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	67
23 9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	68 LOAD_PROJECT_ENTRIES = '''
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	69 SELECT
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	70 id,
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	71 project_id,
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	72 start_time as "[timestamp]",
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	73 stop_time as "[timestamp]",
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	74 description
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	75 FROM
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	76 entries
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	77 WHERE
350 f581752317fd Correctly escape sql query Björn Ricks <bjoern.ricks@intevation.de> parents: 349 diff changeset	78 project_id = :project_id
23 9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	79 ORDER BY
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	80 id
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	81 DESC
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	82 '''
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	83
351 b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	84 LOAD_PROJECT_ENTRIES_YEAR = '''
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	85 SELECT
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	86 id,
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	87 project_id,
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	88 start_time as "[timestamp]",
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	89 stop_time as "[timestamp]",
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	90 'no description' AS description
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	91 FROM entries
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	92 WHERE
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	93 project_id = :project_id AND
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	94 (description IS NULL or length(description) = 0)
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	95 GROUP BY round(julianday(start_time))
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	96 UNION
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	97 SELECT
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	98 id,
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	99 project_id,
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	100 start_time as "[timestamp]",
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	101 stop_time as "[timestamp]",
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	102 description
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	103 FROM entries
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	104 WHERE
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	105 project_id = :project_id AND
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	106 (strftime('%Y', start_time) ) = :year AND
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	107 description IS NOT NULL AND length(description) > 0
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	108 '''
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	109
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	110 LOAD_PROJECT_ENTRIES_WEEK = '''
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	111 SELECT
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	112 id,
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	113 project_id,
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	114 start_time as "[timestamp]",
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	115 stop_time as "[timestamp]",
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	116 'no description' AS description
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	117 FROM entries
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	118 WHERE
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	119 project_id = :project_id AND
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	120 (strftime('%Y', start_time) ) = :year AND
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	121 (description IS NULL or length(description) = 0)
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	122 AND (strftime('%W', start_time) = :week
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	123 OR strftime('%W', stop_time) = :week)
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	124 GROUP BY round(julianday(start_time))
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	125 UNION
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	126 SELECT
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	127 id,
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	128 project_id,
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	129 start_time as "[timestamp]",
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	130 stop_time as "[timestamp]",
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	131 description
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	132 FROM entries
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	133 WHERE
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	134 project_id = :project_id AND
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	135 (strftime('%Y', start_time) ) = :year AND
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	136 description IS NOT NULL AND length(description) > 0
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	137 AND (strftime('%W', start_time) = :week
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	138 OR strftime('%W', stop_time) = :week)
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	139 ORDER BY start_time
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	140 '''
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	141
23 9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	142 INSERT_PROJECT_ENTRY = '''
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	143 INSERT INTO entries (project_id, start_time, stop_time, description)
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	144 VALUES(?,?,?,?)
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	145 '''
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	146
49 062ce001abd1 add backend method to create new projects Björn Ricks <bjoern.ricks@intevation.de> parents: 40 diff changeset	147 INSERT_PROJECT = '''
52 9cb57c37369a fix insert new project sql statemenent and api Björn Ricks <bjoern.ricks@intevation.de> parents: 50 diff changeset	148 INSERT INTO projects (key, description, active) VALUES (?,?,1)
49 062ce001abd1 add backend method to create new projects Björn Ricks <bjoern.ricks@intevation.de> parents: 40 diff changeset	149 '''
062ce001abd1 add backend method to create new projects Björn Ricks <bjoern.ricks@intevation.de> parents: 40 diff changeset	150
23 9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	151 DELETE_PROJECT_ENTRY = 'DELETE FROM entries WHERE id = %i'
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	152
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	153 MOVE_ENTRY = 'UPDATE entries SET project_id = ? WHERE id = ?'
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	154
68 a25dab413260 Add new method to update a description of an entry in the database backend Björn Ricks <bjoern.ricks@intevation.de> parents: 52 diff changeset	155 UPDATE_ENTRY = 'UPDATE entries SET description = ? WHERE id = ?'
a25dab413260 Add new method to update a description of an entry in the database backend Björn Ricks <bjoern.ricks@intevation.de> parents: 52 diff changeset	156
174 36b9f2d31810 Add backend method to update a project in the database Björn Ricks <bjoern.ricks@intevation.de> parents: 165 diff changeset	157 UPDATE_PROJECT = "UPDATE projects SET key = ?, description = ? WHERE id = ?"
36b9f2d31810 Add backend method to update a project in the database Björn Ricks <bjoern.ricks@intevation.de> parents: 165 diff changeset	158
23 9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	159 logger = logging.getLogger()
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	160
348 328ec66517b0 Fix coding style for pep8 Björn Ricks <bjoern.ricks@intevation.de> parents: 284 diff changeset	161
165 83ba64c9a51e A project entry in the db must have a key Björn Ricks <bjoern.ricks@intevation.de> parents: 146 diff changeset	162 class InvalidProjectKeyError(Exception):
83ba64c9a51e A project entry in the db must have a key Björn Ricks <bjoern.ricks@intevation.de> parents: 146 diff changeset	163 pass
83ba64c9a51e A project entry in the db must have a key Björn Ricks <bjoern.ricks@intevation.de> parents: 146 diff changeset	164
348 328ec66517b0 Fix coding style for pep8 Björn Ricks <bjoern.ricks@intevation.de> parents: 284 diff changeset	165
143 9f4da22fb1f1 Use new style class for backend Björn Ricks <bjoern.ricks@intevation.de> parents: 135 diff changeset	166 class Backend(object):
23 9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	167
348 328ec66517b0 Fix coding style for pep8 Björn Ricks <bjoern.ricks@intevation.de> parents: 284 diff changeset	168 def __init__(self, database=DEFAULT_DATABASE):
23 9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	169 self.database = database
40 e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	170 self.ensure_exists()
348 328ec66517b0 Fix coding style for pep8 Björn Ricks <bjoern.ricks@intevation.de> parents: 284 diff changeset	171 self.con = db.connect(database,
328ec66517b0 Fix coding style for pep8 Björn Ricks <bjoern.ricks@intevation.de> parents: 284 diff changeset	172 detect_types=db.PARSE_DECLTYPES \|
328ec66517b0 Fix coding style for pep8 Björn Ricks <bjoern.ricks@intevation.de> parents: 284 diff changeset	173 db.PARSE_COLNAMES)
23 9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	174 self.con.text_factory = lambda x: unicode(x, "utf-8", "ignore")
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	175
40 e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	176 def ensure_exists(self):
e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	177 """ Creates the database file if it does not exist. """
348 328ec66517b0 Fix coding style for pep8 Björn Ricks <bjoern.ricks@intevation.de> parents: 284 diff changeset	178 if os.path.isfile(self.database):
328ec66517b0 Fix coding style for pep8 Björn Ricks <bjoern.ricks@intevation.de> parents: 284 diff changeset	179 return
40 e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	180
e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	181 con, cur = None, None
e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	182 try:
348 328ec66517b0 Fix coding style for pep8 Björn Ricks <bjoern.ricks@intevation.de> parents: 284 diff changeset	183 con = db.connect(self.database)
40 e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	184 cur = con.cursor()
e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	185 try:
e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	186 for sql in CREATE_TABLES:
e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	187 cur.execute(sql)
e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	188 con.commit()
e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	189 except:
e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	190 con.rollback()
e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	191 raise
e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	192 finally:
348 328ec66517b0 Fix coding style for pep8 Björn Ricks <bjoern.ricks@intevation.de> parents: 284 diff changeset	193 if cur:
328ec66517b0 Fix coding style for pep8 Björn Ricks <bjoern.ricks@intevation.de> parents: 284 diff changeset	194 cur.close()
328ec66517b0 Fix coding style for pep8 Björn Ricks <bjoern.ricks@intevation.de> parents: 284 diff changeset	195 if con:
328ec66517b0 Fix coding style for pep8 Björn Ricks <bjoern.ricks@intevation.de> parents: 284 diff changeset	196 con.close()
40 e4759cc8f5e7 Create database file if it does not exist. Sascha L. Teichmann <teichmann@intevation.de> parents: 33 diff changeset	197
23 9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	198 def load_projects(self):
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	199 """ Loads active projects from database and returns them as array """
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	200 logger.debug("load active projects from database.")
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	201 cur = None
348 328ec66517b0 Fix coding style for pep8 Björn Ricks <bjoern.ricks@intevation.de> parents: 284 diff changeset	202 try:
23 9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	203 cur = self.con.cursor()
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	204 cur.execute(LOAD_ACTIVE_PROJECTS)
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	205
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	206 projects = []
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	207 while True:
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	208 row = cur.fetchone()
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	209
135 ce707fbb9666 Change coding style of if clauses Björn Ricks <bjoern.ricks@intevation.de> parents: 97 diff changeset	210 if not row:
ce707fbb9666 Change coding style of if clauses Björn Ricks <bjoern.ricks@intevation.de> parents: 97 diff changeset	211 break
165 83ba64c9a51e A project entry in the db must have a key Björn Ricks <bjoern.ricks@intevation.de> parents: 146 diff changeset	212 # check key
83ba64c9a51e A project entry in the db must have a key Björn Ricks <bjoern.ricks@intevation.de> parents: 146 diff changeset	213 if not row[1]:
348 328ec66517b0 Fix coding style for pep8 Björn Ricks <bjoern.ricks@intevation.de> parents: 284 diff changeset	214 raise InvalidProjectKeyError("Project with id %s needs "
328ec66517b0 Fix coding style for pep8 Björn Ricks <bjoern.ricks@intevation.de> parents: 284 diff changeset	215 "a key" % row[0])
353 5ded6192b85b Pass the Backend to the Project constructor Björn Ricks <bjoern.ricks@intevation.de> parents: 351 diff changeset	216 proj = Project(self, *row)
23 9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	217 projects.append(proj)
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	218
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	219 logger.info("found %i active projects." % len(projects))
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	220 return projects
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	221
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	222 finally:
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	223 close(cur)
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	224
351 b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	225 def load_project(self, key):
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	226 logger.debug("load active projects from database.")
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	227 cur = None
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	228 try:
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	229 cur = self.con.cursor()
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	230 cur.execute(LOAD_ACTIVE_PROJECT, {"project_id": key})
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	231 row = cur.fetchone()
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	232 if not row:
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	233 raise InvalidProjectKeyError("Project with key %s not "
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	234 "found." % key)
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	235 # check key
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	236 if not row[1]:
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	237 raise InvalidProjectKeyError("Project with id %s needs "
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	238 "a key" % row[0])
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	239 return Project(self, *row)
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	240
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	241 finally:
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	242 close(cur)
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	243
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	244 def load_entries(self, project_id, year=None, week=None):
23 9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	245 """ Loads all entries that belong to a specific project """
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	246 logger.debug("load entries that belong to project %s" % project_id)
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	247 cur = None
351 b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	248
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	249 if week and isinstance(week, int):
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	250 week = "%02d" % (week)
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	251
23 9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	252 try:
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	253 cur = self.con.cursor()
351 b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	254
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	255 if year is None and week is None:
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	256 cur.execute(LOAD_PROJECT_ENTRIES,
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	257 {"project_id": project_id})
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	258 elif week is None:
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	259 cur.execute(LOAD_PROJECT_ENTRIES_YEAR,
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	260 {'project_id': project_id, 'year': str(year)})
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	261 else:
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	262 cur.execute(LOAD_PROJECT_ENTRIES_WEEK,
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	263 {'project_id': project_id,
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	264 'week': week,
b5dc92631561 Allow to load one project and specific entries from the Backend Björn Ricks <bjoern.ricks@intevation.de> parents: 350 diff changeset	265 'year': str(year)})
23 9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	266
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	267 entries = []
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	268 while True:
146 bb70dde875fd Always fail if an entry couldn't be loaded Björn Ricks <bjoern.ricks@intevation.de> parents: 144 diff changeset	269 row = cur.fetchone()
23 9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	270
146 bb70dde875fd Always fail if an entry couldn't be loaded Björn Ricks <bjoern.ricks@intevation.de> parents: 144 diff changeset	271 if not row:
bb70dde875fd Always fail if an entry couldn't be loaded Björn Ricks <bjoern.ricks@intevation.de> parents: 144 diff changeset	272 break
bb70dde875fd Always fail if an entry couldn't be loaded Björn Ricks <bjoern.ricks@intevation.de> parents: 144 diff changeset	273 entries.append(Entry(*row))
23 9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	274
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	275 logger.debug("Found %i entries that belong to project '%i'"
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	276 % (len(entries), project_id))
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	277 return entries
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	278 finally:
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	279 close(cur)
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	280
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	281 def insert_project_entry(self, project, stop_time, desc):
135 ce707fbb9666 Change coding style of if clauses Björn Ricks <bjoern.ricks@intevation.de> parents: 97 diff changeset	282 if project is None:
ce707fbb9666 Change coding style of if clauses Björn Ricks <bjoern.ricks@intevation.de> parents: 97 diff changeset	283 return
23 9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	284 cur = None
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	285 try:
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	286 cur = self.con.cursor()
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	287 cur.execute(INSERT_PROJECT_ENTRY, (
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	288 project.id, project.start, stop_time, desc))
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	289 self.con.commit()
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	290 logger.debug("Added new entry '%s' of project '%s' into db"
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	291 % (desc, project.desc))
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	292
362 90c09cca49c3 Fix loading entries of a project Björn Ricks <bjoern.ricks@intevation.de> parents: 353 diff changeset	293 project.load_entries()
23 9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	294 finally:
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	295 close(cur)
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	296
52 9cb57c37369a fix insert new project sql statemenent and api Björn Ricks <bjoern.ricks@intevation.de> parents: 50 diff changeset	297 def insert_project(self, key, description):
9cb57c37369a fix insert new project sql statemenent and api Björn Ricks <bjoern.ricks@intevation.de> parents: 50 diff changeset	298 if key is None or description is None:
49 062ce001abd1 add backend method to create new projects Björn Ricks <bjoern.ricks@intevation.de> parents: 40 diff changeset	299 return
062ce001abd1 add backend method to create new projects Björn Ricks <bjoern.ricks@intevation.de> parents: 40 diff changeset	300 cur = None
50 9dbb6ee443a4 don't forget to save before commit :-/ BjÃ¶rn Ricks <bjoern.ricks@intevation.de> parents: 49 diff changeset	301 try:
49 062ce001abd1 add backend method to create new projects Björn Ricks <bjoern.ricks@intevation.de> parents: 40 diff changeset	302 cur = self.con.cursor()
52 9cb57c37369a fix insert new project sql statemenent and api Björn Ricks <bjoern.ricks@intevation.de> parents: 50 diff changeset	303 cur.execute(INSERT_PROJECT, (key, description))
49 062ce001abd1 add backend method to create new projects Björn Ricks <bjoern.ricks@intevation.de> parents: 40 diff changeset	304 self.con.commit()
52 9cb57c37369a fix insert new project sql statemenent and api Björn Ricks <bjoern.ricks@intevation.de> parents: 50 diff changeset	305 logger.debug("Added a new project '%s' into db" % description)
50 9dbb6ee443a4 don't forget to save before commit :-/ BjÃ¶rn Ricks <bjoern.ricks@intevation.de> parents: 49 diff changeset	306 finally:
9dbb6ee443a4 don't forget to save before commit :-/ BjÃ¶rn Ricks <bjoern.ricks@intevation.de> parents: 49 diff changeset	307 close(cur)
23 9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	308
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	309 def delete_entries(self, entries):
135 ce707fbb9666 Change coding style of if clauses Björn Ricks <bjoern.ricks@intevation.de> parents: 97 diff changeset	310 if entries is None:
ce707fbb9666 Change coding style of if clauses Björn Ricks <bjoern.ricks@intevation.de> parents: 97 diff changeset	311 return
23 9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	312
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	313 cur = None
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	314 try:
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	315 cur = self.con.cursor()
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	316 for entry in entries:
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	317 cur.execute(DELETE_PROJECT_ENTRY % entry.id)
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	318 logger.debug("Deleted entry: %s (%d)" % (entry.desc, entry.id))
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	319 self.con.commit()
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	320 finally:
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	321 close(cur)
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	322
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	323 def move_entries(self, entries, new_project_id):
135 ce707fbb9666 Change coding style of if clauses Björn Ricks <bjoern.ricks@intevation.de> parents: 97 diff changeset	324 if entries is None or new_project_id is None:
ce707fbb9666 Change coding style of if clauses Björn Ricks <bjoern.ricks@intevation.de> parents: 97 diff changeset	325 return
23 9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	326
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	327 cur = None
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	328 try:
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	329 cur = self.con.cursor()
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	330 for entry in entries:
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	331 cur.execute(MOVE_ENTRY, (new_project_id, entry.id))
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	332 logger.debug("Moved entry '%s' (id=%d) to project with id %d."
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	333 % (entry.desc, entry.id, new_project_id))
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	334 self.con.commit()
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	335 finally:
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	336 close(cur)
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	337
68 a25dab413260 Add new method to update a description of an entry in the database backend Björn Ricks <bjoern.ricks@intevation.de> parents: 52 diff changeset	338 def update_entry(self, entry):
a25dab413260 Add new method to update a description of an entry in the database backend Björn Ricks <bjoern.ricks@intevation.de> parents: 52 diff changeset	339 if not entry:
a25dab413260 Add new method to update a description of an entry in the database backend Björn Ricks <bjoern.ricks@intevation.de> parents: 52 diff changeset	340 return
a25dab413260 Add new method to update a description of an entry in the database backend Björn Ricks <bjoern.ricks@intevation.de> parents: 52 diff changeset	341
a25dab413260 Add new method to update a description of an entry in the database backend Björn Ricks <bjoern.ricks@intevation.de> parents: 52 diff changeset	342 cur = None
a25dab413260 Add new method to update a description of an entry in the database backend Björn Ricks <bjoern.ricks@intevation.de> parents: 52 diff changeset	343 try:
a25dab413260 Add new method to update a description of an entry in the database backend Björn Ricks <bjoern.ricks@intevation.de> parents: 52 diff changeset	344 cur = self.con.cursor()
a25dab413260 Add new method to update a description of an entry in the database backend Björn Ricks <bjoern.ricks@intevation.de> parents: 52 diff changeset	345 cur.execute(UPDATE_ENTRY, (entry.desc, entry.id))
a25dab413260 Add new method to update a description of an entry in the database backend Björn Ricks <bjoern.ricks@intevation.de> parents: 52 diff changeset	346 self.con.commit()
a25dab413260 Add new method to update a description of an entry in the database backend Björn Ricks <bjoern.ricks@intevation.de> parents: 52 diff changeset	347 logger.debug("Updated entry: '%s' (%d)" % (entry.desc, entry.id))
a25dab413260 Add new method to update a description of an entry in the database backend Björn Ricks <bjoern.ricks@intevation.de> parents: 52 diff changeset	348 finally:
a25dab413260 Add new method to update a description of an entry in the database backend Björn Ricks <bjoern.ricks@intevation.de> parents: 52 diff changeset	349 close(cur)
a25dab413260 Add new method to update a description of an entry in the database backend Björn Ricks <bjoern.ricks@intevation.de> parents: 52 diff changeset	350
174 36b9f2d31810 Add backend method to update a project in the database Björn Ricks <bjoern.ricks@intevation.de> parents: 165 diff changeset	351 def update_project(self, project):
36b9f2d31810 Add backend method to update a project in the database Björn Ricks <bjoern.ricks@intevation.de> parents: 165 diff changeset	352 if not project:
36b9f2d31810 Add backend method to update a project in the database Björn Ricks <bjoern.ricks@intevation.de> parents: 165 diff changeset	353 return
36b9f2d31810 Add backend method to update a project in the database Björn Ricks <bjoern.ricks@intevation.de> parents: 165 diff changeset	354
36b9f2d31810 Add backend method to update a project in the database Björn Ricks <bjoern.ricks@intevation.de> parents: 165 diff changeset	355 cur = None
36b9f2d31810 Add backend method to update a project in the database Björn Ricks <bjoern.ricks@intevation.de> parents: 165 diff changeset	356 try:
36b9f2d31810 Add backend method to update a project in the database Björn Ricks <bjoern.ricks@intevation.de> parents: 165 diff changeset	357 cur = self.con.cursor()
348 328ec66517b0 Fix coding style for pep8 Björn Ricks <bjoern.ricks@intevation.de> parents: 284 diff changeset	358 cur.execute(
328ec66517b0 Fix coding style for pep8 Björn Ricks <bjoern.ricks@intevation.de> parents: 284 diff changeset	359 UPDATE_PROJECT, (project.key, project.desc, project.id))
174 36b9f2d31810 Add backend method to update a project in the database Björn Ricks <bjoern.ricks@intevation.de> parents: 165 diff changeset	360 self.con.commit()
36b9f2d31810 Add backend method to update a project in the database Björn Ricks <bjoern.ricks@intevation.de> parents: 165 diff changeset	361 logger.debug("Updated project: (%d) %s %s" % (project.id,
349 c97a4b6a6887 Fix textwidth <= 80 Björn Ricks <bjoern.ricks@intevation.de> parents: 348 diff changeset	362 project.key,
c97a4b6a6887 Fix textwidth <= 80 Björn Ricks <bjoern.ricks@intevation.de> parents: 348 diff changeset	363 project.desc))
174 36b9f2d31810 Add backend method to update a project in the database Björn Ricks <bjoern.ricks@intevation.de> parents: 165 diff changeset	364 finally:
36b9f2d31810 Add backend method to update a project in the database Björn Ricks <bjoern.ricks@intevation.de> parents: 165 diff changeset	365 close(cur)
36b9f2d31810 Add backend method to update a project in the database Björn Ricks <bjoern.ricks@intevation.de> parents: 165 diff changeset	366
348 328ec66517b0 Fix coding style for pep8 Björn Ricks <bjoern.ricks@intevation.de> parents: 284 diff changeset	367
23 9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	368 def close(cur):
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	369 """ This function closes a database cursor if it is existing """
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	370 if cur:
135 ce707fbb9666 Change coding style of if clauses Björn Ricks <bjoern.ricks@intevation.de> parents: 97 diff changeset	371 try:
ce707fbb9666 Change coding style of if clauses Björn Ricks <bjoern.ricks@intevation.de> parents: 97 diff changeset	372 cur.close()
23 9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	373 except:
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	374 logger.warn("could not close database cursor.")
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	375
9c4e8ba3c4fa Added a new implementation of 'getan' based on urwid, a python console user interface library. Ingo Weinzierl <ingo_weinzierl@web.de> parents: diff changeset	376 # vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8:

Mercurial > getan > getan

annotate getan/backend.py @ 362:90c09cca49c3