django.contrib.contenttypes.management: 34 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/django/contrib/contenttypes/management.py

Stats: 0 executed, 31 missed, 3 excluded, 45 ignored

1. from django.contrib.contenttypes.models import ContentType
2. from django.db.models import get_apps, get_models, signals
3. from django.utils.encoding import smart_unicode
5. def update_contenttypes(app, created_models, verbosity=2, **kwargs):
6. """
7. Creates content types for models in the given app, removing any model
8. entries that no longer have a matching model class.
9. """
10. ContentType.objects.clear_cache()
11. app_models = get_models(app)
12. if not app_models:
13. return
14. # They all have the same app_label, get the first one.
15. app_label = app_models[0]._meta.app_label
16. app_models = dict(
17. (model._meta.object_name.lower(), model)
18. for model in app_models
19. )
20. # Get all the content types
21. content_types = dict(
22. (ct.model, ct)
23. for ct in ContentType.objects.filter(app_label=app_label)
24. )
25. to_remove = [
26. ct
27. for (model_name, ct) in content_types.iteritems()
28. if model_name not in app_models
29. ]
31. cts = ContentType.objects.bulk_create([
32. ContentType(
33. name=smart_unicode(model._meta.verbose_name_raw),
34. app_label=app_label,
35. model=model_name,
36. )
37. for (model_name, model) in app_models.iteritems()
38. if model_name not in content_types
39. ])
40. if verbosity >= 2:
41. for ct in cts:
42. print "Adding content type '%s | %s'" % (ct.app_label, ct.model)
44. # Confirm that the content type is stale before deletion.
45. if to_remove:
46. if kwargs.get('interactive', False):
47. content_type_display = '\n'.join([
48. ' %s | %s' % (ct.app_label, ct.model)
49. for ct in to_remove
50. ])
51. ok_to_delete = raw_input("""The following content types are stale and need to be deleted:
53. %s
55. Any objects related to these content types by a foreign key will also
56. be deleted. Are you sure you want to delete these content types?
57. If you're unsure, answer 'no'.
59. Type 'yes' to continue, or 'no' to cancel: """ % content_type_display)
60. else:
61. ok_to_delete = False
63. if ok_to_delete == 'yes':
64. for ct in to_remove:
65. if verbosity >= 2:
66. print "Deleting stale content type '%s | %s'" % (ct.app_label, ct.model)
67. ct.delete()
68. else:
69. if verbosity >= 2:
70. print "Stale content types remain."
72. def update_all_contenttypes(verbosity=2, **kwargs):
73. for app in get_apps():
74. update_contenttypes(app, None, verbosity, **kwargs)
76. signals.post_syncdb.connect(update_contenttypes)
78. if __name__ == "__main__":
79. update_all_contenttypes()