changeset e7f0e72148f6 in modules/account_payment_stripe:4.6
details:
https://hg.tryton.org/modules/account_payment_stripe?cmd=changeset;node=e7f0e72148f6
description:
Lock records when processing
issue8012
review60511002
(grafted from d404a49a74d2d906ce2b2d3bf14978b5d5c51427)
diffstat:
payment.py | 21 +++++++++++++++++++++
1 files changed, 21 insertions(+), 0 deletions(-)
diffs (45 lines):
diff -r e811095d3911 -r e7f0e72148f6 payment.py
--- a/payment.py Fri Jan 05 00:06:03 2018 +0100
+++ b/payment.py Wed Feb 06 22:51:18 2019 +0100
@@ -313,6 +313,7 @@
('stripe_charge_id', '=', None),
])
for payment in payments:
+ cls.__lock([payment])
try:
charge = stripe.Charge.create(**payment._charge_parameters())
except (stripe.error.RateLimitError,
@@ -372,6 +373,7 @@
or payment.stripe_captured
or payment.state != 'processing'):
continue
+ cls.__lock([payment])
try:
charge = stripe.Charge.retrieve(
payment.stripe_charge_id,
@@ -405,6 +407,25 @@
'amount': self.stripe_amount,
}
+ @classmethod
+ def __lock(cls, records):
+ from trytond.tools import grouped_slice, reduce_ids
+ from sql import Literal, For
+ transaction = Transaction()
+ database = transaction.database
+ connection = transaction.connection
+ table = cls.__table__()
+
+ if database.has_select_for():
+ for sub_records in grouped_slice(records):
+ where = reduce_ids(table.id, sub_records)
+ query = table.select(
+ Literal(1), where=where, for_=For('UPDATE', nowait=True))
+ with connection.cursor() as cursor:
+ cursor.execute(*query)
+ else:
+ database.lock(connection, cls._table)
+
class Account(ModelSQL, ModelView):
"Stripe Account"
