diff --git a/mysqlred.py b/mysqlred.py index 589c648..83dd48c 100644 --- a/mysqlred.py +++ b/mysqlred.py @@ -89,14 +89,19 @@ class MysqlRed: # Execute insert query if cls._cached_insert_data: - print( cls._cached_insert_data ) - cursor.executemany( cls._insert_query, ( cls._cached_insert_data[ key ] for key in cls._cached_insert_data ) ) + # Since cls._cached_insert_data is a dict, we need to have a custom + # Generator to iterate over it + cursor.executemany( cls._insert_query, + ( cls._cached_insert_data[ key ] + for key in cls._cached_insert_data ) ) + # Reset after writing cls._cached_insert_data = {} # Execute update query # Use executemany since update could not be reduced to one query if cls._cached_update_data: cursor.executemany( cls._update_query, cls._cached_update_data ) + # Reset after writing cls._cached_update_data = [] # Commit db changes