SWAMPy

✎ CARVE FLESH

EDITING: _diffcommand.py

from __future__ import absolute_import

import optparse
import sys
import re
import os
from .diff import htmldiff

description = """\
"""

parser = optparse.OptionParser(
    usage="%prog [OPTIONS] FILE1 FILE2\n"
    "%prog --annotate [OPTIONS] INFO1 FILE1 INFO2 FILE2 ...",
    description=description,
    )

parser.add_option(
    '-o', '--output',
    metavar="FILE",
    dest="output",
    default="-",
    help="File to write the difference to",
    )

parser.add_option(
    '-a', '--annotation',
    action="store_true",
    dest="annotation",
    help="Do an annotation")

def main(args=None):
    if args is None:
        args = sys.argv[1:]
    options, args = parser.parse_args(args)
    if options.annotation:
        return annotate(options, args)
    if len(args) != 2:
        print('Error: you must give two files')
        parser.print_help()
        sys.exit(1)
    file1, file2 = args
    input1 = read_file(file1)
    input2 = read_file(file2)
    body1 = split_body(input1)[1]
    pre, body2, post = split_body(input2)
    result = htmldiff(body1, body2)
    result = pre + result + post
    if options.output == '-':
        if not result.endswith('\n'):
            result += '\n'
        sys.stdout.write(result)
    else:
        f = open(options.output, 'wb')
        f.write(result)
        f.close()

def read_file(filename):
    if filename == '-':
        c = sys.stdin.read()
    elif not os.path.exists(filename):
        raise OSError(
            "Input file %s does not exist" % filename)
    else:
        f = open(filename, 'rb')
        c = f.read()
        f.close()
    return c

body_start_re = re.compile(
    r"<body.*?>", re.I|re.S)
body_end_re = re.compile(
    r"</body.*?>", re.I|re.S)
    
def split_body(html):
    pre = post = ''
    match = body_start_re.search(html)
    if match:
        pre = html[:match.end()]
        html = html[match.end():]
    match = body_end_re.search(html)
    if match:
        post = html[match.start():]
        html = html[:match.start()]
    return pre, html, post

def annotate(options, args):
    print("Not yet implemented")
    sys.exit(1)

CANCEL

Name	Type	Size	Modified	Actions
↩ ..	DIR	—	—
📁 __pycache__/	DIR	—	2026-07-18 12:50
📄 __init__.py	PY	63.4 KB	2018-03-13 20:13	EDIT
📄 _diffcommand.py	PY	2.1 KB	2017-09-17 07:43	EDIT
📄 _html5builder.py	PY	3.2 KB	2012-09-28 19:13	EDIT
📄 _setmixin.py	PY	1.1 KB	2016-05-05 07:08	EDIT
📄 builder.py	PY	4.2 KB	2011-09-25 16:58	EDIT
📄 clean.cpython-36m-x86_64-linux-gnu.so	SO	276.8 KB	2022-04-18 17:59	EDIT
📄 clean.py	PY	26.4 KB	2022-04-18 17:58	EDIT
📄 defs.py	PY	4.1 KB	2022-04-18 17:58	EDIT
📄 diff.cpython-36m-x86_64-linux-gnu.so	SO	371.5 KB	2022-04-18 17:59	EDIT
📄 diff.py	PY	29.8 KB	2017-05-01 14:11	EDIT
📄 ElementSoup.py	PY	319 B	2011-09-25 16:58	EDIT
📄 formfill.py	PY	9.5 KB	2017-06-03 16:47	EDIT
📄 html5parser.py	PY	8.4 KB	2017-09-17 07:43	EDIT
📄 soupparser.py	PY	10 KB	2017-09-17 07:43	EDIT
📄 usedoctest.py	PY	249 B	2011-09-25 16:58	EDIT