1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-04 21:48:17 +00:00
trezor-firmware/core/src/trezor/sdcard.py

81 lines
2.0 KiB
Python
Raw Normal View History

try:
from trezorio import fatfs, sdcard
HAVE_SDCARD = True
is_present = sdcard.is_present
capacity = sdcard.capacity
except Exception:
HAVE_SDCARD = False
2020-02-17 14:44:58 +00:00
def is_present() -> bool:
return False
def capacity() -> int:
return 0
from typing import TYPE_CHECKING
if TYPE_CHECKING:
from typing import Any, Callable, TypeVar
from typing_extensions import ParamSpec
2020-02-26 18:25:22 +00:00
P = ParamSpec("P")
R = TypeVar("R")
2020-02-17 14:44:58 +00:00
class FilesystemWrapper:
_INSTANCE: "FilesystemWrapper | None" = None
2020-02-17 14:44:58 +00:00
def __init__(self, mounted: bool) -> None:
if not HAVE_SDCARD:
raise RuntimeError
2020-02-17 14:44:58 +00:00
self.mounted = mounted
self.counter = 0
@classmethod
def get_instance(cls, mounted: bool = True) -> "FilesystemWrapper":
if cls._INSTANCE is None:
cls._INSTANCE = cls(mounted=mounted)
if cls._INSTANCE.mounted is not mounted:
raise RuntimeError # cannot request mounted and non-mounted instance at the same time
return cls._INSTANCE
def _deinit_instance(self) -> None:
2020-02-26 18:25:22 +00:00
fatfs.unmount()
sdcard.power_off()
FilesystemWrapper._INSTANCE = None
2020-02-26 18:25:22 +00:00
def __enter__(self) -> None:
try:
if self.counter <= 0:
self.counter = 0
sdcard.power_on()
if self.mounted:
2020-02-26 18:25:22 +00:00
fatfs.mount()
self.counter += 1
except Exception:
self._deinit_instance()
raise
2020-02-17 14:44:58 +00:00
def __exit__(self, exc_type: Any, exc_val: Any, tb: Any) -> None:
self.counter -= 1
if self.counter <= 0:
self.counter = 0
self._deinit_instance()
2020-02-17 14:44:58 +00:00
2020-02-26 18:25:22 +00:00
def filesystem(mounted: bool = True) -> FilesystemWrapper:
2020-02-17 14:44:58 +00:00
return FilesystemWrapper.get_instance(mounted=mounted)
def with_filesystem(func: Callable[P, R]) -> Callable[P, R]:
def wrapped_func(*args: P.args, **kwargs: P.kwargs) -> R:
2020-02-26 18:25:22 +00:00
with filesystem():
return func(*args, **kwargs)
return wrapped_func