123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148 |
- import operator
- from django.db import DataError, InterfaceError
- from django.db.backends.base.features import BaseDatabaseFeatures
- from django.db.backends.postgresql.psycopg_any import is_psycopg3
- from django.utils.functional import cached_property
- class DatabaseFeatures(BaseDatabaseFeatures):
- minimum_database_version = (12,)
- allows_group_by_selected_pks = True
- can_return_columns_from_insert = True
- can_return_rows_from_bulk_insert = True
- has_real_datatype = True
- has_native_uuid_field = True
- has_native_duration_field = True
- has_native_json_field = True
- can_defer_constraint_checks = True
- has_select_for_update = True
- has_select_for_update_nowait = True
- has_select_for_update_of = True
- has_select_for_update_skip_locked = True
- has_select_for_no_key_update = True
- can_release_savepoints = True
- supports_comments = True
- supports_tablespaces = True
- supports_transactions = True
- can_introspect_materialized_views = True
- can_distinct_on_fields = True
- can_rollback_ddl = True
- schema_editor_uses_clientside_param_binding = True
- supports_combined_alters = True
- nulls_order_largest = True
- closed_cursor_error_class = InterfaceError
- greatest_least_ignores_nulls = True
- can_clone_databases = True
- supports_temporal_subtraction = True
- supports_slicing_ordering_in_compound = True
- create_test_procedure_without_params_sql = """
- CREATE FUNCTION test_procedure () RETURNS void AS $$
- DECLARE
- V_I INTEGER;
- BEGIN
- V_I := 1;
- END;
- $$ LANGUAGE plpgsql;"""
- create_test_procedure_with_int_param_sql = """
- CREATE FUNCTION test_procedure (P_I INTEGER) RETURNS void AS $$
- DECLARE
- V_I INTEGER;
- BEGIN
- V_I := P_I;
- END;
- $$ LANGUAGE plpgsql;"""
- create_test_table_with_composite_primary_key = """
- CREATE TABLE test_table_composite_pk (
- column_1 INTEGER NOT NULL,
- column_2 INTEGER NOT NULL,
- PRIMARY KEY(column_1, column_2)
- )
- """
- requires_casted_case_in_updates = True
- supports_over_clause = True
- only_supports_unbounded_with_preceding_and_following = True
- supports_aggregate_filter_clause = True
- supported_explain_formats = {"JSON", "TEXT", "XML", "YAML"}
- supports_deferrable_unique_constraints = True
- has_json_operators = True
- json_key_contains_list_matching_requires_list = True
- supports_update_conflicts = True
- supports_update_conflicts_with_target = True
- supports_covering_indexes = True
- supports_stored_generated_columns = True
- supports_virtual_generated_columns = False
- can_rename_index = True
- test_collations = {
- "deterministic": "C",
- "non_default": "sv-x-icu",
- "swedish_ci": "sv-x-icu",
- "virtual": "sv-x-icu",
- }
- test_now_utc_template = "STATEMENT_TIMESTAMP() AT TIME ZONE 'UTC'"
- insert_test_table_with_defaults = "INSERT INTO {} DEFAULT VALUES"
- django_test_skips = {
- "opclasses are PostgreSQL only.": {
- "indexes.tests.SchemaIndexesNotPostgreSQLTests."
- "test_create_index_ignores_opclasses",
- },
- "PostgreSQL requires casting to text.": {
- "lookup.tests.LookupTests.test_textfield_exact_null",
- },
- }
- @cached_property
- def django_test_expected_failures(self):
- expected_failures = set()
- if self.uses_server_side_binding:
- expected_failures.update(
- {
- # Parameters passed to expressions in SELECT and GROUP BY
- # clauses are not recognized as the same values when using
- # server-side binding cursors (#34255).
- "aggregation.tests.AggregateTestCase."
- "test_group_by_nested_expression_with_params",
- }
- )
- return expected_failures
- @cached_property
- def uses_server_side_binding(self):
- options = self.connection.settings_dict["OPTIONS"]
- return is_psycopg3 and options.get("server_side_binding") is True
- @cached_property
- def prohibits_null_characters_in_text_exception(self):
- if is_psycopg3:
- return DataError, "PostgreSQL text fields cannot contain NUL (0x00) bytes"
- else:
- return ValueError, "A string literal cannot contain NUL (0x00) characters."
- @cached_property
- def introspected_field_types(self):
- return {
- **super().introspected_field_types,
- "PositiveBigIntegerField": "BigIntegerField",
- "PositiveIntegerField": "IntegerField",
- "PositiveSmallIntegerField": "SmallIntegerField",
- }
- @cached_property
- def is_postgresql_13(self):
- return self.connection.pg_version >= 130000
- @cached_property
- def is_postgresql_14(self):
- return self.connection.pg_version >= 140000
- @cached_property
- def is_postgresql_15(self):
- return self.connection.pg_version >= 150000
- has_bit_xor = property(operator.attrgetter("is_postgresql_14"))
- supports_covering_spgist_indexes = property(operator.attrgetter("is_postgresql_14"))
- supports_unlimited_charfield = True
- supports_nulls_distinct_unique_constraints = property(
- operator.attrgetter("is_postgresql_15")
- )
|