Python工具之SQLAlchemy

457475451

#!/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()     # 关闭会话