mirror of https://github.com/tLDP/python-tldp
CLI-tool friendly handling of EPIPE and INT
And, correcting from the name of the Python class to the format name processed by the Python class (class.__name__ vs. class.formatname).
This commit is contained in:
parent
2f71c66b5a
commit
61d55a9f69
|
@ -6,6 +6,7 @@ from __future__ import absolute_import, division, print_function
|
||||||
import os
|
import os
|
||||||
import sys
|
import sys
|
||||||
import errno
|
import errno
|
||||||
|
import signal
|
||||||
import shutil
|
import shutil
|
||||||
import logging
|
import logging
|
||||||
import inspect
|
import inspect
|
||||||
|
@ -21,6 +22,13 @@ from tldp.utils import arg_isloglevel, arg_isdirectory
|
||||||
from tldp.utils import swapdirs, sameFilesystem
|
from tldp.utils import swapdirs, sameFilesystem
|
||||||
from tldp.doctypes.common import preamble, postamble
|
from tldp.doctypes.common import preamble, postamble
|
||||||
|
|
||||||
|
# -- Don't freak out with IOError when our STDOUT, handled with
|
||||||
|
# head, sed, awk, grep, etc; and, also deal with a user's ctrl-C
|
||||||
|
# the same way (i.e. no traceback, just stop)
|
||||||
|
#
|
||||||
|
signal.signal(signal.SIGPIPE, signal.SIG_DFL)
|
||||||
|
signal.signal(signal.SIGINT, signal.SIG_DFL)
|
||||||
|
|
||||||
logformat = '%(levelname)-9s %(name)s %(filename)s#%(lineno)s ' \
|
logformat = '%(levelname)-9s %(name)s %(filename)s#%(lineno)s ' \
|
||||||
+ '%(funcName)s %(message)s'
|
+ '%(funcName)s %(message)s'
|
||||||
logging.basicConfig(stream=sys.stderr, format=logformat, level=logging.ERROR)
|
logging.basicConfig(stream=sys.stderr, format=logformat, level=logging.ERROR)
|
||||||
|
@ -122,7 +130,7 @@ def summary(config, *args, **kwargs):
|
||||||
print('', 'By Document Type', '----------------', sep='\n', file=file)
|
print('', 'By Document Type', '----------------', sep='\n', file=file)
|
||||||
summarybytype = collections.defaultdict(list)
|
summarybytype = collections.defaultdict(list)
|
||||||
for doc in inv.source.values():
|
for doc in inv.source.values():
|
||||||
name = doc.doctype.__name__
|
name = doc.doctype.formatname
|
||||||
summarybytype[name].append(doc.stem)
|
summarybytype[name].append(doc.stem)
|
||||||
for doctype, docs in summarybytype.items():
|
for doctype, docs in summarybytype.items():
|
||||||
count = len(docs)
|
count = len(docs)
|
||||||
|
|
Loading…
Reference in New Issue