Skip to content

Commit

Permalink
Combining get() methods in Variables.
Browse files Browse the repository at this point in the history
  • Loading branch information
cecily_carver committed Jan 28, 2025
1 parent 233c21b commit 538e7b2
Showing 1 changed file with 6 additions and 9 deletions.
15 changes: 6 additions & 9 deletions mysql_mimic/variables.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down Expand Up @@ -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)]

Expand Down

0 comments on commit 538e7b2

Please sign in to comment.