Hopiu/django-cachalot
1
0
Fork 0
mirror of https://github.com/Hopiu/django-cachalot.git synced 2026-03-16 21:30:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

table invalidation condition enhanced (#213)

Browse source
This commit is contained in:
Jano Valaska 2022-02-25 00:31:04 +01:00 committed by GitHub
parent 9ba75c2eec
commit c8791fec4b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
cachalot/monkey_patch.py
View file

@ -1,3 +1,4 @@
import re
from collections.abc import Iterable
from functools import wraps
from time import time
@ -21,6 +22,13 @@ from .utils import (
WRITE_COMPILERS = (SQLInsertCompiler, SQLUpdateCompiler, SQLDeleteCompiler)
SQL_DATA_CHANGE_RE = re.compile(
'|'.join([
fr'(\W|\A){re.escape(keyword)}(\W|\Z)'
for keyword in ['update', 'insert', 'delete', 'alter', 'create', 'drop']
]),
flags=re.IGNORECASE,
)
def _unset_raw_connection(original):
def inner(compiler, *args, **kwargs):
@ -133,9 +141,7 @@ def _patch_cursor():
if isinstance(sql, bytes):
sql = sql.decode('utf-8')
sql = sql.lower()
if 'update' in sql or 'insert' in sql or 'delete' in sql \
or 'alter' in sql or 'create' in sql \
or 'drop' in sql:
if SQL_DATA_CHANGE_RE.search(sql):
tables = filter_cachable(
_get_tables_from_sql(connection, sql))
if tables: