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:
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()
test.create_user()
#test.get_users(NAME=input('Test:'))