1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-24 15:28:10 +00:00
trezor-firmware/src/trezor/utils.py
2016-09-27 17:23:59 +02:00

32 lines
769 B
Python

import sys
from TrezorUtils import Utils
_utils = Utils()
type_gen = type((lambda: (yield))())
def memaccess(address, length):
return _utils.memaccess(address, length)
def unimport_func(func):
def inner(*args, **kwargs):
mods = set(sys.modules)
try:
ret = func(*args, **kwargs)
finally:
for to_remove in set(sys.modules) - mods:
del sys.modules[to_remove]
return ret
return inner
def unimport_gen(gen):
def inner(*args, **kwargs):
mods = set(sys.modules)
try:
ret = yield from gen(*args, **kwargs)
finally:
for to_remove in set(sys.modules) - mods:
del sys.modules[to_remove]
return ret
return inner