fix get_user function and add as requirement for create_user function
This commit is contained in:
37
timeTrack.py
37
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()
|
||||
test.create_user()
|
||||
#test.get_users(NAME=input('Test:'))
|
||||
Reference in New Issue
Block a user