Adds end_batch functionality
This commit is contained in:
		
							parent
							
								
									3db1256895
								
							
						
					
					
						commit
						5929c0af3c
					
				
					 1 changed files with 5 additions and 3 deletions
				
			
		|  | @ -49,7 +49,11 @@ class BatchController(object): | ||||||
|         cache[cls._NESTING_KEY] -= 1 |         cache[cls._NESTING_KEY] -= 1 | ||||||
| 
 | 
 | ||||||
|         if cache[cls._NESTING_KEY] == 0: |         if cache[cls._NESTING_KEY] == 0: | ||||||
|             # TODO: Handle batch end cases | 
 | ||||||
|  |             for key in cache: | ||||||
|  |                 item = cache[key] | ||||||
|  |                 if hasattr(item, 'end_batch') and callable(item.end_batch): | ||||||
|  |                     item.end_batch() | ||||||
| 
 | 
 | ||||||
|             del cls._user_caches[user] |             del cls._user_caches[user] | ||||||
| 
 | 
 | ||||||
|  | @ -115,6 +119,4 @@ Tests: | ||||||
| - ``end_batch`` behaviour for CartController (use for_user *A LOT*) | - ``end_batch`` behaviour for CartController (use for_user *A LOT*) | ||||||
|   - discounts not calculated until outermost batch point exits. |   - discounts not calculated until outermost batch point exits. | ||||||
|   - Revision number shouldn't change until outermost batch point exits. |   - Revision number shouldn't change until outermost batch point exits. | ||||||
| - Make sure memoisation ONLY happens when we're in a batch. |  | ||||||
| 
 |  | ||||||
| ''' | ''' | ||||||
|  |  | ||||||
		Loading…
	
	Add table
		
		Reference in a new issue
	
	 Christopher Neugebauer
						Christopher Neugebauer