def load_users(self): conn = psycopg2.connect(self.ena.conn_string) cursor = conn.cursor() id = 0 level = 0 cursor.execute("SELECT * FROM Users;") # Remember to add fields added to the user constructor here. # To add the fields to the column add them in settings.py for user_info in cursor: user = User() user.user(user_info[0], user_info[1], user_info[2], user_info[3]) self.ena.users.append(user) id = user_info[0] conn.commit() conn.close() return id
def create_user(self): user = User() conn = psycopg2.connect(self.ena.conn_string) cursor = conn.cursor() username = input("username: "******"Password: "******"Confirm Password: "******"INSERT INTO Users VALUES(%s,%s,%s,%s)", (self.ena.id, username, password, level)) conn.commit() conn.close() user.user(id, username, password, level) self.ena.users.append(user) self.ena.users[self.ena.id - 1].logged_in = True self.ena.main_user = user self.ena.commands.load_commands_authenticated_user() else: conn.commit() conn.close() print("passwords did not match")