south.hacks.django_1_0: 48 total statements, 0.0% covered

Generated: Wed 2013-03-13 10:33 CET

Source file: /media/Envs/Envs/filer-gallery/lib/python2.7/site-packages/south/hacks/django_1_0.py

Stats: 0 executed, 42 missed, 6 excluded, 59 ignored

1. """
2. Hacks for the Django 1.0/1.0.2 releases.
3. """
5. from django.conf import settings
6. from django.db.backends.creation import BaseDatabaseCreation
7. from django.db.models.loading import cache
8. from django.core import management
9. from django.core.management.commands.flush import Command as FlushCommand
10. from django.utils.datastructures import SortedDict
12. class SkipFlushCommand(FlushCommand):
13. def handle_noargs(self, **options):
14. # no-op to avoid calling flush
15. return
17. class Hacks:
19. def set_installed_apps(self, apps):
20. """
21. Sets Django's INSTALLED_APPS setting to be effectively the list passed in.
22. """
24. # Make sure it's a list.
25. apps = list(apps)
27. # Make sure it contains strings
28. if apps:
29. assert isinstance(apps[0], basestring), "The argument to set_installed_apps must be a list of strings."
31. # Monkeypatch in!
32. settings.INSTALLED_APPS, settings.OLD_INSTALLED_APPS = (
33. apps,
34. settings.INSTALLED_APPS,
35. )
36. self._redo_app_cache()
39. def reset_installed_apps(self):
40. """
41. Undoes the effect of set_installed_apps.
42. """
43. settings.INSTALLED_APPS = settings.OLD_INSTALLED_APPS
44. self._redo_app_cache()
47. def _redo_app_cache(self):
48. """
49. Used to repopulate AppCache after fiddling with INSTALLED_APPS.
50. """
51. cache.loaded = False
52. cache.handled = {}
53. cache.postponed = []
54. cache.app_store = SortedDict()
55. cache.app_models = SortedDict()
56. cache.app_errors = {}
57. cache._populate()
60. def clear_app_cache(self):
61. """
62. Clears the contents of AppCache to a blank state, so new models
63. from the ORM can be added.
64. """
65. self.old_app_models, cache.app_models = cache.app_models, {}
68. def unclear_app_cache(self):
69. """
70. Reversed the effects of clear_app_cache.
71. """
72. cache.app_models = self.old_app_models
73. cache._get_models_cache = {}
76. def repopulate_app_cache(self):
77. """
78. Rebuilds AppCache with the real model definitions.
79. """
80. cache._populate()
82. def store_app_cache_state(self):
83. self.stored_app_cache_state = dict(**cache.__dict__)
85. def restore_app_cache_state(self):
86. cache.__dict__ = self.stored_app_cache_state
88. def patch_flush_during_test_db_creation(self):
89. """
90. Patches BaseDatabaseCreation.create_test_db to not flush database
91. """
93. def patch(f):
94. def wrapper(*args, **kwargs):
95. # hold onto the original and replace flush command with a no-op
96. original_flush_command = management._commands['flush']
97. try:
98. management._commands['flush'] = SkipFlushCommand()
99. # run create_test_db
100. f(*args, **kwargs)
101. finally:
102. # unpatch flush back to the original
103. management._commands['flush'] = original_flush_command
104. return wrapper
106. BaseDatabaseCreation.create_test_db = patch(BaseDatabaseCreation.create_test_db)