Fix LOAD DATA LOCAL INFILE write EOF packet on closed connection. (#1116)

methane · web-flow · commit bd3bd0149994 · 2023-05-25T00:33:50.000+09:00
Fix #989
diff --git a/.gitignore b/.gitignore
@@ -13,3 +13,4 @@
 /pymysql/tests/databases.json
 __pycache__
 Pipfile.lock
+pdm.lock
diff --git a/pymysql/connections.py b/pymysql/connections.py
@@ -1370,7 +1370,7 @@ def send_data(self):
         """Send data packets from the local file to the server"""
         if not self.connection._sock:
             raise err.InterfaceError(0, "")
-        conn = self.connection
+        conn: Connection = self.connection
 
         try:
             with open(self.filename, "rb") as open_file:
@@ -1388,5 +1388,6 @@ def send_data(self):
                 f"Can't find file '{self.filename}'",
             )
         finally:
-            # send the empty packet to signify we are done sending data
-            conn.write_packet(b"")
+            if not conn._closed:
+                # send the empty packet to signify we are done sending data
+                conn.write_packet(b"")
