From 1aa92eb252179c32d9f2e294188d734c3c5ec238 Mon Sep 17 00:00:00 2001 From: "Karl O. Pinc" Date: Tue, 30 Jan 2024 11:19:49 -0600 Subject: [PATCH] Raise an exception if we don't find the view code --- doc/extract_view.py | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) diff --git a/doc/extract_view.py b/doc/extract_view.py index 9752da9..e1f9bcf 100755 --- a/doc/extract_view.py +++ b/doc/extract_view.py @@ -27,6 +27,17 @@ GRANT = re.compile(r"\s*GRANT ") SQL_COMMENT = re.compile(r"\s*(?:--.*)|$") +class Error(Exception): + pass + + +class ViewNotFound(Error): + def __init__(self): + super().__init__( + f'Cannot find a view when searching for the regexp "{VIEW_RE}"' + ) + + def get_body(fd): """Return a list of every line between (exclusive) the line matching VIEW_RE and the line matching GRANT. @@ -39,7 +50,11 @@ def get_body(fd): break body.append(line) else: - found = VIEW_RE.match(line) is not None + if VIEW_RE.match(line) is not None: + body.append(line) + found = True + if not body: + raise ViewNotFound() return body -- 2.34.1