using rstrip

GoogleCloudPlatform · Oct 10, 2022 · 3775b51 · 3775b51
1 parent 2160649
commit 3775b51
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/data_validation/validation_builder.py b/data_validation/validation_builder.py
@@ -450,9 +450,7 @@ def get_query_from_file(self, filename):
         try:
             file = open(filename, "r")
             query = file.read()
-            query_semicolon = query[query.rfind(";") :]
-            if set(query_semicolon).issubset({";", " ", "\n"}):
-                query = query[0 : query.rfind(";")]
+            query = query.rstrip(";\n")
         except IOError:
             logging.warning("Cannot read query file: ", filename)