38 lines
824 B
38 lines
824 B
import typing
|
|
|
|
|
|
class NullRegistry:
|
|
"""A NullRegistry that masquerades as a pint.UnitRegistry."""
|
|
|
|
def __init__(self):
|
|
"""Initialize a null registry."""
|
|
|
|
def __getattr__(self, item: typing.Any) -> int:
|
|
"""Return a Scalar 1 to simulate a unit."""
|
|
return 1
|
|
|
|
def __bool__(self):
|
|
"""Return False since a NullRegistry is not a pint.UnitRegistry."""
|
|
return False
|
|
|
|
def define(self, *args, **kwargs):
|
|
"""Pretend to add unit to the registry."""
|
|
|
|
|
|
def _build_unit_registry():
|
|
try:
|
|
import pint
|
|
|
|
registry = pint.UnitRegistry()
|
|
registry.define('FPS = 1 * hertz')
|
|
|
|
pint.set_application_registry(registry)
|
|
return registry
|
|
except ModuleNotFoundError:
|
|
pass
|
|
|
|
return NullRegistry()
|
|
|
|
|
|
units = _build_unit_registry()
|