136 lines
3.9 KiB
Python
136 lines
3.9 KiB
Python
#!/bin/python3
|
|
#########################
|
|
#
|
|
# timeTrack.py
|
|
# by 4nima
|
|
# v.0.1
|
|
#
|
|
#########################
|
|
# simple time tracking with database
|
|
#########################
|
|
|
|
import datetime
|
|
import sqlite3
|
|
import json
|
|
import os
|
|
import logging
|
|
|
|
class TimeTrack:
|
|
def __init__(self, DATABASE='timetrack.db', CONFIG='timetrack.conf'):
|
|
self.DATABASE = DATABASE
|
|
self.CONFIG = CONFIG
|
|
self.USER = 0
|
|
self.LOGFILE = 'timetrack.log'
|
|
logging.basicConfig(filename=self.LOGFILE,
|
|
level=logging.DEBUG,
|
|
format='%(asctime)s - %(process)d-%(levelname)s: %(message)s',
|
|
datefmt='%Y-%m-%d %H:%M:%S'
|
|
)
|
|
self.load_config()
|
|
self.db_setup()
|
|
|
|
## Läd oder erstellt das config file
|
|
def load_config(self):
|
|
if os.path.isfile(self.CONFIG):
|
|
logging.info('Config file found')
|
|
try:
|
|
with open(self.CONFIG) as config_data:
|
|
data = json.load(config_data)
|
|
except ValueError:
|
|
logging.error('Config file has no valid JSON syntax')
|
|
print('TimeTrack wird beendet. Fehler in der JSON-Konfig')
|
|
quit()
|
|
else:
|
|
logging.info('Config file loaded successfully')
|
|
self.USER = data['user']
|
|
|
|
else:
|
|
logging.info('Config file not found')
|
|
##> ausgabe nicht in einer zeile
|
|
config = {
|
|
'user' : 1,
|
|
'default' : 'interactive'
|
|
}
|
|
|
|
##> logging erweitert falls nicht möglich
|
|
with open(self.CONFIG, "w") as outfile:
|
|
json.dump(config, outfile)
|
|
logging.info('config file successfully created')
|
|
|
|
def db_setup(self):
|
|
sql = []
|
|
sql.append("""
|
|
CREATE TABLE IF NOT EXISTS time_entries (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
starttime DATETIME NOT NULL,
|
|
endtime DATETIME NOT NULL,
|
|
user_id INTEGER NOT NULL,
|
|
activity TEXT,
|
|
catrgory_id INTEGER,
|
|
client_id INTEGER,
|
|
lock BOOLEAN
|
|
)
|
|
""")
|
|
|
|
sql.append("""
|
|
CREATE TABLE IF NOT EXISTS active_events (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
starttime DATETIME NOT NULL,
|
|
user_id INT NOT NULL
|
|
)
|
|
""")
|
|
|
|
sql.append("""
|
|
CREATE TABLE IF NOT EXISTS users (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
name TEXT NOT NULL,
|
|
worktime INT,
|
|
worktime_span INT
|
|
)
|
|
""")
|
|
|
|
sql.append("""
|
|
CREATE TABLE IF NOT EXISTS clients (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
name TEXT NOT NULL
|
|
)
|
|
""")
|
|
|
|
sql.append("""
|
|
CREATE TABLE IF NOT EXISTS categories (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
name TEXT NOT NULL
|
|
)
|
|
""")
|
|
self.db_connect(sql)
|
|
|
|
def db_connect(self, SQL, DATA=''):
|
|
if os.path.isfile(self.DATABASE):
|
|
logging.info('DB gefunden')
|
|
else:
|
|
logging.info('Keine DB gefunden, neue wird angelegt')
|
|
|
|
connect = sqlite3.connect(self.DATABASE)
|
|
cursor = connect.cursor()
|
|
##> vorgang optimieren...
|
|
for sql in SQL:
|
|
try:
|
|
if isinstance(sql, list):
|
|
cursor.executemany(sql, DATA)
|
|
else:
|
|
cursor.execute(sql)
|
|
except:
|
|
logging.error('Fehler beim schreiben in DB')
|
|
logging.debug(sql)
|
|
else:
|
|
logging.info('Erfolgreich in DB geschrieben')
|
|
logging.debug(sql)
|
|
connect.commit()
|
|
pass
|
|
connect.close()
|
|
|
|
def set_user(self):
|
|
pass
|
|
|
|
|
|
test = TimeTrack() |