Python工具之SQLAlchemy

#!/usr/bin/env python
# -*- coding:utf-8 -*-

from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, Integer, String, DATETIME
from sqlalchemy.orm import sessionmaker
from sqlalchemy import create_engine

# 以字典的形式配置好Mysql数据库的连接信息
mysql_dic = {
   'mysql_user':'test',
   'mysql_pass':'1234',
   'mysql_ip':'192.168.192.168',
   'mysql_port':3306,
   'mysql_db':'test',
}

class Connect(object):          # 创建一个专门连接数据库的类
   Base = declarative_base()    # 创建对象的基类
   def __init__(self):
      mysql_str = "mysql+mysqldb://{mysql_user}:{mysql_pass}@{mysql_ip}:{mysql_port}/{mysql_db}"    # 连接数据库的命令行
      mysql_con = mysql_str.format(**mysql_dic)                 # 格式化命令
      self.engine = create_engine(mysql_con, max_overflow=5)    # 初始化数据库连接
      Session = sessionmaker(bind=self.engine)                 # 创建一个会话来操作数据库
      self.session = Session()

   def create_table(self):
      '''寻找Base的所有子类，按照子类的结构在数据库中生成对应的数据表信息'''
      Connect.Base.metadata.create_all(self.engine)

class Server(Connect.Base):    # 定义一个类（其实就是一张数据库表）继承Base基类
   __tablename__ = 'server'    # 表名，以下均为表字段
   id = Column(Integer,autoincrement=1,primary_key=True)
   date = Column(DATETIME)
   user = Column(String(20))
   ip = Column(String(50))
   group = Column(String(50))

if __name__ == '__main__':
   c = Connect()    # 连接数据库并创建程序与数据库之间的会话
   c.create_table() # 创建定义的server表 
   
      # 增加数据
    s = Server(user='test',ip='192.168.100.1',group='monitor') # 实例化一个类对象
    c.session.add(s) # 通过session会话向表中添加一行数据（一个类对象）
    
    # c.session.add_all([
    #     Server(user='test',ip='192.168.100.2',group='monitor'),
    #     Server(user='test',ip='192.168.100.3',group='monitor')，
    # ])    # 一次添加多条数据
    
    c.session.commit()    # 提交
     
    
    # 删除数据（删除server表中id大于2的数据）
    c.session.query(Server).filter(Server.id > 2).delete()
    c.session.commit()
    
    # 修改数据（修改server表中id大于2的数据的group字段值为JAVA）
    c.session.query(Server).filter(Server.id > 2).update({'group' : 'JAVA'})
    c.session.commit()
    
    # 查询数据（查询server表中所有user字段值为test的数据）
    ret = c.session.query(Server).filter_by(user='test').all()
    for i in ret:
        print i.ip    # 获取满足条件的数据中ip字段的值
         
    c.session.close()     # 关闭会话