Changeset View
Changeset View
Standalone View
Standalone View
python_modules/dagster/dagster/seven/compat/pendulum.py
from contextlib import contextmanager | from contextlib import contextmanager | ||||
import packaging.version | import packaging.version | ||||
import pendulum | import pendulum | ||||
_IS_PENDULUM_2 = ( | _IS_PENDULUM_2 = ( | ||||
hasattr(pendulum, "__version__") and packaging.version.parse(pendulum.__version__).major == 2 | hasattr(pendulum, "__version__") | ||||
and getattr(packaging.version.parse(pendulum.__version__), "major") == 2 | |||||
) | ) | ||||
@contextmanager | @contextmanager | ||||
def mock_pendulum_timezone(override_timezone): | def mock_pendulum_timezone(override_timezone): | ||||
if _IS_PENDULUM_2: | if _IS_PENDULUM_2: | ||||
with pendulum.tz.test_local_timezone( # pylint: disable=no-member | with pendulum.tz.test_local_timezone( # pylint: disable=no-member | ||||
pendulum.tz.timezone(override_timezone) # pylint: disable=no-member | pendulum.tz.timezone(override_timezone) # pylint: disable=no-member | ||||
Show All 35 Lines |