Simplify _to_bytes

This version is smaller and faster.
CourtBouillon · Jan 5, 2024 · 322a602 · 322a602
1 parent 9ff462e
commit 322a602
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/pydyf/__init__.py b/pydyf/__init__.py
@@ -16,15 +16,13 @@ def _to_bytes(item):
     """Convert item to bytes."""
     if isinstance(item, bytes):
         return item
-    elif isinstance(item, Object):
-        return item.data
     elif isinstance(item, float):
         if item.is_integer():
-            return f'{int(item):d}'.encode('ascii')
+            return str(int(item)).encode('ascii')
         else:
             return f'{item:f}'.rstrip('0').encode('ascii')
-    elif isinstance(item, int):
-        return f'{item:d}'.encode('ascii')
+    elif isinstance(item, Object):
+        return item.data
     return str(item).encode('ascii')