support stdin

This commit is contained in:
PyYoshi 2017-05-15 09:51:37 +09:00
parent d51386c77a
commit a7d649871c

View file

@ -1,9 +1,12 @@
#!/usr/bin/env python #!/usr/bin/env python
from __future__ import print_function, unicode_literals from __future__ import print_function, unicode_literals
import argparse import argparse
import sys
import cchardet import cchardet
PY2 = sys.version_info.major == 2
def read_chunks(f, chunk_size): def read_chunks(f, chunk_size):
chunk = f.read(chunk_size) chunk = f.read(chunk_size)
while chunk: while chunk:
@ -13,7 +16,7 @@ def read_chunks(f, chunk_size):
def main(): def main():
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument('files', nargs='+', help="Files to detect encoding of", type=argparse.FileType('rb')) parser.add_argument('files', nargs='*', help="Files to detect encoding of", type=argparse.FileType('rb'), default=[sys.stdin if PY2 else sys.stdin.buffer])
parser.add_argument('--chunk-size', type=int, default=(256 * 1024)) parser.add_argument('--chunk-size', type=int, default=(256 * 1024))
args = parser.parse_args() args = parser.parse_args()