summaryrefslogtreecommitdiff
path: root/lib/sqlalchemy/dialects/sqlite
diff options
context:
space:
mode:
authorMichael Gorven <michael@gorven.net>2022-11-29 18:36:19 -0500
committerMike Bayer <mike_mp@zzzcomputing.com>2022-11-30 15:13:22 -0500
commit07760011b5176be03c7811e9a45933b473b8b80b (patch)
treefe7c1958ef41ab47604e765cfcdbd35daaff7ef1 /lib/sqlalchemy/dialects/sqlite
parent7857a1de32169858367446d11089c34f8daee957 (diff)
downloadsqlalchemy-07760011b5176be03c7811e9a45933b473b8b80b.tar.gz
[sqlite] Reflect DEFERRABLE and INITIALLY options for foreign keys
Added support for the SQLite backend to reflect the "DEFERRABLE" and "INITIALLY" keywords which may be present on a foreign key construct. Pull request courtesy Michael Gorven. Fixes: #8903 Closes: #8904 Pull-request: https://github.com/sqlalchemy/sqlalchemy/pull/8904 Pull-request-sha: 52aa4cf77482c4051899e21bea75b9830e4c3efa Change-Id: I713906db1a458d8f1be39625841ca3bbc03ec835
Diffstat (limited to 'lib/sqlalchemy/dialects/sqlite')
-rw-r--r--lib/sqlalchemy/dialects/sqlite/base.py12
1 files changed, 11 insertions, 1 deletions
diff --git a/lib/sqlalchemy/dialects/sqlite/base.py b/lib/sqlalchemy/dialects/sqlite/base.py
index 11554fcc0..fb9a19c89 100644
--- a/lib/sqlalchemy/dialects/sqlite/base.py
+++ b/lib/sqlalchemy/dialects/sqlite/base.py
@@ -2462,6 +2462,8 @@ class SQLiteDialect(default.DefaultDialect):
r'REFERENCES +(?:(?:"(.+?)")|([a-z0-9_]+)) *\((.+?)\) *'
r"((?:ON (?:DELETE|UPDATE) "
r"(?:SET NULL|SET DEFAULT|CASCADE|RESTRICT|NO ACTION) *)*)"
+ r"((?:NOT +)?DEFERRABLE)?"
+ r"(?: +INITIALLY +(DEFERRED|IMMEDIATE))?"
)
for match in re.finditer(FK_PATTERN, table_data, re.I):
(
@@ -2471,7 +2473,9 @@ class SQLiteDialect(default.DefaultDialect):
referred_name,
referred_columns,
onupdatedelete,
- ) = match.group(1, 2, 3, 4, 5, 6)
+ deferrable,
+ initially,
+ ) = match.group(1, 2, 3, 4, 5, 6, 7, 8)
constrained_columns = list(
self._find_cols_in_sig(constrained_columns)
)
@@ -2493,6 +2497,12 @@ class SQLiteDialect(default.DefaultDialect):
onupdate = token[6:].strip()
if onupdate and onupdate != "NO ACTION":
options["onupdate"] = onupdate
+
+ if deferrable:
+ options["deferrable"] = "NOT" not in deferrable.upper()
+ if initially:
+ options["initially"] = initially.upper()
+
yield (
constraint_name,
constrained_columns,