File: client.py

package info (click to toggle)
mysql-connector-python 1.2.3-2
  • links: PTS, VCS
  • area: main
  • in suites: jessie, jessie-kfreebsd
  • size: 1,716 kB
  • ctags: 5,129
  • sloc: python: 23,339; makefile: 28
file content (144 lines) | stat: -rw-r--r-- 4,220 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
#!/usr/bin/env python
# -*- coding: utf-8 -*-

# MySQL Connector/Python - MySQL driver written in Python.
# Copyright (c) 2009, 2013, Oracle and/or its affiliates. All rights reserved.

# MySQL Connector/Python is licensed under the terms of the GPLv2
# <http://www.gnu.org/licenses/old-licenses/gpl-2.0.html>, like most
# MySQL Connectors. There are special exceptions to the terms and
# conditions of the GPLv2 as it is applied to this software, see the
# FOSS License Exception
# <http://www.mysql.com/about/legal/licensing/foss-exception.html>.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA

"""
Simple CLI using the Connector/Python. It does not take arguments so
you'll need to set the proper login information manually.

Example output:

$ ./client.py 
Welcome to MySQL Python CLI.
Your MySQL connection ID is 21.
Server version: 5.0.62-enterprise-gpl

Python 2.5.1 (r251:54863, Jan 17 2008, 19:35:17) 
[GCC 4.0.1 (Apple Inc. build 5465)] on darwin
Type "help", "copyright", "credits" or "license" for more information.
(MySQLConsole)
>>> show tables;
1046 (3D000): No database selected
>>> use test;
>>> show tables;
>>> create table t1 (id int);
>>> show tables;
(u't1')
"""

from __future__ import print_function

import sys
import os
import readline
import code
import atexit
import re

# We're actually just a demo and supposed to run in the source directory.
sys.path.append('.')

import mysql.connector

class MySQLConsole(code.InteractiveConsole):
    mysql = None
    regexp = {}
    
    def __init__(self, mysql, locals=None, filename="<console>",
                 histfile=os.path.expanduser("~/.mysql_history")):
        code.InteractiveConsole.__init__(self)
        self.init_history(histfile)
        self.mysql = mysql

        self.regexp['USE'] = re.compile('USE (\w*)', re.IGNORECASE)

    def init_history(self, histfile):
        readline.parse_and_bind("tab: complete")
        if hasattr(readline, "read_history_file"):
            try:
                readline.read_history_file(histfile)
            except IOError:
                pass
            atexit.register(self.save_history, histfile)

    def save_history(self, histfile):
        readline.write_history_file(histfile)
    
    def send_query(self, line):
      rows = ()
      try:
          cursor = self.mysql.cursor()
          cursor.execute(line)
      except mysql.connector.errors.Error as e:
          print(e.errmsglong)
          return
      
      try:
          rows = cursor.fetchall()
          for row in rows:
              print(row)
      except:
          pass
    
    def _do_use(db):
      try:
          my.cmd_init_db(db)
      except mysql.connector.errors.InterfaceError as e:
          print(e)
            
    def push(self, line):
      try:
        res = self.regexp['USE'].findall(line)
        db = res[0]
        self._do_use(db)
      except:
        pass
      self.send_query(line)
      code.InteractiveConsole(self, line)


if __name__ == '__main__':

    print("Welcome to MySQL Python CLI.")
    
    try:
        db = mysql.connector.Connect(unix_socket='/tmp/mysql.sock', user='root', password='')
    except mysql.connector.errors.InterfaceError as e:
        print(e)
        sys.exit(1)
    
    console = MySQLConsole(db)
    myconnpy_version = "%s-%s" % (
        '.'.join(map(str,mysql.connector.__version__[0:3])),
        mysql.connector.__version__[3])
    
    print("Your MySQL connection ID is %d." % (db.get_server_threadid()))
    print("Server version: %s" % (db.get_server_info()))
    print("MySQL Connector/Python v%s" % (myconnpy_version))
    print()
    
    console.interact()