Mega Code Archive

 

Categories / Python Tutorial / Database

 

Create a table and insert data

import psycopg 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("""CREATE TABLE myTable(mynum integer UNIQUE,mystring  varchar(30))""") cur.execute("INSERT INTO myTable VALUES (5, 'Five')") cur.execute("INSERT INTO myTable VALUES (0)") dbh.commit() dbh.close()