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
|
#!/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()
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue