diff --git a/index.py b/index.py index 83759b9..1f3721e 100644 --- a/index.py +++ b/index.py @@ -7,16 +7,16 @@ } def get_data_by_config_value(value): - # This might look suspicious due to string concatenation with values from CONFIG. - query = "SELECT * FROM " + CONFIG["default_table"] + " WHERE " + CONFIG["default_column"] + " = '" + value + "'" + # Use a parameterized query to prevent SQL injection + query = f"SELECT * FROM {CONFIG['default_table']} WHERE {CONFIG['default_column']} = ?" connection = sqlite3.connect("database.db") cursor = connection.cursor() - cursor.execute(query) + cursor.execute(query, (value,)) result = cursor.fetchall() connection.close() return result # Test -print(get_data_by_config_value("admin")) +print(get_data_by_config_value("admin")) \ No newline at end of file