Skip to content

Commit 2b341f2

Browse files
committed
[IMP] util/pg: deduplicate "{parallel_filter}"
As usual, it reduces chances of stupid mistakes (e.g. typos).
1 parent e800de2 commit 2b341f2

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

src/util/pg.py

+4-3
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@
4545

4646
ON_DELETE_ACTIONS = frozenset(("SET NULL", "CASCADE", "RESTRICT", "NO ACTION", "SET DEFAULT"))
4747
MAX_BUCKETS = int(os.getenv("MAX_BUCKETS", "150000"))
48+
PARALLEL_FILTER = "{parallel_filter}"
4849

4950

5051
class PGRegexp(str):
@@ -252,9 +253,9 @@ def explode_query_range(cr, query, table, alias=None, bucket_size=10000, prefix=
252253

253254
alias = alias or table
254255

255-
if "{parallel_filter}" not in query:
256+
if PARALLEL_FILTER not in query:
256257
sep_kw = " AND " if re.search(r"\sWHERE\s", query, re.M | re.I) else " WHERE "
257-
query += sep_kw + "{parallel_filter}"
258+
query += sep_kw + PARALLEL_FILTER
258259

259260
cr.execute(format_query(cr, "SELECT min(id), max(id) FROM {}", table))
260261
min_id, max_id = cr.fetchone()
@@ -300,7 +301,7 @@ def explode_query_range(cr, query, table, alias=None, bucket_size=10000, prefix=
300301
return [query.replace("{parallel_filter}", parallel_filter)]
301302

302303
parallel_filter = "{alias}.id BETWEEN %(lower-bound)s AND %(upper-bound)s".format(alias=alias)
303-
query = query.replace("%", "%%").replace("{parallel_filter}", parallel_filter)
304+
query = query.replace("%", "%%").replace(PARALLEL_FILTER, parallel_filter)
304305
return [
305306
cr.mogrify(query, {"lower-bound": ids[i], "upper-bound": ids[i + 1] - 1}).decode() for i in range(len(ids) - 1)
306307
]

0 commit comments

Comments
 (0)