create time_start function and clear_screan function.
This commit is contained in:
92
timeTrack.py
92
timeTrack.py
@@ -3,15 +3,13 @@
|
||||
#
|
||||
# timeTrack.py
|
||||
# by 4nima
|
||||
# v.0.3.0
|
||||
# v.0.4.0
|
||||
#
|
||||
#########################
|
||||
# simple time tracking with database
|
||||
#########################
|
||||
|
||||
import datetime
|
||||
from sqlite3.dbapi2 import Cursor
|
||||
import time
|
||||
import sqlite3
|
||||
import json
|
||||
import os
|
||||
@@ -24,6 +22,7 @@ class TimeTrack:
|
||||
self.CONFIG = CONFIG
|
||||
self.USERID = 0
|
||||
self.USERNAME = ''
|
||||
self.OLDEVENT = 2
|
||||
self.LOGFILE = 'timetrack.log'
|
||||
logging.basicConfig(
|
||||
filename=self.LOGFILE,
|
||||
@@ -35,6 +34,15 @@ class TimeTrack:
|
||||
self.load_config()
|
||||
|
||||
## Prepartation
|
||||
### Check OS and clear screen
|
||||
def clear_screen(self):
|
||||
if os.name == 'posix':
|
||||
logging.debug('Unix/Linux system detected')
|
||||
_ = os.system('clear')
|
||||
else:
|
||||
logging.debug('Winwos System detected')
|
||||
_ = os.system('cls')
|
||||
|
||||
### Creates a database if none is found
|
||||
def db_setup(self):
|
||||
if os.path.isfile(self.DATABASE):
|
||||
@@ -118,14 +126,16 @@ class TimeTrack:
|
||||
else:
|
||||
logging.info('Config file was loaded successfully')
|
||||
self.USERID = data['user']
|
||||
self.OLDEVENT = data['oldevent']
|
||||
logging.debug('UserID {} was used'.format(data['user']))
|
||||
self.set_user()
|
||||
|
||||
else:
|
||||
logging.warning('Config file not found')
|
||||
config = {
|
||||
'default' : 'interactive',
|
||||
'user' : 1,
|
||||
'default' : 'interactive'
|
||||
'oldevent' : 2
|
||||
}
|
||||
with open(self.CONFIG, "w") as outfile:
|
||||
json.dump(config, outfile, indent=4, sort_keys=True)
|
||||
@@ -221,7 +231,7 @@ class TimeTrack:
|
||||
|
||||
## Time handling
|
||||
### Creates an active event if none exists for the user
|
||||
def set_event(self, TIME=datetime.datetime.now()):
|
||||
def save_event(self, TIME=datetime.datetime.now()):
|
||||
if not self.get_event(USERID=self.USERID):
|
||||
logging.debug('No active events found for the user: {}'.format(self.USERID))
|
||||
connect = sqlite3.connect(self.DATABASE)
|
||||
@@ -234,13 +244,16 @@ class TimeTrack:
|
||||
logging.error('Event could not be created')
|
||||
logging.debug(sql)
|
||||
print('Event konnte nicht gespeichert werden.')
|
||||
return False
|
||||
else:
|
||||
logging.info('Event was created successfully')
|
||||
connect.commit()
|
||||
|
||||
connect.close()
|
||||
return True
|
||||
else:
|
||||
logging.warning('Active events found for the user, new event could not be created')
|
||||
return False
|
||||
|
||||
### Deletes an active event based on a user or event ID
|
||||
def delete_event(self, ENTRYID='', USERID=''):
|
||||
@@ -309,14 +322,64 @@ class TimeTrack:
|
||||
else:
|
||||
logging.debug('{} events found'.format(len(data)))
|
||||
return data[0]
|
||||
####################################################################################################
|
||||
####################################################################################################
|
||||
####################################################################################################
|
||||
####################################################################################################
|
||||
def time_start(self):
|
||||
pass
|
||||
|
||||
def time_end(self):
|
||||
def time_start(self, AUTOFORWARD=True):
|
||||
self.clear_screen()
|
||||
starttime = datetime.datetime.now()
|
||||
logging.info('New Event process started at {}'.format(starttime))
|
||||
if self.save_event(starttime):
|
||||
print('Neues Event gestartet um {}'.format(starttime.strftime("%H:%M")))
|
||||
else:
|
||||
data = self.get_event(USERID=self.USERID)
|
||||
print('Es existiert bereits ein aktives Event.')
|
||||
if (data[1] + datetime.timedelta(hours=self.OLDEVENT)) <= datetime.datetime.now():
|
||||
logging.info('Event exceeds allowed duration')
|
||||
print('Zeiteintrag ist zu alt laut den Einstellungen (älter {} Stunden)'.format(self.OLDEVENT))
|
||||
|
||||
if datetime.date.today() == data[1].date():
|
||||
print('Start um: {}'.format(data[1].strftime("%H:%M")))
|
||||
else:
|
||||
print('Start am: {}'.format(data[1].strftime("%d.%m.%Y um %H:%M")))
|
||||
|
||||
elapsed = datetime.datetime.now() - data[1]
|
||||
if elapsed.days:
|
||||
logging.debug('Event older than 1 day ({} days)'.format(elapsed.days))
|
||||
print('Vergangene Zeit: >{} Tage'.format(elapsed.days))
|
||||
else:
|
||||
logging.debug('Event younger than 1 day')
|
||||
print('Vergangene Zeit: >{} Stunden'.format(int(elapsed.seconds/3600)))
|
||||
|
||||
userinput = ''
|
||||
while True:
|
||||
if userinput == "A" or userinput == "a" or userinput == "1" \
|
||||
or userinput == "F" or userinput == "f" or userinput == "2" \
|
||||
or userinput == "L" or userinput == "l" or userinput == "3":
|
||||
break
|
||||
print('Soll das Event fortgesetzt oder gelöscht werden?')
|
||||
print('[1/F/f] für fortsetzen')
|
||||
print('[2/L/l] für löschen')
|
||||
print('[3/A/a] für abbrechen')
|
||||
userinput = input('Antwort: ')
|
||||
logging.debug('User input: {}'.format(userinput))
|
||||
self.clear_screen()
|
||||
|
||||
if userinput == "F" or userinput == "f" or userinput == "1":
|
||||
logging.debug('Event should be continued')
|
||||
self.time_stop()
|
||||
elif userinput == "L" or userinput == "l" or userinput == "2":
|
||||
logging.info('Event should be deleted (eventid: {})'.format(data[0]))
|
||||
self.delete_event(data[0])
|
||||
self.time_start()
|
||||
else:
|
||||
logging.debug('Terminated by the user')
|
||||
exit()
|
||||
|
||||
else:
|
||||
logging.debug('Event continues (eventid{})'.format(data[0]))
|
||||
print('Event von {} Uhr geladen'.format(data[1].strftime("%H:%M")))
|
||||
self.time_stop()
|
||||
|
||||
def time_stop(self):
|
||||
pass
|
||||
|
||||
def get_time(self):
|
||||
@@ -325,5 +388,6 @@ class TimeTrack:
|
||||
|
||||
test = TimeTrack()
|
||||
#test.delete_event(USERID=1)
|
||||
#test.set_event()
|
||||
test.get_event()
|
||||
#test.save_event()
|
||||
#test.get_event()
|
||||
test.time_start()
|
||||
Reference in New Issue
Block a user