support stdin
This commit is contained in:
parent
d51386c77a
commit
a7d649871c
1 changed files with 4 additions and 1 deletions
|
@ -1,9 +1,12 @@
|
|||
#!/usr/bin/env python
|
||||
from __future__ import print_function, unicode_literals
|
||||
import argparse
|
||||
import sys
|
||||
import cchardet
|
||||
|
||||
|
||||
PY2 = sys.version_info.major == 2
|
||||
|
||||
def read_chunks(f, chunk_size):
|
||||
chunk = f.read(chunk_size)
|
||||
while chunk:
|
||||
|
@ -13,7 +16,7 @@ def read_chunks(f, chunk_size):
|
|||
|
||||
def main():
|
||||
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))
|
||||
args = parser.parse_args()
|
||||
|
||||
|
|
Loading…
Reference in a new issue