Rietveld Code Review Tool
Help | Bug tracker | Discussion group | Source code

Unified Diff: sitescripts/subscriptions/combineSubscriptions.py

Issue 29345242: Noissue - Adapt quotes for compliance with our coding style in sitescripts (Closed)
Patch Set: Fixed raw string Created May 30, 2016, 8:47 a.m.
Use n/p to move between diff chunks; N/P to move between comments.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: sitescripts/subscriptions/combineSubscriptions.py
===================================================================
--- a/sitescripts/subscriptions/combineSubscriptions.py
+++ b/sitescripts/subscriptions/combineSubscriptions.py
@@ -28,9 +28,9 @@
import tempfile
from getopt import getopt, GetoptError
-accepted_extensions = set([".txt"])
-ignore = set(["Apache.txt", "CC-BY-SA.txt", "GPL.txt", "MPL.txt"])
-verbatim = set(["COPYING"])
+accepted_extensions = set(['.txt'])
+ignore = set(['Apache.txt', 'CC-BY-SA.txt', 'GPL.txt', 'MPL.txt'])
+verbatim = set(['COPYING'])
def combine_subscriptions(sources, target_dir, timeout=30, tempdir=None):
@@ -38,26 +38,26 @@
os.makedirs(target_dir, 0755)
def save_file(filename, data):
- handle = tempfile.NamedTemporaryFile(mode="wb", dir=tempdir, delete=False)
- handle.write(data.encode("utf-8"))
+ handle = tempfile.NamedTemporaryFile(mode='wb', dir=tempdir, delete=False)
+ handle.write(data.encode('utf-8'))
handle.close()
- if hasattr(os, "chmod"):
+ if hasattr(os, 'chmod'):
os.chmod(handle.name, 0644)
try:
- subprocess.check_output(["7za", "a", "-tgzip", "-mx=9", "-bd", "-mpass=5", handle.name + ".gz", handle.name])
+ subprocess.check_output(['7za', 'a', '-tgzip', '-mx=9', '-bd', '-mpass=5', handle.name + '.gz', handle.name])
except:
- print >>sys.stderr, "Failed to compress file %s. Please ensure that p7zip is installed on the system." % handle.name
+ print >>sys.stderr, 'Failed to compress file %s. Please ensure that p7zip is installed on the system.' % handle.name
path = os.path.join(target_dir, filename)
os.rename(handle.name, path)
- os.rename(handle.name + ".gz", path + ".gz")
+ os.rename(handle.name + '.gz', path + '.gz')
known = set()
for source_name, source in sources.iteritems():
for filename in source.list_top_level_files():
- if filename in ignore or filename.startswith("."):
+ if filename in ignore or filename.startswith('.'):
continue
if filename in verbatim:
process_verbatim_file(source, save_file, filename)
@@ -70,13 +70,13 @@
print >>sys.stderr, 'Error processing subscription file "%s"' % filename
traceback.print_exc()
print >>sys.stderr
- known.add(os.path.splitext(filename)[0] + ".tpl")
- known.add(os.path.splitext(filename)[0] + ".tpl.gz")
+ known.add(os.path.splitext(filename)[0] + '.tpl')
+ known.add(os.path.splitext(filename)[0] + '.tpl.gz')
known.add(filename)
- known.add(filename + ".gz")
+ known.add(filename + '.gz')
for filename in os.listdir(target_dir):
- if filename.startswith("."):
+ if filename.startswith('.'):
continue
if not filename in known:
os.remove(os.path.join(target_dir, filename))
@@ -90,19 +90,19 @@
source = sources[source_name]
lines = source.read_file(filename).splitlines()
- header = ""
+ header = ''
if len(lines) > 0:
header = lines.pop(0)
- if not re.search(r"\[Adblock(?:\s*Plus\s*([\d\.]+)?)?\]", header, re.I):
- raise Exception("This is not a valid Adblock Plus subscription file.")
+ if not re.search(r'\[Adblock(?:\s*Plus\s*([\d\.]+)?)?\]', header, re.I):
+ raise Exception('This is not a valid Adblock Plus subscription file.')
lines = resolve_includes(source_name, sources, lines, timeout)
- seen = set(["checksum", "version"])
+ seen = set(['checksum', 'version'])
def check_line(line):
- if line == "":
+ if line == '':
return False
- match = re.search(r"^\s*!\s*(Redirect|Homepage|Title|Checksum|Version|Expires)\s*:", line, re.M | re.I)
+ match = re.search(r'^\s*!\s*(Redirect|Homepage|Title|Checksum|Version|Expires)\s*:', line, re.M | re.I)
if not match:
return True
key = match.group(1).lower()
@@ -112,29 +112,29 @@
return True
lines = filter(check_line, lines)
- write_tpl(save_file, os.path.splitext(filename)[0] + ".tpl", lines)
+ write_tpl(save_file, os.path.splitext(filename)[0] + '.tpl', lines)
- lines.insert(0, "! Version: %s" % time.strftime("%Y%m%d%H%M", time.gmtime()))
+ lines.insert(0, '! Version: %s' % time.strftime('%Y%m%d%H%M', time.gmtime()))
checksum = hashlib.md5()
- checksum.update("\n".join([header] + lines).encode("utf-8"))
- lines.insert(0, "! Checksum: %s" % base64.b64encode(checksum.digest()).rstrip("="))
+ checksum.update('\n'.join([header] + lines).encode('utf-8'))
+ lines.insert(0, '! Checksum: %s' % base64.b64encode(checksum.digest()).rstrip('='))
lines.insert(0, header)
- save_file(filename, "\n".join(lines))
+ save_file(filename, '\n'.join(lines))
def resolve_includes(source_name, sources, lines, timeout, level=0):
if level > 5:
- raise Exception("There are too many nested includes, which is probably the result of a circular reference somewhere.")
+ raise Exception('There are too many nested includes, which is probably the result of a circular reference somewhere.')
result = []
for line in lines:
- match = re.search(r"^\s*%include\s+(.*)%\s*$", line)
+ match = re.search(r'^\s*%include\s+(.*)%\s*$', line)
if match:
filename = match.group(1)
newlines = None
- if re.match(r"^https?://", filename):
- result.append("! *** Fetched from: %s ***" % filename)
+ if re.match(r'^https?://', filename):
+ result.append('! *** Fetched from: %s ***' % filename)
for i in range(3):
try:
@@ -151,14 +151,14 @@
# We should really get the charset from the headers rather than assuming
# that it is UTF-8. However, some of the Google Code mirrors are
# misconfigured and will return ISO-8859-1 as charset instead of UTF-8.
- newlines = data.decode("utf-8").splitlines()
- newlines = filter(lambda l: not re.search(r"^\s*!\s*(Redirect|Homepage|Title|Version|Expires)\s*:", l, re.M | re.I), newlines)
+ newlines = data.decode('utf-8').splitlines()
+ newlines = filter(lambda l: not re.search(r'^\s*!\s*(Redirect|Homepage|Title|Version|Expires)\s*:', l, re.M | re.I), newlines)
else:
- result.append("! *** %s ***" % filename)
+ result.append('! *** %s ***' % filename)
include_source = source_name
- if ":" in filename:
- include_source, filename = filename.split(":", 1)
+ if ':' in filename:
+ include_source, filename = filename.split(':', 1)
if not include_source in sources:
raise Exception('Cannot include file from repository "%s", this repository is unknown' % include_source)
@@ -166,34 +166,34 @@
newlines = source.read_file(filename).splitlines()
newlines = resolve_includes(include_source, sources, newlines, timeout, level + 1)
- if len(newlines) and re.search(r"\[Adblock(?:\s*Plus\s*([\d\.]+)?)?\]", newlines[0], re.I):
+ if len(newlines) and re.search(r'\[Adblock(?:\s*Plus\s*([\d\.]+)?)?\]', newlines[0], re.I):
del newlines[0]
result.extend(newlines)
else:
- if line.find("%timestamp%") >= 0:
+ if line.find('%timestamp%') >= 0:
if level == 0:
- line = line.replace("%timestamp%", time.strftime("%d %b %Y %H:%M UTC", time.gmtime()))
+ line = line.replace('%timestamp%', time.strftime('%d %b %Y %H:%M UTC', time.gmtime()))
else:
- line = ""
+ line = ''
result.append(line)
return result
def write_tpl(save_file, filename, lines):
result = []
- result.append("msFilterList")
+ result.append('msFilterList')
for line in lines:
- if re.search(r"^\s*!", line):
+ if re.search(r'^\s*!', line):
# This is a comment. Handle "Expires" comment in a special way, keep the rest.
- match = re.search(r"^\s*!\s*Expires\s*:\s*(\d+)\s*(h)?", line, re.I)
+ match = re.search(r'^\s*!\s*Expires\s*:\s*(\d+)\s*(h)?', line, re.I)
if match:
interval = int(match.group(1))
if match.group(2):
interval = int(interval / 24)
- result.append(": Expires=%i" % interval)
+ result.append(': Expires=%i' % interval)
else:
- result.append(re.sub(r"^\s*!", "#", re.sub(r"--!$", "--#", line)))
- elif line.find("#") >= 0:
+ result.append(re.sub(r'^\s*!', '#', re.sub(r'--!$', '--#', line)))
+ elif line.find('#') >= 0:
# Element hiding rules are not supported in MSIE, drop them
pass
else:
@@ -201,37 +201,37 @@
origline = line
is_exception = False
- if line.startswith("@@"):
+ if line.startswith('@@'):
is_exception = True
line = line[2:]
has_unsupported = False
requires_script = False
- match = re.search(r"^(.*?)\$(.*)", line)
+ match = re.search(r'^(.*?)\$(.*)', line)
if match:
# This rule has options, check whether any of them are important
line = match.group(1)
- options = match.group(2).replace("_", "-").lower().split(",")
+ options = match.group(2).replace('_', '-').lower().split(',')
# Remove first-party only exceptions, we will allow an ad server everywhere otherwise
- if is_exception and "~third-party" in options:
+ if is_exception and '~third-party' in options:
has_unsupported = True
# A number of options are not supported in MSIE but can be safely ignored, remove them
- options = filter(lambda o: not o in ("", "third-party", "~third-party", "match-case", "~match-case", "~other", "~donottrack"), options)
+ options = filter(lambda o: not o in ('', 'third-party', '~third-party', 'match-case', '~match-case', '~other', '~donottrack'), options)
# Also ignore domain negation of whitelists
if is_exception:
- options = filter(lambda o: not o.startswith("domain=~"), options)
+ options = filter(lambda o: not o.startswith('domain=~'), options)
- unsupported = filter(lambda o: o in ("other", "elemhide"), options)
+ unsupported = filter(lambda o: o in ('other', 'elemhide'), options)
if unsupported and len(unsupported) == len(options):
# The rule only applies to types that are not supported in MSIE
has_unsupported = True
- elif "donottrack" in options:
+ elif 'donottrack' in options:
# Do-Not-Track rules have to be removed even if $donottrack is combined with other options
has_unsupported = True
- elif "script" in options and len(options) == len(unsupported) + 1:
+ elif 'script' in options and len(options) == len(unsupported) + 1:
# Mark rules that only apply to scripts for approximate conversion
requires_script = True
elif len(options) > 0:
@@ -240,62 +240,62 @@
# remaining options to avoid potential false positives. Other rules
# simply aren't included in the TPL file.
if is_exception:
- has_unsupported = any([o.startswith("domain=") for o in options])
+ has_unsupported = any([o.startswith('domain=') for o in options])
else:
has_unsupported = True
if has_unsupported:
# Do not include filters with unsupported options
- result.append("# " + origline)
+ result.append('# ' + origline)
else:
- line = line.replace("^", "/") # Assume that separator placeholders mean slashes
+ line = line.replace('^', '/') # Assume that separator placeholders mean slashes
# Try to extract domain info
domain = None
- match = re.search(r"^(\|\||\|\w+://)([^*:/]+)(:\d+)?(/.*)", line)
+ match = re.search(r'^(\|\||\|\w+://)([^*:/]+)(:\d+)?(/.*)', line)
if match:
domain = match.group(2)
line = match.group(4)
else:
# No domain info, remove anchors at the rule start
- line = re.sub(r"^\|\|", "http://", line)
- line = re.sub(r"^\|", "", line)
+ line = re.sub(r'^\|\|', 'http://', line)
+ line = re.sub(r'^\|', '', line)
# Remove anchors at the rule end
- line = re.sub(r"\|$", "", line)
+ line = re.sub(r'\|$', '', line)
# Remove unnecessary asterisks at the ends of lines
- line = re.sub(r"\*$", "", line)
+ line = re.sub(r'\*$', '', line)
# Emulate $script by appending *.js to the rule
if requires_script:
- line += "*.js"
- if line.startswith("/*"):
+ line += '*.js'
+ if line.startswith('/*'):
line = line[2:]
if domain:
- line = "%sd %s %s" % ("+" if is_exception else "-", domain, line)
- line = re.sub(r"\s+/$", "", line)
+ line = '%sd %s %s' % ('+' if is_exception else '-', domain, line)
+ line = re.sub(r'\s+/$', '', line)
result.append(line)
elif is_exception:
# Exception rules without domains are unsupported
- result.append("# " + origline)
+ result.append('# ' + origline)
else:
- result.append("- " + line)
- save_file(filename, "\n".join(result) + "\n")
+ result.append('- ' + line)
+ save_file(filename, '\n'.join(result) + '\n')
class FileSource:
def __init__(self, dir):
self._dir = dir
- if os.path.exists(os.path.join(dir, ".hg")):
+ if os.path.exists(os.path.join(dir, '.hg')):
# This is a Mercurial repository, try updating
- subprocess.call(["hg", "-q", "-R", dir, "pull", "--update"])
+ subprocess.call(['hg', '-q', '-R', dir, 'pull', '--update'])
def get_path(self, filename):
- return os.path.join(self._dir, *filename.split("/"))
+ return os.path.join(self._dir, *filename.split('/'))
def read_file(self, filename):
path = self.get_path(filename)
- if os.path.relpath(path, self._dir).startswith("."):
- raise Exception("Attempt to access a file outside the repository")
- with codecs.open(path, "rb", encoding="utf-8") as handle:
+ if os.path.relpath(path, self._dir).startswith('.'):
+ raise Exception('Attempt to access a file outside the repository')
+ with codecs.open(path, 'rb', encoding='utf-8') as handle:
return handle.read()
def list_top_level_files(self):
@@ -306,38 +306,38 @@
def usage():
- print """Usage: %s source_name=source_dir ... [output_dir]
+ print '''Usage: %s source_name=source_dir ... [output_dir]
Options:
-h --help Print this message and exit
-t seconds --timeout=seconds Timeout when fetching remote subscriptions
-""" % os.path.basename(sys.argv[0])
+''' % os.path.basename(sys.argv[0])
-if __name__ == "__main__":
+if __name__ == '__main__':
try:
- opts, args = getopt(sys.argv[1:], "ht:", ["help", "timeout="])
+ opts, args = getopt(sys.argv[1:], 'ht:', ['help', 'timeout='])
except GetoptError, e:
print str(e)
usage()
sys.exit(2)
- target_dir = "subscriptions"
+ target_dir = 'subscriptions'
sources = {}
for arg in args:
- if "=" in arg:
- source_name, source_dir = arg.split("=", 1)
+ if '=' in arg:
+ source_name, source_dir = arg.split('=', 1)
sources[source_name] = FileSource(source_dir)
else:
target_dir = arg
if not sources:
- sources[""] = FileSource(".")
+ sources[''] = FileSource('.')
timeout = 30
for option, value in opts:
- if option in ("-h", "--help"):
+ if option in ('-h', '--help'):
usage()
sys.exit()
- elif option in ("-t", "--timeout"):
+ elif option in ('-t', '--timeout'):
timeout = int(value)
combine_subscriptions(sources, target_dir, timeout)

Powered by Google App Engine
This is Rietveld