From 538e7b2e74e2b0a7b2ea953cf11a96556d2777e0 Mon Sep 17 00:00:00 2001 From: cecily_carver Date: Tue, 28 Jan 2025 13:20:35 -0800 Subject: [PATCH] Combining get() methods in Variables. --- mysql_mimic/variables.py | 15 ++++++--------- 1 file changed, 6 insertions(+), 9 deletions(-) diff --git a/mysql_mimic/variables.py b/mysql_mimic/variables.py index aad34d0..021c6c7 100644 --- a/mysql_mimic/variables.py +++ b/mysql_mimic/variables.py @@ -67,7 +67,12 @@ def __init__(self) -> None: self._values: dict[str, Any] = {} def __getitem__(self, key: str) -> Any | None: - return self._get_variable(key) + key = key.lower() + if key in self._values: + return self._values[key] + _, default, _ = self.get_schema(key) + + return default def __setitem__(self, key: str, value: Any) -> None: return self.set(key, value) @@ -103,14 +108,6 @@ def set(self, name: str, value: Any, force: bool = False) -> None: else: self._values[name] = type_(value) - def _get_variable(self, name: str) -> Any | None: - name = name.lower() - if name in self._values: - return self._values[name] - _, default, _ = self.get_schema(name) - - return default - def list(self) -> list[tuple[str, Any]]: return [(name, self.get(name)) for name in sorted(self.schema)]