fix get_user function and add as requirement for create_user function

This commit is contained in:
2021-05-13 19:10:20 +02:00
parent efff88d5fd
commit 5df707da83

View File

@@ -175,6 +175,7 @@ class TimeTrack:
except: except:
logging.error('Fail to execute SQL command') logging.error('Fail to execute SQL command')
logging.debug(sql) logging.debug(sql)
logging.debug(data)
else: else:
logging.info('Success execute SQL command') logging.info('Success execute SQL command')
logging.debug(sql) logging.debug(sql)
@@ -184,7 +185,6 @@ class TimeTrack:
for data in cursor: for data in cursor:
output.append(data) output.append(data)
connect.close() connect.close()
print(output)
return output return output
def create_user(self, USER=''): def create_user(self, USER=''):
@@ -193,6 +193,11 @@ class TimeTrack:
else: else:
data = USER data = USER
while self.get_users(NAME=data):
print('Name ist schon vergeben')
data = input('Wähle einen anderen Namen: ')
sql = """ sql = """
INSERT INTO users ( INSERT INTO users (
name name
@@ -201,29 +206,33 @@ class TimeTrack:
) )
""" """
self.db_connect(sql, data) self.db_connect(sql, data)
pass
##> verbugt noch nicht funktional ##> verbugt noch nicht funktional
def get_users(self, UID=0, NAME=''): def get_users(self, UID=0, NAME=''):
data = input('User ID: ') if not UID == 0:
if UID == 0: logging.info('UID Used')
sql = [ 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: else:
sql = [ logging.info('nothing used')
sql = """
SELECT * FROM users WHERE
""" """
SELET id, name FROM users WHERE id=?
""" return self.db_connect(sql, data)
]
print(self.db_connect(sql))
def set_user(self): def set_user(self):
pass pass
test = TimeTrack() test = TimeTrack()
#test.create_user() test.create_user()
test.get_users() #test.get_users(NAME=input('Test:'))