bin: use binary stdio; fixes msoulier#113

again possible to upload from stdin, download to stdout

Author: 9001

partftpy/TftpContexts.py

@@ -350,7 +350,7 @@ def __init__(
         if hasattr(input, "read"):
             self.fileobj = input
         elif input == "-":
-            self.fileobj = sys.stdin.buffer
+            self.fileobj = sys.stdin if PY2 else sys.stdin.buffer
         else:
             self.fileobj = open(input, "rb")
 
@@ -438,7 +438,7 @@ def __init__(
             self.filelike_fileobj = True
         # If the output filename is -, then use stdout
         elif output == "-":
-            self.fileobj = sys.stdout
+            self.fileobj = sys.stdout if PY2 else sys.stdout.buffer
             self.filelike_fileobj = True
         else:
             self.fileobj = open(output, "wb")

partftpy/TftpShared.py

@@ -2,6 +2,8 @@
 # vim: ts=4 sw=4 et ai:
 from __future__ import print_function, unicode_literals
 
+import sys
+
 """This module holds all objects shared by all other modules in partftpy."""
 
 
@@ -22,6 +24,15 @@
 # be skipped to simulate lost packets.
 
 
+PY2 = sys.version_info < (3,)
+
+
+try:
+    from typing import TYPE_CHECKING
+except:
+    TYPE_CHECKING = False
+
+
 def tftpassert(condition, msg):
     """This function is a simple utility that will check the condition
     passed for a false state. If it finds one, it throws a TftpException