From 5df707da834d2ea8eb00de36c57444e9a492121d Mon Sep 17 00:00:00 2001
From: anima
Date: Thu, 13 May 2021 19:10:20 +0200
Subject: [PATCH] fix get_user function and add as requirement for create_user
function
---
timeTrack.py | 37 +++++++++++++++++++++++--------------
1 file changed, 23 insertions(+), 14 deletions(-)
diff --git a/timeTrack.py b/timeTrack.py
index 73c2f13..e57a820 100644
--- a/timeTrack.py
+++ b/timeTrack.py
@@ -175,6 +175,7 @@ class TimeTrack:
except:
logging.error('Fail to execute SQL command')
logging.debug(sql)
+ logging.debug(data)
else:
logging.info('Success execute SQL command')
logging.debug(sql)
@@ -184,7 +185,6 @@ class TimeTrack:
for data in cursor:
output.append(data)
connect.close()
- print(output)
return output
def create_user(self, USER=''):
@@ -193,6 +193,11 @@ class TimeTrack:
else:
data = USER
+
+ while self.get_users(NAME=data):
+ print('Name ist schon vergeben')
+ data = input('Wähle einen anderen Namen: ')
+
sql = """
INSERT INTO users (
name
@@ -201,29 +206,33 @@ class TimeTrack:
)
"""
self.db_connect(sql, data)
- pass
##> verbugt noch nicht funktional
def get_users(self, UID=0, NAME=''):
- data = input('User ID: ')
- if UID == 0:
- sql = [
+ if not UID == 0:
+ logging.info('UID Used')
+ data = UID
+ sql = """
+ SELECT name FROM users WHERE id = ?
"""
- SELET id, name FROM users
+ elif not NAME == '':
+ logging.info('NAME used')
+ data = NAME
+ sql = """
+ SELECT name FROM users WHERE name = ?
"""
- ]
else:
- sql = [
+ logging.info('nothing used')
+ sql = """
+ SELECT * FROM users WHERE
"""
- SELET id, name FROM users WHERE id=?
- """
- ]
- print(self.db_connect(sql))
+
+ return self.db_connect(sql, data)
def set_user(self):
pass
test = TimeTrack()
-#test.create_user()
-test.get_users()
\ No newline at end of file
+test.create_user()
+#test.get_users(NAME=input('Test:'))
\ No newline at end of file