From 8c6c144ba289fd74f3c848f4849c36b96e460be7 Mon Sep 17 00:00:00 2001 From: Carlos Quintana Date: Mon, 23 May 2022 16:36:13 +0200 Subject: [PATCH] Fix global Partner instance --- app/proton/proton_callback_handler.py | 10 +++------- 1 file changed, 3 insertions(+), 7 deletions(-) diff --git a/app/proton/proton_callback_handler.py b/app/proton/proton_callback_handler.py index f3a5404d..bdb32671 100644 --- a/app/proton/proton_callback_handler.py +++ b/app/proton/proton_callback_handler.py @@ -11,7 +11,6 @@ from app.proton.proton_client import ProtonClient, ProtonUser from app.utils import random_string PROTON_PARTNER_NAME = "Proton" -_PROTON_PARTNER_ID: Optional[int] = None _PROTON_PARTNER: Optional[Partner] = None @@ -21,16 +20,13 @@ def get_proton_partner() -> Partner: partner = Partner.get_by(name=PROTON_PARTNER_NAME) if partner is None: raise ProtonPartnerNotSetUp + Session.expunge(partner) + _PROTON_PARTNER = partner return _PROTON_PARTNER def get_proton_partner_id() -> int: - global _PROTON_PARTNER - if _PROTON_PARTNER is None: - get_proton_partner() - return _PROTON_PARTNER.id - - return _PROTON_PARTNER_ID + return get_proton_partner().id class Action(enum.Enum):