anima/timetrack
1
1
Fork 0
Code Issues 18 Projects 2 Releases 4 Wiki Activity

add set_user function an fix some bugs

Browse Source
This commit is contained in:
Schreiner Pascal
2021-05-31 00:12:31 +02:00
parent 3db70a8486
commit 3a189ad8ed
1 changed files with 20 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
timeTrack.py
View File

@@ -19,7 +19,8 @@ class TimeTrack:
def __init__(self, DATABASE='timetrack.db', CONFIG='timetrack.conf'): def __init__(self, DATABASE='timetrack.db', CONFIG='timetrack.conf'):
self.DATABASE = DATABASE self.DATABASE = DATABASE
self.CONFIG = CONFIG self.CONFIG = CONFIG
self.USER = 0 self.USERID = 0
self.USERNAME = ''
self.LOGFILE = 'timetrack.log' self.LOGFILE = 'timetrack.log'
logging.basicConfig(filename=self.LOGFILE, logging.basicConfig(filename=self.LOGFILE,
level=logging.DEBUG, level=logging.DEBUG,
@@ -28,6 +29,7 @@ class TimeTrack:
) )
self.load_config() self.load_config()
self.db_setup() self.db_setup()
self.set_user()
## Läd oder erstellt das config file ## Läd oder erstellt das config file
def load_config(self): def load_config(self):
@@ -42,7 +44,7 @@ class TimeTrack:
quit() quit()
else: else:
logging.info('Config file loaded successfully') logging.info('Config file loaded successfully')
self.USER = data['user'] self.USERID = data['user']
else: else:
logging.info('Config file not found') logging.info('Config file not found')
@@ -119,13 +121,9 @@ class TimeTrack:
logging.debug(SQL) logging.debug(SQL)
else: else:
logging.info('Success execute SQL command') logging.info('Success execute SQL command')
connect.commit
connect.close() connect.commit
connect.close()
##> Kürzer möglich ?
def db_connect(self, SQL, DATA=''):
pass
def create_user(self, USER=''): def create_user(self, USER=''):
if USER == '': if USER == '':
@@ -150,18 +148,18 @@ class TimeTrack:
else: else:
logging.info('Success execute SQL command') logging.info('Success execute SQL command')
connect.commit() connect.commit()
connect.close()
connect.close()
##> verbugt noch nicht funktional
def get_users(self, UID=0, NAME=''): def get_users(self, UID=0, NAME=''):
if not UID == 0: if not UID == 0:
logging.info('UID Used') logging.info('UID Used')
data = UID data = UID
sql = "SELECT name FROM users WHERE id = ?" sql = "SELECT * FROM users WHERE id = ?"
elif not NAME == '': elif not NAME == '':
logging.info('NAME used') logging.info('NAME used')
data = NAME data = NAME
sql = "SELECT name FROM users WHERE name = ?" sql = "SELECT * FROM users WHERE name = ?"
else: else:
logging.info('nothing used') logging.info('nothing used')
data = '' data = ''
@@ -186,15 +184,19 @@ class TimeTrack:
row = cursor.fetchall() row = cursor.fetchall()
output = [] output = []
for data in row: for data in row:
print(data)
output.append(data) output.append(data)
connect.close() connect.close()
return output return output
def set_user(self): def set_user(self):
pass data = self.get_users()
if not data:
logging.info("No user found. Create new.")
print("Es wurde kein User gefunden, bitte legen sie einen neuen an.")
self.create_user()
data = self.get_users(UID=self.USERID)
self.USERNAME = data[0][1]
test = TimeTrack() test = TimeTrack()
test.create_user()
#test.get_users(NAME=input('Test:'))