/bangkokhotel/lib/python2.5/site-packages/django/test/signals.py

https://bitbucket.org/luisrodriguez/bangkokhotel · Python · 23 lines · 18 code · 5 blank · 0 comment · 9 complexity · 411c902e1ce6970900aadc8dee439f77 MD5 · raw file

  1. from django.conf import settings
  2. from django.db import connections
  3. from django.dispatch import Signal
  4. template_rendered = Signal(providing_args=["template", "context"])
  5. setting_changed = Signal(providing_args=["setting", "value"])
  6. def update_connections_time_zone(**kwargs):
  7. if kwargs['setting'] == 'USE_TZ' and settings.TIME_ZONE != 'UTC':
  8. USE_TZ, TIME_ZONE = kwargs['value'], settings.TIME_ZONE
  9. elif kwargs['setting'] == 'TIME_ZONE' and not settings.USE_TZ:
  10. USE_TZ, TIME_ZONE = settings.USE_TZ, kwargs['value']
  11. else: # no need to change the database connnections' time zones
  12. return
  13. tz = 'UTC' if USE_TZ else TIME_ZONE
  14. for conn in connections.all():
  15. tz_sql = conn.ops.set_time_zone_sql()
  16. if tz_sql:
  17. conn.cursor().execute(tz_sql, [tz])
  18. setting_changed.connect(update_connections_time_zone)