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