File: config.py

package info (click to toggle)
python-django-pgtransaction 2.1.0-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 504 kB
  • sloc: python: 551; makefile: 101; sh: 9
file content (26 lines) | stat: -rw-r--r-- 731 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
from django.conf import settings
from django.core.exceptions import ImproperlyConfigured

try:
    import psycopg.errors as psycopg_errors
except ImportError:
    import psycopg2.errors as psycopg_errors
except Exception as exc:  # pragma: no cover
    raise ImproperlyConfigured("Error loading psycopg2 or psycopg module") from exc


def retry_exceptions():
    """The default errors caught when retrying.

    Note that these must be psycopg errors.
    """
    return getattr(
        settings,
        "PGTRANSACTION_RETRY_EXCEPTIONS",
        (psycopg_errors.SerializationFailure, psycopg_errors.DeadlockDetected),
    )


def retry():
    """The default retry amount"""
    return getattr(settings, "PGTRANSACTION_RETRY", 0)