transit-python 0.8.321__cp310-abi3-win_amd64.whl

transit/write_handlers.py

@@ -0,0 +1,353 @@
+## Copyright 2014 Cognitect. All Rights Reserved.
+## Copyright 2026 Vendekagon Labs LLC.
+##
+## Licensed under the Apache License, Version 2.0 (the "License");
+## you may not use this file except in compliance with the License.
+## You may obtain a copy of the License at
+##
+##      http://www.apache.org/licenses/LICENSE-2.0
+##
+## Unless required by applicable law or agreed to in writing, software
+## distributed under the License is distributed on an "AS-IS" BASIS,
+## WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+## See the License for the specific language governing permissions and
+## limitations under the License.
+
+import datetime
+import struct
+import uuid
+from decimal import Decimal
+from math import isnan
+
+from transit.class_hash import ClassDict
+from transit.transit_types import Keyword, Symbol, URI, frozendict, TaggedValue, Link, Boolean
+
+MAX_INT = 2**63 - 1
+MIN_INT = -2**63
+
+## This file contains Write Handlers - all the top-level objects used when
+## writing Transit data.  These object must all be immutable and pickleable.
+
+
+class TaggedMap:
+    def __init__(self, tag, rep, str):
+        self._tag = tag
+        self._rep = rep
+        self._str = str
+
+    def tag(self):
+        return self._tag
+
+    def rep(self):
+        return self._rep
+
+    def string_rep(self):
+        return self._str
+
+
+class NoneHandler:
+    @staticmethod
+    def tag(_):
+        return '_'
+
+    @staticmethod
+    def rep(_):
+        return None
+
+    @staticmethod
+    def string_rep(n):
+        return None
+
+
+class IntHandler:
+    @staticmethod
+    def tag(n):
+        if MIN_INT <= n <= MAX_INT:
+            return "i"
+        return "n"
+
+    @staticmethod
+    def rep(n):
+        return n
+
+    @staticmethod
+    def string_rep(n):
+        return str(n)
+
+
+class BigIntHandler:
+    @staticmethod
+    def tag(_):
+        return "n"
+
+    @staticmethod
+    def rep(n):
+        return str(n)
+
+    @staticmethod
+    def string_rep(n):
+        return str(n)
+
+
+class BigDecimalHandler:
+    @staticmethod
+    def tag(_):
+        return "f"
+
+    @staticmethod
+    def rep(n):
+        return str(n)
+
+    @staticmethod
+    def string_rep(n):
+        return str(n)
+
+
+class FloatHandler:
+    @staticmethod
+    def tag(f):
+        return "z" if isnan(f) or f in (float('Inf'), float('-Inf')) else "d"
+
+    @staticmethod
+    def rep(f):
+        if isnan(f):
+            return "NaN"
+        if f == float('Inf'):
+            return "INF"
+        if f == float("-Inf"):
+            return "-INF"
+        return f
+
+    @staticmethod
+    def string_rep(f):
+        return str(f)
+
+
+class StringHandler:
+    @staticmethod
+    def tag(s):
+        return 's'
+
+    @staticmethod
+    def rep(s):
+        return s
+
+    @staticmethod
+    def string_rep(s):
+        return s
+
+
+class BooleanHandler:
+    @staticmethod
+    def tag(_):
+        return '?'
+
+    @staticmethod
+    def rep(b):
+        return bool(b)
+
+    @staticmethod
+    def string_rep(b):
+        return 't' if b else 'f'
+
+
+class ArrayHandler:
+    @staticmethod
+    def tag(a):
+        return 'array'
+
+    @staticmethod
+    def rep(a):
+        return a
+
+    @staticmethod
+    def string_rep(a):
+        return None
+
+
+class MapHandler:
+    @staticmethod
+    def tag(m):
+        return 'map'
+
+    @staticmethod
+    def rep(m):
+        return m
+
+    @staticmethod
+    def string_rep(m):
+        return None
+
+
+class KeywordHandler:
+    @staticmethod
+    def tag(k):
+        return ':'
+
+    @staticmethod
+    def rep(k):
+        return str(k)
+
+    @staticmethod
+    def string_rep(k):
+        return str(k)
+
+
+class SymbolHandler:
+    @staticmethod
+    def tag(s):
+        return '$'
+
+    @staticmethod
+    def rep(s):
+        return str(s)
+
+    @staticmethod
+    def string_rep(s):
+        return str(s)
+
+
+class UuidHandler:
+    @staticmethod
+    def tag(_):
+        return "u"
+
+    @staticmethod
+    def rep(u):
+        return struct.unpack('>qq', u.bytes)
+
+    @staticmethod
+    def string_rep(u):
+        return str(u)
+
+
+class UriHandler:
+    @staticmethod
+    def tag(_):
+        return "r"
+
+    @staticmethod
+    def rep(u):
+        return u.rep
+
+    @staticmethod
+    def string_rep(u):
+        return u.rep
+
+
+class DateTimeHandler:
+    epoch = datetime.datetime(1970, 1, 1, tzinfo=datetime.timezone.utc)
+
+    @staticmethod
+    def tag(_):
+        return "m"
+
+    @staticmethod
+    def rep(d):
+        return (d - DateTimeHandler.epoch) // datetime.timedelta(milliseconds=1)
+
+    @staticmethod
+    def verbose_handler():
+        return VerboseDateTimeHandler
+
+    @staticmethod
+    def string_rep(d):
+        return str(DateTimeHandler.rep(d))
+
+
+def format_rfc3339(d):
+    """d as an RFC 3339 UTC time to the millisecond, as transit-java writes it."""
+    ms = DateTimeHandler.rep(d)
+    t = DateTimeHandler.epoch + datetime.timedelta(milliseconds=ms)
+    return "%04d-%02d-%02dT%02d:%02d:%02d.%03dZ" % (t.year, t.month, t.day, t.hour, t.minute, t.second,
+                                                     ms % 1000)
+
+
+class VerboseDateTimeHandler:
+    @staticmethod
+    def tag(_):
+        return "t"
+
+    @staticmethod
+    def rep(d):
+        return format_rfc3339(d)
+
+    @staticmethod
+    def string_rep(d):
+        return format_rfc3339(d)
+
+
+class SetHandler:
+    @staticmethod
+    def tag(_):
+        return "set"
+
+    @staticmethod
+    def rep(s):
+        return TaggedMap("array", tuple(s), None)
+
+    @staticmethod
+    def string_rep(_):
+        return None
+
+
+class TaggedValueHandler:
+    @staticmethod
+    def tag(tv):
+        return tv.tag
+
+    @staticmethod
+    def rep(tv):
+        return tv.rep
+
+    @staticmethod
+    def string_rep(_):
+        return None
+
+
+class LinkHandler:
+    @staticmethod
+    def tag(_):
+        return "link"
+
+    @staticmethod
+    def rep(l):
+        return l.as_map
+
+    @staticmethod
+    def string_rep(_):
+        return None
+
+
+class WriteHandler(ClassDict):
+    """This is the top-level handler for encoding/writing Python data into
+    Transit data, based on its type.
+    The Handler itself is a dispatch map, that resolves on full type/object
+    inheritance.
+
+    These handlers can be overriden during the creation of a Transit Writer.
+    """
+
+    def __init__(self):
+        super(WriteHandler, self).__init__()
+        self[type(None)] = NoneHandler
+        self[bool] = BooleanHandler
+        self[Boolean] = BooleanHandler
+        self[str] = StringHandler
+        self[list] = ArrayHandler
+        self[tuple] = ArrayHandler
+        self[int] = IntHandler
+        self[float] = FloatHandler
+        self[Keyword] = KeywordHandler
+        self[Symbol] = SymbolHandler
+        self[uuid.UUID] = UuidHandler
+        self[URI] = UriHandler
+        self[datetime.datetime] = DateTimeHandler
+        self[set] = SetHandler
+        self[frozenset] = SetHandler
+        self[TaggedMap] = TaggedMap
+        self[dict] = MapHandler
+        self[frozendict] = MapHandler
+        self[TaggedValue] = TaggedValueHandler
+        self[Link] = LinkHandler
+        self[Decimal] = BigDecimalHandler