From cd09674c6a264bbdbd0421c18f017ad1c79d0ef5 Mon Sep 17 00:00:00 2001 From: "Karl O. Pinc" Date: Wed, 20 Sep 2023 11:16:42 -0500 Subject: [PATCH] Reformat with black --- doc/extract_view.py | 20 ++++++++++++-------- 1 file changed, 12 insertions(+), 8 deletions(-) diff --git a/doc/extract_view.py b/doc/extract_view.py index 98cee2a..45d2c97 100755 --- a/doc/extract_view.py +++ b/doc/extract_view.py @@ -14,9 +14,9 @@ # You should have received a copy of the GNU Affero General Public License # along with this program. If not, see . # -'''Read from stdin and (heuristically) extract the SQL that makes up +"""Read from stdin and (heuristically) extract the SQL that makes up the body of the view. -''' +""" # Karl O. Pinc # import re @@ -26,10 +26,11 @@ CREATE_VIEW = re.compile(r"\s*CREATE OR REPLACE VIEW \S+ AS\s*$") GRANT = re.compile(r"\s*GRANT ") SQL_COMMENT = re.compile(r"\s*(?:--.*)|$") + def get_body(fd): - '''Return a list of every line between (exclusive) the line matching + """Return a list of every line between (exclusive) the line matching CREATE_VIEW and the line matching GRANT. - ''' + """ body = [] found = False for line in fd: @@ -41,12 +42,13 @@ def get_body(fd): found = CREATE_VIEW.match(line) is not None return body + def cleanup(body): - '''Mutate the list that is body, with leading and trailing lines that + """Mutate the list that is body, with leading and trailing lines that are empty or consist only of whitespace removed, and all trailing lines that match SQL_COMMENT removed. - ''' - while body[0].strip == '': + """ + while body[0].strip == "": del body[0] for pos in range(len(body) - 1, -1, -1): @@ -54,12 +56,14 @@ def cleanup(body): break del body[pos] + def main(): - with open('/dev/stdin', 'r') as fd: + with open("/dev/stdin", "r") as fd: body = get_body(fd) cleanup(body) for line in body: print(line, end="") + if __name__ == "__main__": main() -- 2.34.1