Skip to content

Commit e3b6af6

Browse files
committed
replaced decimal.localcontext usage with compat version (bugfix JarryShaw#139)
1 parent 9af2733 commit e3b6af6

File tree

4 files changed

+8
-8
lines changed

4 files changed

+8
-8
lines changed

pcapkit/protocols/data/misc/pcapng.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88
from pcapkit.corekit.infoclass import info_final
99
from pcapkit.corekit.multidict import MultiDict
1010
from pcapkit.protocols.data.data import Data
11+
from pcapkit.utilities.compat import localcontext
1112

1213
__all__ = [
1314
'PCAPNG',
@@ -480,7 +481,7 @@ class SimplePacketBlock(PCAPNG):
480481

481482
def __post_init__(self) -> 'None':
482483
"""Post-initialization handling."""
483-
with decimal.localcontext(prec=64):
484+
with localcontext(prec=64):
484485
self.__update__(
485486
interface_id=0,
486487
timestamp=datetime.datetime.fromtimestamp(0, datetime.timezone.utc),

pcapkit/protocols/misc/pcap/frame.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@
3434
from pcapkit.protocols.data.misc.pcap.frame import FrameInfo as Data_FrameInfo
3535
from pcapkit.protocols.protocol import Protocol
3636
from pcapkit.protocols.schema.misc.pcap.frame import Frame as Schema_Frame
37+
from pcapkit.utilities.compat import localcontext
3738
from pcapkit.utilities.exceptions import UnsupportedCall, stacklevel
3839
from pcapkit.utilities.warnings import ProtocolWarning, RegistryWarning, warn
3940

@@ -209,7 +210,7 @@ def read(self, length: 'Optional[int]' = None, *, _read: 'bool' = True, **kwargs
209210
_ilen = schema.incl_len
210211
_olen = schema.orig_len
211212

212-
with decimal.localcontext(prec=64):
213+
with localcontext(prec=64):
213214
if self._nsec:
214215
_epch = _tsss + decimal.Decimal(_tsus) / 1_000_000_000
215216
else:
@@ -409,7 +410,7 @@ def _make_timestamp(self, timestamp: 'Optional[float | Decimal | dt_type | int]'
409410
Second and microsecond/nanosecond value of timestamp.
410411
411412
"""
412-
with decimal.localcontext(prec=64):
413+
with localcontext(prec=64):
413414
if timestamp is None:
414415
if py37 and nanosecond:
415416
timestamp = decimal.Decimal(time.time_ns()) / 1_000_000_000

pcapkit/protocols/misc/pcapng.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -169,7 +169,7 @@
169169
from pcapkit.protocols.schema.misc.pcapng import WireGuardKeyLog as Schema_WireGuardKeyLog
170170
from pcapkit.protocols.schema.misc.pcapng import ZigBeeAPSKey as Schema_ZigBeeAPSKey
171171
from pcapkit.protocols.schema.misc.pcapng import ZigBeeNWKKey as Schema_ZigBeeNWKKey
172-
from pcapkit.utilities.compat import StrEnum
172+
from pcapkit.utilities.compat import StrEnum, localcontext
173173
from pcapkit.utilities.exceptions import ProtocolError, UnsupportedCall, stacklevel
174174
from pcapkit.utilities.warnings import (AttributeWarning, DeprecatedFormatWarning, ProtocolWarning,
175175
RegistryWarning, warn)
@@ -1189,7 +1189,7 @@ def _read_timestamp(self, timestamp_high: 'int', timestamp_low: 'int', *,
11891189
tzone = self._get_timezone(interface_id)
11901190

11911191
timestamp_raw = (timestamp_high << 32) | timestamp_low
1192-
with decimal.localcontext(prec=64):
1192+
with localcontext(prec=64):
11931193
timestamp_epoch = decimal.Decimal(timestamp_raw) / self._get_resolution(interface_id) + \
11941194
self._get_offset(interface_id)
11951195
ts_decimal = timestamp_epoch + decimal.Decimal(
@@ -1234,7 +1234,7 @@ def _make_timestamp(self, timestamp: 'Optional[float | Decimal | dt_type | int]'
12341234
based on the given offset and timezone conversion.
12351235
12361236
"""
1237-
with decimal.localcontext(prec=64):
1237+
with localcontext(prec=64):
12381238
if timestamp is None:
12391239
if py37 and self.nanosecond:
12401240
timestamp = decimal.Decimal(time.time_ns()) / 1_000_000_000

pcapkit/protocols/schema/transport/tcp.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,8 +41,6 @@
4141
from ipaddress import IPv4Address, IPv6Address
4242
from typing import Any, Optional
4343

44-
from typing_extensions import Literal
45-
4644
from pcapkit.corekit.fields.field import _Field as Field
4745
from pcapkit.protocols.protocol import Protocol
4846

0 commit comments

Comments
 (0)