home-assistant/core
1
0
Fork 1
mirror of https://github.com/home-assistant/core.git synced 2025-08-03 12:45:28 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add lock to token validity check (#117912)

Browse Source
This commit is contained in:
Joost Lekkerkerker
2024-05-22 20:10:23 +02:00
committed by GitHub
parent 55c8ef1c7b
commit 0c5296b38f
1 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
homeassistant/helpers/config_entry_oauth2_flow.py
View File

@@ -10,6 +10,7 @@ from __future__ import annotations
from abc import ABC, ABCMeta, abstractmethod from abc import ABC, ABCMeta, abstractmethod
import asyncio import asyncio
from asyncio import Lock
from collections.abc import Awaitable, Callable from collections.abc import Awaitable, Callable
from http import HTTPStatus from http import HTTPStatus
from json import JSONDecodeError from json import JSONDecodeError
@@ -506,6 +507,7 @@ class OAuth2Session:
self.hass = hass self.hass = hass
self.config_entry = config_entry self.config_entry = config_entry
self.implementation = implementation self.implementation = implementation
self._token_lock = Lock()
@property @property
def token(self) -> dict: def token(self) -> dict:
@@ -522,6 +524,7 @@ class OAuth2Session:
async def async_ensure_token_valid(self) -> None: async def async_ensure_token_valid(self) -> None:
"""Ensure that the current token is valid.""" """Ensure that the current token is valid."""
async with self._token_lock:
if self.valid_token: if self.valid_token:
return return