From 681866339c076fce572e66e756aa6c13078af4cc Mon Sep 17 00:00:00 2001 From: jesopo Date: Sun, 7 Oct 2018 11:24:13 +0100 Subject: [PATCH] Support multiple types of comments in utils.get_hashflags --- src/utils/__init__.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/src/utils/__init__.py b/src/utils/__init__.py index b762e3d2..a700dd95 100644 --- a/src/utils/__init__.py +++ b/src/utils/__init__.py @@ -128,15 +128,23 @@ def export(setting, value): return module return _export_func +COMMENT_TYPES = ["#", "//"] def get_hashflags(filename): hashflags = {} with io.open(filename, mode="r", encoding="utf8") as f: for line in f: line = line.strip("\n") - if not line.startswith("#"): + found = False + for comment_type in COMMENT_TYPES: + if line.startswith(comment_type): + line = line.replace(comment_type, "", 1).lstrip() + found = True + break + + if not found: break - elif line.startswith("#--"): - hashflag, sep, value = line[3:].partition(" ") + elif line.startswith("--"): + hashflag, sep, value = line[2:].partition(" ") hashflags[hashflag] = value if sep else None return hashflags.items()