check local_infile option

methane · methane · commit b5e17cee46e0 · 2017-02-14T21:27:57.000+09:00
diff --git a/pymysql/connections.py b/pymysql/connections.py
@@ -600,7 +600,8 @@ def __init__(self, host=None, user=None, password="",
         if compress or named_pipe:
             raise NotImplementedError("compress and named_pipe arguments are not supported")
 
-        if local_infile:
+        self._local_infile = bool(local_infile)
+        if self._local_infile:
             client_flag |= CLIENT.LOCAL_FILES
 
         self.ssl = False
@@ -1374,6 +1375,9 @@ def _read_ok_packet(self, first_packet):
         self.has_next = ok_packet.has_next
 
     def _read_load_local_packet(self, first_packet):
+        if not self.connection._local_infile:
+            raise RuntimeError(
+                "**WARN**: Received LOAD_LOCAL packet but local_infile option is false.")
         load_packet = LoadLocalPacketWrapper(first_packet)
         sender = LoadLocalFile(load_packet.filename, self.connection)
         try:
