sqlalchemy 简单实例

本文地址：http://blog.youkuaiyun.com/spch2008/article/details/9014559

[python]  view plain copy print ?

1. from sqlalchemy import *  
2. from sqlalchemy.orm import sessionmaker, mapper  
3. from obj.dbobj import User  
4.   
5.   
6. def init_db():  
7.     engine = create_engine("mysql://root:sqlPwd@localhost/chat")  
8.     metadata = MetaData()  
9.   
10.     table = Table('users', metadata,  
11.                     Column('user_id', String(20), primary_key=True),  
12.                     Column('user_pwd', String(20))  
13.                  )  
14.     metadata.create_all(engine)  
15.   
16.     mapper(User, table)  
17.       
18.     Session = sessionmaker(engine)  
19.     return Session()  
20.   
21.   
22. class database:  
23.     def __init__(self, session):  
24.         self.session = session  
25.         self.query   = self.session.query(User)  
26.       
27.     def getUser(self, user_id):  
28.         return self.query.get(user_id)  
29.    
30.     def getPwd(self, user_id):  
31.         user = self.getUser(user_id)  
32.         return user.pwd()  
33.           
34.       
35.     def insertInfo(self, user_id, pwd):  
36.         user = User(user_id, pwd)  
37.         self.session.add(user)  
38.         self.session.commit()  
39.       
40.     def deleteInfo(self, user_id):  
41.         user = self.getUser(user_id)  
42.         self.session.delete(user)  
43.         self.session.commit()  
44.       
45.     def getCount(self):  
46.         return self.query.count()  
47.       
48.     def getAllUser(self):  
49.         return self.query.all()  
50.       
51.     def modifyPwd(self, user_id, pwd):  
52.         self.query.filter(User.user_id == user_id).update({"user_pwd" : pwd})  
53.         self.session.commit()  
54.   
55.   
56.   
57. def getDB():  
58.     engine = init_db()  
59.     db = database(engine)  
60.     return db