Postgres: Extend regex in load script

Author: szarnyasg

postgres/scripts/load.py

@@ -14,6 +14,7 @@ def __init__(self):
         self.user = os.environ.get("POSTGRES_USER", "postgres")
         self.password = os.environ.get("POSTGRES_PASSWORD", "mysecretpassword")
 
+
     def run_script(self, conn, filename):
 
         with open(filename, "r") as f:
@@ -26,7 +27,7 @@ def run_script(self, conn, filename):
                     if query.isspace():
                         continue
 
-                    sql_statement = re.findall(r"^((CREATE|INSERT|DROP|DELETE|SELECT|COPY) [A-Za-z0-9_ ]*)", query, re.MULTILINE|re.IGNORECASE)
+                    sql_statement = re.findall(r"^((CREATE|INSERT|DROP|DELETE|SELECT|COPY|ALTER) [A-Za-z0-9_ ]*)", query, re.MULTILINE|re.IGNORECASE)
                     print(f"{sql_statement[0][0].strip()} ...")
                     start = time.time()
                     cur = conn.cursor()
@@ -45,6 +46,7 @@ def vacuum(self, conn):
         conn.cursor().execute("ANALYZE")
         conn.autocommit=False
 
+
     def main(self):
         with psycopg.connect(
             dbname=self.database,