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:
|
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:'))
|
||||||
Reference in New Issue
Block a user