Merge pull request #170 from methane/select-db

Add con.select_db() method. (fixes #80)

pymysql/connections.py

@@ -706,6 +706,11 @@ class Connection(object):
         self._execute_command(COM_QUERY, "ROLLBACK")
         self._read_ok_packet()
 
+    def select_db(self, db):
+        '''Set current db'''
+        self._execute_command(COM_INIT_DB, db)
+        self._read_ok_packet()
+
     def escape(self, obj):
         ''' Escape whatever value you pass to it  '''
         if isinstance(obj, str_type):

pymysql/tests/test_connection.py

@@ -41,6 +41,19 @@ class TestConnection(base.PyMySQLTestCase):
         cur.execute("SELECT @@AUTOCOMMIT")
         self.assertEqual(cur.fetchone()[0], 0)
 
+    def test_select_db(self):
+        con = self.connections[0]
+        current_db = self.databases[0]['db']
+        other_db = self.databases[1]['db']
+
+        cur = con.cursor()
+        cur.execute('SELECT database()')
+        self.assertEqual(cur.fetchone()[0], current_db)
+
+        con.select_db(other_db)
+        cur.execute('SELECT database()')
+        self.assertEqual(cur.fetchone()[0], other_db)
+
 
 if __name__ == "__main__":
     try: