Use _quote_ident to escape channel names.

Author: sqwishy

asyncpg/connection.py

@@ -112,7 +112,7 @@ async def add_listener(self, channel, callback):
         """
         self._check_open()
         if channel not in self._listeners:
-            await self.fetch('LISTEN "{}"'.format(channel))
+            await self.fetch('LISTEN {}'.format(utils._quote_ident(channel)))
             self._listeners[channel] = set()
         self._listeners[channel].add(callback)
 
@@ -127,7 +127,7 @@ async def remove_listener(self, channel, callback):
         self._listeners[channel].remove(callback)
         if not self._listeners[channel]:
             del self._listeners[channel]
-            await self.fetch('UNLISTEN "{}"'.format(channel))
+            await self.fetch('UNLISTEN {}'.format(utils._quote_ident(channel)))
 
     def add_log_listener(self, callback):
         """Add a listener for Postgres log messages.

tests/test_listeners.py

@@ -87,14 +87,14 @@ async def test_listen_notletters(self):
                 def listener1(*args):
                     q1.put_nowait(args)
 
-                await con1.add_listener('12+34', listener1)
-                await con2.execute("""NOTIFY "12+34", 'hello'""")
+                await con1.add_listener('12+"34', listener1)
+                await con2.execute("""NOTIFY "12+""34", 'hello'""")
 
                 self.assertEqual(
                     await q1.get(),
-                    (con1, con2.get_server_pid(), '12+34', 'hello'))
+                    (con1, con2.get_server_pid(), '12+"34', 'hello'))
 
-                await con1.remove_listener('12+34', listener1)
+                await con1.remove_listener('12+"34', listener1)
 
     async def test_dangling_listener_warns(self):
         async with self.create_pool(database='postgres') as pool: