Fix #454

aiomysql/connection.py

@@ -10,6 +10,10 @@
 import configparser
 import getpass
 from functools import partial
+try:
+    from asyncio.streams import IncompleteReadError
+except ImportError:
+    from asyncio.exceptions import IncompleteReadError
 
 from pymysql.charset import charset_by_name, charset_by_id
 from pymysql.constants import SERVER_STATUS
@@ -596,7 +600,7 @@ async def _read_packet(self, packet_type=MysqlPacket):
     async def _read_bytes(self, num_bytes):
         try:
             data = await self._reader.readexactly(num_bytes)
-        except asyncio.streams.IncompleteReadError as e:
+        except IncompleteReadError as e:
             msg = "Lost connection to MySQL server during query"
             raise OperationalError(2013, msg) from e
         except (IOError, OSError) as e: