Chinaunix

标题: Python连接PostgreSQL数据库 [打印本页]
作者: genedna    时间: 2005-05-31 11:43
标题: Python连接PostgreSQL数据库
我来贴一个Python和PostgreSQL数据库的例子，庆祝开版。
  1. from mx.DateTime import *
  2. import pg
  3. import random

  5. def insertdb(id, initNum):
  6.   nrange = range(initNum - 15, initNum +20)
  7.   marchbegin = Date(2005, 3, 1)
  8.   marchend = Date(2005, 4, 1)
  9.   i = 10
  10.   insertstrlist = []
  11.   while True:
  12.     intever = marchbegin + RelativeDateTime(day=1,hour=0,minute=i,second=0)
  13.     n = random.choice(nrange)
  14.     sql = "INSERT INTO srdata (\"srid\", \"srdtime\", \"srdcount\", \"srdstat\") VALUES ('" + str(id) + "', '" + str(intever) + "', '" + str(n) + "', '" + str(initNum) +"')"
  15.     insertstrlist.append(sql)
  16.     if marchend == intever :
  17.       break
  18.     i = i + 10
  19.   return insertstrlist
  20.   
  21. region50list = {'1':1120, '2':1222, '3': 1103, '4': 902}
  22. region64list = {'5':1000, '6':1200, '7': 973, '8': 823}
  23. region73list = {'9':1020, '10':1287, '11': 1122, '12': 1244}

  25. db = pg.connect(dbname = 'stat', host = '192.168.0.2', user = 'bbs', passwd = '2037423')
  26. for region in db.query("SELECT * FROM sregions").dictresult():
  27.   if (region["srid"] < 5):
  28.     num = region50list[str(region["srid"])]
  29.     sql50 = insertdb(region["srid"], num)
  30.     for i in sql50:
  31.       db.query(i)
  32.       print "Insert OK : " + i
  33.    
  34.   if (region["srid"] >; 5 and region["srid"] < 9):
  35.     num = region64list[str(region["srid"])]
  36.     sql64 = insertdb(region["srid"], num)
  37.     for i in sql64:
  38.       db.query(i)
  39.       print "Insert OK : " + i
  40.    
  41.   if (region["srid"] >; 8 and region["srid"] < 13):
  42.     num = region73list[str(region["srid"])]
  43.     sql70 = insertdb(region["srid"], num)
  44.     for i in sql70:
  45.       db.query(i)
  46.       print "Insert OK : " + i
复制代码




欢迎光临 Chinaunix (http://bbs.chinaunix.net/) Powered by Discuz! X3.2