support -l and -s arguments

pull/8/head
Nico 2015-11-04 22:11:42 +01:00
rodzic c4e58669e9
commit ddab6ff576
1 zmienionych plików z 9 dodań i 2 usunięć

Wyświetl plik

@ -74,10 +74,17 @@ def _get_message(arguments):
yield line.encode(encoding='utf-8')
except:
logger.error("%s Failed to read file '%s'" % (client.client_id, arguments['-f']))
if arguments['-s']:
if arguments['-l']:
import sys
for line in sys.stdin:
if line:
yield line.encode(encoding='utf-8')
if arguments['-s']:
import sys
message = bytearray()
for line in sys.stdin:
message.extend(line.encode(encoding='utf-8'))
yield message