堡垒机-数据库设计

 

 数据库表结构设计：

 

 

 层次结构：

 

 

 

 

 

import sqlalchemy
from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
#Column导入
from  sqlalchemy import  Column,Enum, Integer, String,UniqueConstraint,ForeignKey,Table
from sqlalchemy.orm import sessionmaker,relationship
from sqlalchemy_utils import ChoiceType,PasswordType


Base = declarative_base()#生成orm基类

user_m2m_bindhost = Table(\'user_m2m_bindhost\', Base.metadata,
                        Column(\'userprofile_id\', Integer, ForeignKey(\'user_profile.id\')),
                        Column(\'bindhost_id\', Integer, ForeignKey(\'bind_host.id\')),
                        )
bindhost_m2m_hostgroup=Table(\'bindhost_m2m_hostgroup\', Base.metadata,
                      Column(\'bindhost_id\',Integer,ForeignKey(\'bind_host.id\')),
                      Column(\'hostgroup_id\',Integer,ForeignKey(\'host_group.id\'))
)

user_m2m_hostgroup=Table(\'user_m2m_hostgroup\', Base.metadata,
                      Column(\'userprofile_id\',Integer,ForeignKey(\'user_profile.id\')),
                      Column(\'hostgroup_id\',Integer,ForeignKey(\'host_group.id\'))
)

class  Host(Base):
    __tablename__ = \'host\'
    id =Column(Integer,primary_key=True)
    hostname =Column(String(64),unique=True)
    ip = Column(String(64),unique=True)
    port=Column(Integer,default=22)
    # remote_users =relationship(\'RemoteUser\', secondary = host_m2m_remoteuser, backref=\'hosts\')
    def __repr__(self):
        return self.hostname
class HostGroup(Base):
    __tablename__ = \'host_group\'
    id = Column(Integer, primary_key=True)
    name = Column(String(64), unique=True)
    bind_hosts = relationship(\'BindHost\',secondary="bindhost_m2m_hostgroup",backref="host_groups")
    def __repr__(self):
        return self.name
class RemoteUser(Base):
    __tablename__ = \'remote_user\'
    #创建联合唯一auth_type、username、password
    __table_args__ = (UniqueConstraint(\'auth_type\', \'username\', \'password\', name=\'_user_passwd_uc\'),)
    AuthTypes = [
        (\'ssh-password\', \'SSH/Password\'),#第一个是存到数据库的，第二个是显示给我们看的
        (\'ssh-key\', \'SSH/KEY\'),
    ]
    id = Column(Integer, primary_key=True, autoincrement=True)
    auth_type = Column(ChoiceType(AuthTypes))
    username = Column(String(64), nullable=False)
    password = Column(String(255))



    def __repr__(self):
        return self.username


class BindHost(Base):
    \'\'\'
    192.168.1.11    web
    192.168.1.11    mysql

    \'\'\'
    __tablename__ = "bind_host"
    __table_args__ = (UniqueConstraint(\'host_id\',\'remoteuser_id\', name=\'_host_remoteuser_uc\'),)

    id = Column(Integer, primary_key=True)
    host_id = Column(Integer,ForeignKey(\'host.id\'))
    #group_id = Column(Integer,ForeignKey(\'group.id\'))
    remoteuser_id = Column(Integer, ForeignKey(\'remote_user.id\'))
    host = relationship("Host",backref="bind_hosts")
    #host_group = relationship("HostGroup",backref="bind_hosts")
    remote_user = relationship("RemoteUser",backref="bind_hosts")
    def __repr__(self):
        return "<%s -- %s >" %(self.host.ip,
                                   self.remote_user.username
                                  )

class UserProfile(Base):
    __tablename__ = \'user_profile\'
    id = Column(Integer, primary_key=True)
    username = Column(String(32), unique=True)
    password = Column(String(128))

    bindhosts =relationship(\'BindHost\', secondary=user_m2m_bindhost, backref = \'user_profiles\')
    host_groups = relationship("HostGroup", secondary = user_m2m_hostgroup, backref = "user_profiles")
    def __repr__(self):
        return self.username


# class AuditLog(Base):
#     pass


if __name__ == \'__main__\':
    Base.metadata.create_all()  # 创建所有表结构