diff options
Diffstat (limited to 'django/contrib/postgres/apps.py')
| -rw-r--r-- | django/contrib/postgres/apps.py | 32 |
1 files changed, 18 insertions, 14 deletions
diff --git a/django/contrib/postgres/apps.py b/django/contrib/postgres/apps.py index b8ec85b7a4..d917201f05 100644 --- a/django/contrib/postgres/apps.py +++ b/django/contrib/postgres/apps.py @@ -1,6 +1,4 @@ -from psycopg2.extras import ( - DateRange, DateTimeRange, DateTimeTZRange, NumericRange, -) +from psycopg2.extras import DateRange, DateTimeRange, DateTimeTZRange, NumericRange from django.apps import AppConfig from django.core.signals import setting_changed @@ -25,7 +23,11 @@ def uninstall_if_needed(setting, value, enter, **kwargs): Undo the effects of PostgresConfig.ready() when django.contrib.postgres is "uninstalled" by override_settings(). """ - if not enter and setting == 'INSTALLED_APPS' and 'django.contrib.postgres' not in set(value): + if ( + not enter + and setting == "INSTALLED_APPS" + and "django.contrib.postgres" not in set(value) + ): connection_created.disconnect(register_type_handlers) CharField._unregister_lookup(Unaccent) TextField._unregister_lookup(Unaccent) @@ -43,21 +45,23 @@ def uninstall_if_needed(setting, value, enter, **kwargs): class PostgresConfig(AppConfig): - name = 'django.contrib.postgres' - verbose_name = _('PostgreSQL extensions') + name = "django.contrib.postgres" + verbose_name = _("PostgreSQL extensions") def ready(self): setting_changed.connect(uninstall_if_needed) # Connections may already exist before we are called. for conn in connections.all(): - if conn.vendor == 'postgresql': - conn.introspection.data_types_reverse.update({ - 3904: 'django.contrib.postgres.fields.IntegerRangeField', - 3906: 'django.contrib.postgres.fields.DecimalRangeField', - 3910: 'django.contrib.postgres.fields.DateTimeRangeField', - 3912: 'django.contrib.postgres.fields.DateRangeField', - 3926: 'django.contrib.postgres.fields.BigIntegerRangeField', - }) + if conn.vendor == "postgresql": + conn.introspection.data_types_reverse.update( + { + 3904: "django.contrib.postgres.fields.IntegerRangeField", + 3906: "django.contrib.postgres.fields.DecimalRangeField", + 3910: "django.contrib.postgres.fields.DateTimeRangeField", + 3912: "django.contrib.postgres.fields.DateRangeField", + 3926: "django.contrib.postgres.fields.BigIntegerRangeField", + } + ) if conn.connection is not None: register_type_handlers(conn) connection_created.connect(register_type_handlers) |