luxferre/databankr
1
0
Fork 0
Code Actions Activity

Fixed stdio flows

Browse Source
This commit is contained in:
Luxferre
2024-05-21 20:24:06 +03:00
parent b035c1c12e
commit f08b73a186
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
databankr.py
+6 -2
View File
@@ -139,7 +139,9 @@ if __name__ == '__main__': # main app start
else:
infd = open(args.input_file, mode='rb')
indata = infd.read()
if infd != sys.stdin:
if infd == sys.stdin: # perform additional conversion
indata = indata.encode('utf-8')
else:
infd.close()
except:
print('Error reading the input data!')
@@ -171,10 +173,12 @@ if __name__ == '__main__': # main app start
try:
if args.output_file == '-':
outfd = sys.stdout
if type(outdata) == type(b''):
outdata = outdata.decode('utf-8')
outfd.write(outdata)
else:
outfd = open(args.output_file, mode='wb')
if type(outdata) == 'str':
if type(outdata) == type(''):
outdata = outdata.encode('utf-8')
outfd.write(outdata)
if outfd != sys.stdout: