Python連接PostgreSQL數(shù)據(jù)庫(kù)

genedna

我來(lái)貼一個(gè)Python和PostgreSQL數(shù)據(jù)庫(kù)的例子，慶祝開版。

1. from mx.DateTime import *
   
2. import pg
   
3. import random
   
4. 
   
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}
    
24. 
    
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

復(fù)制代碼