| Index: abp/filters/diff_script.py |
| =================================================================== |
| new file mode 100644 |
| --- /dev/null |
| +++ b/abp/filters/diff_script.py |
| @@ -0,0 +1,65 @@ |
| +# This file is part of Adblock Plus <https://adblockplus.org/>, |
| +# Copyright (C) 2006-present eyeo GmbH |
| +# |
| +# Adblock Plus is free software: you can redistribute it and/or modify |
| +# it under the terms of the GNU General Public License version 3 as |
| +# published by the Free Software Foundation. |
| +# |
| +# Adblock Plus 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 General Public License for more details. |
| +# |
| +# You should have received a copy of the GNU General Public License |
| +# along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. |
| + |
| +"""Command line script for rendering Adblock Plus filter list diffs.""" |
| + |
| +import argparse |
| +import io |
| +import sys |
| + |
| +from collections import namedtuple |
| + |
| +from .sources import NotFound |
| +from .renderer import render_diff |
| +__all__ = ['main'] |
| + |
| + |
| +def parse_args(): |
| + parser = argparse.ArgumentParser(description='Render a filter list diff.') |
| + parser.add_argument( |
| + 'base', help='the older filter list that needs to be updated', |
| + default='-', nargs='?') |
| + parser.add_argument( |
| + 'latest', help='the most recent version of the filter list', |
| + default='-', nargs='?') |
| + parser.add_argument( |
| + 'outfile', help='output file for filter list diff', |
| + default='-', nargs='?') |
| + return parser.parse_args() |
| + |
| + |
| +def main(args_in=None): |
| + """Entry point for the diff rendering script (diffrender).""" |
| + if args_in: |
| + Parse = namedtuple('Test', 'base, latest, outfile') |
| + args = Parse(args_in['base'], args_in['latest'], args_in['outfile']) |
| + else: |
| + args = parse_args() |
| + |
| + base = open(args.base, 'r') |
| + latest = open(args.latest, 'r') |
|
Sebastian Noack
2018/09/18 16:56:12
Any reason you don't use io.open(..., encoding="ut
Sebastian Noack
2018/09/18 16:56:12
Please make sure all files are explicitly closed.
rhowell
2018/09/18 17:23:33
Done.
|
| + |
| + try: |
| + lines = render_diff(base, latest) |
| + if args.outfile == '-': |
| + for line in lines: |
| + sys.stdout.write(line + '\n') |
| + else: |
| + open(args.outfile, 'w') |
|
rhowell
2018/09/18 16:01:54
This feels hacky..
Sebastian Noack
2018/09/18 16:56:12
Well, this is redundant. Why did you put it here?
rhowell
2018/09/18 17:23:33
Wow, you're right. I swore I was getting an error
Sebastian Noack
2018/09/18 19:13:18
Both, open(..., 'w') and io.open(..., 'w') create
|
| + with io.open(args.outfile, 'w', encoding='utf-8') as out_fp: |
| + for line in lines: |
| + out_fp.write(line + '\n') |
| + except NotFound as exc: |
| + sys.exit(exc) |