| Left: | ||
| Right: |
| OLD | NEW |
|---|---|
| (Empty) | |
| 1 # This file is part of Adblock Plus <https://adblockplus.org/>, | |
| 2 # Copyright (C) 2006-present eyeo GmbH | |
| 3 # | |
| 4 # Adblock Plus is free software: you can redistribute it and/or modify | |
| 5 # it under the terms of the GNU General Public License version 3 as | |
| 6 # published by the Free Software Foundation. | |
| 7 # | |
| 8 # Adblock Plus is distributed in the hope that it will be useful, | |
| 9 # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
| 10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
| 11 # GNU General Public License for more details. | |
| 12 # | |
| 13 # You should have received a copy of the GNU General Public License | |
| 14 # along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. | |
| 15 | |
| 16 """Command line script for rendering Adblock Plus filter list diffs.""" | |
| 17 | |
| 18 import argparse | |
| 19 import io | |
| 20 import sys | |
| 21 | |
| 22 from collections import namedtuple | |
| 23 | |
| 24 from .sources import NotFound | |
| 25 from .renderer import render_diff | |
| 26 __all__ = ['main'] | |
| 27 | |
| 28 | |
| 29 def parse_args(): | |
| 30 parser = argparse.ArgumentParser(description='Render a filter list diff.') | |
| 31 parser.add_argument( | |
| 32 'base', help='the older filter list that needs to be updated', | |
| 33 default='-', nargs='?') | |
| 34 parser.add_argument( | |
| 35 'latest', help='the most recent version of the filter list', | |
| 36 default='-', nargs='?') | |
| 37 parser.add_argument( | |
| 38 'outfile', help='output file for filter list diff', | |
| 39 default='-', nargs='?') | |
| 40 return parser.parse_args() | |
| 41 | |
| 42 | |
| 43 def main(args_in=None): | |
| 44 """Entry point for the diff rendering script (diffrender).""" | |
| 45 if args_in: | |
| 46 Parse = namedtuple('Test', 'base, latest, outfile') | |
| 47 args = Parse(args_in['base'], args_in['latest'], args_in['outfile']) | |
| 48 else: | |
| 49 args = parse_args() | |
| 50 | |
| 51 base = open(args.base, 'r') | |
| 52 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.
| |
| 53 | |
| 54 try: | |
| 55 lines = render_diff(base, latest) | |
| 56 if args.outfile == '-': | |
| 57 for line in lines: | |
| 58 sys.stdout.write(line + '\n') | |
| 59 else: | |
| 60 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
| |
| 61 with io.open(args.outfile, 'w', encoding='utf-8') as out_fp: | |
| 62 for line in lines: | |
| 63 out_fp.write(line + '\n') | |
| 64 except NotFound as exc: | |
| 65 sys.exit(exc) | |
| OLD | NEW |