Don't loop forever at the end of input SQL file

akaihola · May 5, 2024 · 10e8fb8 · 10e8fb8
1 parent 1394845
commit 10e8fb8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pgtricks/pg_dump_splitsort.py b/pgtricks/pg_dump_splitsort.py
@@ -173,6 +173,8 @@ def new_output(filename: str) -> IO[str]:
     with open(sql_filepath) as sql_file:
         while True:
             line = sql_file.readline()
+            if not line:
+                break
             if sorted_data_lines is None:
                 if line in ('\n', '--\n'):
                     buf.append(line)