|
import psycopg
rows = ({'num': 0, 'text': 'Zero'},
{'num': 1, 'text': 'Item One'},
{'num': 2, 'text': 'Item Two'},
{'num': 3, 'text': 'Three'})
def getdsn(db = None, user = None, passwd = None, host = None):
if user == None:
import os, pwd
user = pwd.getpwuid(os.getuid())[0]
if db == None:
db = user
dsn = 'dbname=%s user=%s' % (db, user)
if passwd != None:
dsn += ' password=' + passwd
if host != None:
dsn += ' host=' + host
return dsn
dsn = getdsn()
print "Connecting to %s" % dsn
dbh = psycopg.connect(dsn)
print "Connection successful."
cur = dbh.cursor()
cur.execute("DELETE FROM myTable")
query = "INSERT INTO myTable VALUES (%(num)d, %(text)s)"
for row in rows:
cur.execute(query, row)
dbh.commit()
dbh.close()
|