diff --git a/fab_access/main.py b/fab_access/main.py index ad95f4f..9a014e0 100644 --- a/fab_access/main.py +++ b/fab_access/main.py @@ -22,19 +22,19 @@ def has_permission(user_permissions, machine_id): return False - def gen_display_name(user): # display names must be not longer than 8 chrs if 'firstName' in user.keys() and 'lastName' in user.keys(): full_name = f'{user["firstName"]} {user["lastName"]}' if len(full_name) > 8: - full_name = 'f{user["firstName"][0]}.{user["lastName"][:6]}' + display_name = f'{user["firstName"][0]}.{user["lastName"][:6]}' else: try: display_name = user['username'][:8] except KeyError: print('user has no username') return 'Error' + return(display_name) def handle_request(msg, client): print('') @@ -48,19 +48,12 @@ def handle_request(msg, client): MqttHandler.print_to_display(reader_id, 16, fabcard_id) return - # TODO Database - # - get machine_id from database - # - get last_user from database - # - get machine_status from database, is 0 for off, 1 for on - # - get plug_id from database - db_data = SQLHandler.get_machine_data(reader_id) - #print(data) machine_id = db_data["machine_id"] - last_user = data["last_user"] - machine_status = data["machine_status"] - plug_id = data["plug_id"] + last_user = db_data["last_user"] + machine_status = db_data["machine_status"] + plug_id = db_data["plug_id"] try: user_permissions = json.loads(user['attributes']['FabPermissions'][0]) @@ -82,7 +75,6 @@ def handle_request(msg, client): MqttHandler.switch_plug(plug_id, 1) MqttHandler.print_to_display(reader_id, 20, f'Login\n{display_name}') else: - print("user1"+last_user+"user2"+username) if not (username == last_user or KeycloakHandler.user_is_privileged(username)): MqttHandler.print_to_display(reader_id, 9, last_user) return @@ -90,10 +82,7 @@ def handle_request(msg, client): MqttHandler.switch_plug(plug_id, 0) MqttHandler.print_to_display(reader_id, 20, f'Bitte anmelden') - # TODO Update Database: - # - last user - # - machine_status - SQLHandler.update_machine(reader_id, last_user, machine_status) + SQLHandler.update_machine(reader_id, username, machine_status) MqttHandler.publish(f'/FabLogging/{plug_id}/USER', username) diff --git a/fab_access/sql_handler.py b/fab_access/sql_handler.py index e210d31..d321a98 100644 --- a/fab_access/sql_handler.py +++ b/fab_access/sql_handler.py @@ -79,5 +79,4 @@ class SQLHandler: machines = json.loads(Config.machines) for machine in machines: SQLHandler.cursor.execute("INSERT INTO readerplug (reader_id, plug_id, machine_id, machine_status, last_user) VALUES (%s, %s, %s, False, 'no_user');", (machine[0],machine[1],machine[2])) - print("-------------") SQLHandler.conn.commit() \ No newline at end of file