python_day11のPython操作 pymysql && SQLAchemy

长枪不倒

#!/usr/bin/env python　　
# -*- coding:utf-8 -*-
　　
from sqlalchemy import create_engine,and_,or_,func,Table
　　
from sqlalchemy.ext.declarative import declarative_base
　　
from sqlalchemy import Column, Integer, String,ForeignKey,UniqueConstraint,DateTime
　　
from  sqlalchemy.orm import sessionmaker,relationship
　　

　　
Base = declarative_base() #生成一个SqlORM 基类
　　

　　
# 服务器账号和组
　　
# HostUser2Group = Table('hostuser_2_group',Base.metadata,
　　
#     Column('hostuser_id',ForeignKey('host_user.id'),primary_key=True),
　　
#     Column('group_id',ForeignKey('group.id'),primary_key=True),
　　
# )
　　

　　
# 用户和组关系表，用户可以属于多个组，一个组可以有多个人
　　
UserProfile2Group = Table('userprofile_2_group',Base.metadata,
　　
    Column('userprofile_id',ForeignKey('user_profile.id'),primary_key=True),
　　
    Column('group_id',ForeignKey('group.id'),primary_key=True),
　　
)
　　

　　
# 程序登陆用户和服务器账户，一个人可以有多个服务器账号，一个服务器账号可以给多个人用
　　
UserProfile2HostUser= Table('userprofile_2_hostuser',Base.metadata,
　　
    Column('userprofile_id',ForeignKey('user_profile.id'),primary_key=True),
　　
    Column('hostuser_id',ForeignKey('host_user.id'),primary_key=True),
　　
)
　　

　　

　　
class Host(Base):
　　
    __tablename__='host'
　　
    id = Column(Integer,primary_key=True,autoincrement=True)
　　
    hostname = Column(String(64),unique=True,nullable=False)
　　
    ip_addr = Column(String(128),unique=True,nullable=False)
　　
    port = Column(Integer,default=22)
　　
    def __repr__(self):
　　
        return  "<id=%s,hostname=%s, ip_addr=%s>" %(self.id,
　　
                                                    self.hostname,
　　
                                                    self.ip_addr)
　　

　　

　　
class HostUser(Base):
　　
    __tablename__ = 'host_user'
　　
    id = Column(Integer,primary_key=True)
　　
    AuthTypes = [
　　
        (u'ssh-passwd',u'SSH/Password'),
　　
        (u'ssh-key',u'SSH/KEY'),
　　
    ]
　　
    # auth_type = Column(ChoiceType(AuthTypes))
　　
    auth_type = Column(String(64))
　　
    username = Column(String(64),unique=True,nullable=False)
　　
    password = Column(String(255))
　　

　　
    host_id = Column(Integer,ForeignKey('host.id'))
　　

　　
    # groups = relationship('Group',
　　
    #                       secondary=HostUser2Group,
　　
    #                       backref='host_list')
　　

　　
    __table_args__ = (UniqueConstraint('host_id','username', name='_host_username_uc'),)
　　

　　
    def __repr__(self):
　　
        return  "<id=%s,name=%s>" %(self.id,self.username)
　　

　　

　　
class Group(Base):
　　
    __tablename__ = 'group'
　　
    id = Column(Integer,primary_key=True)
　　
    name = Column(String(64),unique=True,nullable=False)
　　
    def __repr__(self):
　　
        return  "<id=%s,name=%s>" %(self.id,self.name)
　　

　　

　　
class UserProfile(Base):
　　
    __tablename__ = 'user_profile'
　　
    id = Column(Integer,primary_key=True)
　　
    username = Column(String(64),unique=True,nullable=False)
　　
    password = Column(String(255),nullable=False)
　　
    # host_list = relationship('HostUser',
　　
    #                       secondary=UserProfile2HostUser,
　　
    #                       backref='userprofiles')
　　
    # groups = relationship('Group',
　　
    #                       secondary=UserProfile2Group,
　　
    #                       backref='userprofiles')
　　
    def __repr__(self):
　　
        return  "<id=%s,name=%s>" %(self.id,self.username)
　　

　　

　　
class AuditLog(Base):
　　
    __tablename__ = 'audit_log'
　　
    id = Column(Integer,primary_key=True)
　　
    userprofile_id = Column(Integer,ForeignKey('user_profile.id'))
　　
    hostuser_id = Column(Integer,ForeignKey('host_user.id'))
　　
    action_choices2 = [
　　
        (u'cmd',u'CMD'),
　　
        (u'login',u'Login'),
　　
        (u'logout',u'Logout'),
　　
    ]
　　
    action_type = Column(ChoiceType(action_choices2))
　　
    #action_type = Column(String(64))
　　
    cmd = Column(String(255))
　　
    date = Column(DateTime)
　　

　　
    # user_profile = relationship("UserProfile")
　　
    #bind_host = relationship("BindHost")
　　

　　

　　
engine = create_engine("mysql+pymsql://root:123@localhost:3306/stupid_jumpserver",echo=False)
　　
Base.metadata.create_all(engine) #创建所有表结构