src/hwp5/proc/summaryinfo.py

# -*- coding: utf-8 -*-
#
#   pyhwp : hwp file format parser in python
#   Copyright (C) 2010-2023 mete0r <https://github.com/mete0r>
#
#   This program is free software: you can redistribute it and/or modify
#   it under the terms of the GNU Affero General Public License as published by
#   the Free Software Foundation, either version 3 of the License, or
#   (at your option) any later version.
#
#   This program is distributed in the hope that it will be useful,
#   but WITHOUT ANY WARRANTY; without even the implied warranty of
#   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#   GNU Affero General Public License for more details.
#
#   You should have received a copy of the GNU Affero General Public License
#   along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
from __future__ import absolute_import
from __future__ import print_function
from __future__ import unicode_literals
import sys

from ..filestructure import Hwp5File
from ..summaryinfo import HwpSummaryInfoTextFormatter


PY2 = sys.version_info.major == 2


def main(args):
    if PY2:
        output_fp = sys.stdout
    else:
        output_fp = sys.stdout.buffer

    formatter = HwpSummaryInfoTextFormatter()
    hwpfile = Hwp5File(args.hwp5file)
    try:
        for textline in formatter.formatTextLines(hwpfile.summaryinfo):
            line = textline.encode('utf-8')
            output_fp.write(line)
            output_fp.write(b'\n')
    finally:
        hwpfile.close()


def summaryinfo_argparser(subparsers, _):
    parser = subparsers.add_parser(
        'summaryinfo',
        help=_(
            'Print summary informations of .hwp files.'
        ),
        description=_(
            'Print the summary information of <hwp5file>.'
        ),
    )
    parser.add_argument(
        'hwp5file',
        metavar='<hwp5file>',
        help=_('.hwp file to analyze'),
    )
    parser.set_defaults(func=main)
    return parser