openstack/deb-python-pymysql
Code Issues Proposed changes

Merge pull request #306 from boglin/multi_init_cmd

Browse Source 
Multi-statement init_command causes subsequent query to fail
This commit is contained in:
INADA Naoki
2015-02-15 18:31:56 +09:00
parent b2ec828715 ac0cc7e51a
commit 8f599fe643
2 changed files with 11 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
pymysql/connections.py
View File

@@ -841,6 +841,7 @@ class Connection(object):
if self.init_command is not None: if self.init_command is not None:
c = self.cursor() c = self.cursor()
c.execute(self.init_command) c.execute(self.init_command)
c.close()
self.commit() self.commit()
if self.autocommit_mode is not None: if self.autocommit_mode is not None:

10
pymysql/tests/test_connection.py
View File

@@ -69,3 +69,13 @@ class TestConnection(base.PyMySQLTestCase):
# error occures while reading, not writing because of socket buffer. # error occures while reading, not writing because of socket buffer.
#self.assertEquals(cm.exception.args[0], 2006) #self.assertEquals(cm.exception.args[0], 2006)
self.assertIn(cm.exception.args[0], (2006, 2013)) self.assertIn(cm.exception.args[0], (2006, 2013))
def test_init_command(self):
conn = pymysql.connect(
init_command='SELECT "bar"; SELECT "baz"',
**self.databases[0]
)
c = conn.cursor()
c.execute('select "foobar";')
self.assertEqual(('foobar',), c.fetchone())
conn.close()