PythonMySQL: Difference between revisions

From XPUB & Lens-Based wiki
No edit summary
Line 25: Line 25:
         print "Error message:", e.args[1]
         print "Error message:", e.args[1]
         sys.exit(1)
         sys.exit(1)
</source>
== Executing a Query ==
=== Many rows ===
<source lang="python">
import settings
dbconn = settings.connect_db()
cursor = dbconn.cursor()
q = "SELECT id FROM items WHERE status='active' AND name=%s ORDER BY id"
cursor.execute(q, (name,))
while 1:
row = cursor.fetchone()
if (row == None): break
print row[0]
# if you forget this, it happens automatically
cursor.close()
dbconn.close()
</source>
</source>

Revision as of 15:23, 11 June 2008

You'll need the "MySQLdb" library installed.

Connecting to MySQL

If you put this function in a file (say "settings.py"), you can easily import if from different scripts... and also you only need to change the login data in one location (useful for local vs. server-based testing).

import MySQLdb

def connect_db():
    host_name = "localhost"
    db_name = "dbname"
    user_name = "dbuser"
    password = "dbpass"
    
    try:
        conn = MySQLdb.connect (db = db_name,
                                host = host_name,
                                user = user_name,
                                passwd = password)
        return conn
    except MySQLdb.Error, e:
        print "Cannot connect to server"
        print "Error code:", e.args[0]
        print "Error message:", e.args[1]
        sys.exit(1)

Executing a Query

Many rows

import settings

dbconn = settings.connect_db()
cursor = dbconn.cursor()
q = "SELECT id FROM items WHERE status='active' AND name=%s ORDER BY id"
cursor.execute(q, (name,))
while 1:
	row = cursor.fetchone()
	if (row == None): break
	print row[0]

# if you forget this, it happens automatically
cursor.close()
dbconn.close()