1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-01-22 05:10:56 +00:00

core: introduce filesystem wrapper

This commit is contained in:
matejcik 2020-02-17 15:44:58 +01:00
parent b2084a19be
commit d08942be4a

46
core/src/trezor/sdcard.py Normal file
View File

@ -0,0 +1,46 @@
from trezorio import FatFS, sdcard
if False:
from typing import Any, Optional
class FilesystemWrapper:
_INSTANCE = None # type: Optional[FilesystemWrapper]
def __init__(self, mounted: bool = True) -> None:
self.fs = FatFS()
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 __enter__(self) -> "FatFS":
if self.counter <= 0:
self.counter = 0
sdcard.power_on()
if self.mounted:
self.fs.mount()
self.counter += 1
return self.fs
def __exit__(self, exc_type: Any, exc_val: Any, tb: Any) -> None:
self.counter -= 1
if self.counter <= 0:
self.counter = 0
if self.mounted:
self.fs.unmount()
sdcard.power_off()
FilesystemWrapper._INSTANCE = None
def get_filesystem(mounted: bool = True) -> FilesystemWrapper:
return FilesystemWrapper.get_instance(mounted=mounted)
is_present = sdcard.is_present