Django项目：堡垒机(Linux服务器主机管理系统)--02--02堡垒机设计后台交互程序

#main.py
#本文件写所有的连接交互动作程序
#————————————————02堡垒机设计后台交互程序 开始————————————————
from django.contrib.auth import authenticate #验证用户的证书
import getpass#输入密码时不显示密码
#用户登陆堡垒机后的交互程序
class HostManager(object):
def __init__(self):
self.user = None #当前账号 def interactive(self):#交互脚本
# ————————————————登录堡垒机 开始————————————————
print("----开始 运行交互脚本----")
count = 0 #用户名密码试错的次数
while count < 6:#用户名密码试错的次数
username = input("请输入你的用户名:").strip()#用户名
password = getpass.getpass("请输入你的密码:").strip()#密码
user = authenticate(username=username,password=password)#验证用户名密码
if user: #验证成功
print("欢迎 %s 登陆使用！".center(50,'-') % user.name )
self.user = user #验证成功,赋值
break #退出无限循环
else:
print("用户名或密码错误!")
count += 1#用户名密码试错的次数
else:
print("----结束 运行交互脚本----")
exit("用户名密码试错的次数到了,再见。")
"""
#————从外部调用django的方法 开始————
#注意写,从外部调用django的方法 E:\堡垒机\eye\fortess_manage.py
#注意,到admin后台配置主机给账户
# ————————————
#运行 Terminal
#E:\堡垒机\eye>python fortess_manage.py
#Username:admin@qq.com
#Password:admin123456
#————从外部调用django的方法 结束————
"""
# ————————————————登录堡垒机 结束————————————————
#————————————————02堡垒机设计后台交互程序 结束————————————————

#fortess_manage.py
#————————————————02堡垒机设计后台交互程序 开始————————————————
#————从外部调用django的方法 开始————
import os #操作系统
if __name__ == "__main__":
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'eye.settings')#找路径，环境变量
import django #导入django
django.setup()#安装django
from Fortress.backend import main #从backend导入main.py
obj = main.HostManager() #实例化类#main.py的类
obj.interactive() #调用类的函数 #交互脚本
#————从外部调用django的方法 结束————
#————————————————02堡垒机设计后台交互程序 结束————————————————

#models.py
#————————————————01堡垒机重写DJANGO账户表 开始————————————————
from django.db import models #模型
from django.contrib.auth.models import BaseUserManager,AbstractBaseUser,PermissionsMixin#DJANGO账户继承重写
#重写DJAGO账号表
class UserProfileManager(BaseUserManager):
def create_user(self, email, name, password=None):#创建并保存一个普通用户给定的邮箱地址,密码。
if not email:
raise ValueError('用户必须有一个电子邮箱地址')
user = self.model(email=self.normalize_email(email),#验证邮箱格式
name=name,)#创建一条普通用户记录
user.set_password(password)#对密码加密
user.save(using=self._db)#从临时文件读取出来数据，进行保存。
return user
def create_superuser(self, email,name,password):#创建并保存一个超级用户给定的邮箱地址,密码。
user = self.create_user(email,name=name,password=password)#创建一条超级用户记录
user.is_admin = True#管理员
user.save(using=self._db)#从临时文件读取出来数据，进行保存。
return user #重写DJAGO账号表
class UserProfile(AbstractBaseUser,PermissionsMixin):
# ————#DJANGO账号必须要有的字段 开始————
email = models.EmailField( max_length=255,unique=True,verbose_name='邮箱地址')
name = models.CharField(max_length=32,verbose_name='昵称')
is_active = models.BooleanField(default=True,verbose_name='是否启用')
is_admin = models.BooleanField(default=False,verbose_name='管理员',help_text='指定用户是否可以登录管理网站并且拥有全部权限',)
is_staff = models.BooleanField(default=True,verbose_name='员工状态',help_text='指定用户是否可以登录管理网站',)
# ————#DJANGO账号必须要有的字段 结束———— # ————————————————02堡垒机设计后台交互程序 开始————————————————
bind_hosts = models.ManyToManyField('BindHost',blank=True,verbose_name="关联的主机")
host_groups = models.ManyToManyField('HostGroups',blank=True,verbose_name="关联的分组")
# ————————————————02堡垒机设计后台交互程序 结束———————————————— #————#创建用户记录 开始————
objects = UserProfileManager()#创建用户记录，固定的变量名（objects）
USERNAME_FIELD = 'email' #用于登录的字段
REQUIRED_FIELDS = ['name']#必须填写
# ————#创建用户记录 结束————
# ————#DJANGO账号必须要有的方法 开始————
def get_full_name(self):#用户确认的电子邮件地址
return self.email
def get_short_name(self):#用户确认的电子邮件地址
return self.email
# ————#DJANGO账号必须要有的方法 结束————
def __str__(self):# __unicode__ on Python 2
return self.email
# ————#DJANGO账号权限 开始————
def has_perm(self, perm, obj=None):#权限#用户有一个特定的许可吗?
return True
def has_module_perms(self, app_label):#用户有权限查看应用名称吗？
return True
# ————#DJANGO账号权限 结束————
class Meta:
verbose_name_plural='00账户'
# ————自定制admin 开始————
#到admin.py里给DJANGO账户注册显示 #E:\堡垒机\eye\Fortress\admin.py
# ————自定制admin 结束————
# ————注册到配置文件 开始————
#到配置文件里给DJANGO账户指定模型 #E:\堡垒机\eye\eye\settings.py
#AUTH_USER_MODEL = 'web.UserProfile'
# ————注册到配置文件 结束————
#————————————————01堡垒机重写DJANGO账户表 结束————————————————
#————————————————02堡垒机设计后台交互程序 开始————————————————
#01标签
class Tag(models.Model):
name = models.CharField(max_length=64,unique=True) #标签名#CharField定长文本#最长度=64字节#不可以重复
class Meta:#通过一个内嵌类 "class Meta" 给你的 model 定义元数据
verbose_name_plural = "01标签" #verbose_name_plural给你的模型类起一个更可读的名字
def __str__(self): # __str__()是Python的一个“魔幻”方法，这个方法定义了当object调用str()时应该返回的值。
return self.name # 返回 #标签名
#02机房
class IDC(models.Model):
name = models.CharField(max_length=64, unique=True,verbose_name="机房名")
address=models.CharField(max_length=256,blank=True,null=True,verbose_name="机房地址")
memo = models.CharField(max_length=128,blank=True,null=True,verbose_name="机房的备注")
tag = models.ManyToManyField("Tag",blank=True,verbose_name="标签")#多对多关联到 标签表
date = models.DateTimeField(auto_now_add=True,verbose_name='创建时间')
class Meta:
verbose_name_plural='02机房'
def __str__(self):
return self.name
#03主机
class Host(models.Model):
hostname = models.CharField(max_length=64,unique=True,verbose_name="主机名")
ip_addr = models.GenericIPAddressField(verbose_name="主机IP地址")
port = models.SmallIntegerField(default=22,verbose_name="主机端口")
idc = models.ForeignKey('IDC',blank=True,null=True,on_delete=models.CASCADE,verbose_name="主机所在的机房")
system_type_choices = ((0,'Linux'),(1,'Windows'))
system_type = models.SmallIntegerField(choices=system_type_choices,default=0,verbose_name="操作系统类型")
enabled = models.BooleanField(default=1,verbose_name="是否启用本机")
memo = models.CharField(max_length=128,blank=True,null=True,verbose_name="主机的备注")
tag = models.ManyToManyField("Tag",blank=True,verbose_name="标签")#多对多关联到 标签表
date = models.DateTimeField(auto_now_add=True,verbose_name='创建时间')
class Meta:
verbose_name_plural='03主机'
unique_together = ('ip_addr','port')
def __str__(self):
return "%s(%s)" % (self.hostname, self.ip_addr)
#04账号密码
class RemoteUser(models.Model):
auth_type_choices =((0,'ssh-password'), (1,'ssh-key'))
auth_type = models.SmallIntegerField(choices=auth_type_choices,default=0,verbose_name="连接方式,密码或者密钥")
userlevel=models.CharField(max_length=128,blank=True,null=True,verbose_name="用户等级")
username = models.CharField(max_length=128,verbose_name="用户名")
password = models.CharField(max_length=256,verbose_name="密码或者密钥",help_text="如果连接方式选择为ssh-key,那此处就应该是key的路径")
memo = models.CharField(max_length=128,blank=True,null=True,verbose_name="账号密码的备注")
host_prompt=models.ManyToManyField("Host",blank=True,verbose_name="可关联的主机",help_text="关联主机与账号密码提示")
tag = models.ManyToManyField("Tag",blank=True,verbose_name="标签")#多对多关联到 标签表
date = models.DateTimeField(auto_now_add=True,verbose_name='创建时间')
class Meta:
verbose_name_plural='04账号密码'
unique_together = ('auth_type', 'username', 'password') # 联合唯一
def __str__(self):
return "%s：%s：%s" %(self.id,self.username,self.memo)
#05关联主机与账号密码
class BindHost(models.Model):
host = models.ForeignKey('Host',on_delete=models.CASCADE,verbose_name="远程主机名")
remote_user = models.ForeignKey('RemoteUser',on_delete=models.CASCADE,verbose_name="远程用户名")
memo = models.CharField(max_length=128,blank=True,null=True,verbose_name="关联的备注")
tag = models.ManyToManyField("Tag",blank=True,verbose_name="标签")#多对多关联到 标签表
date = models.DateTimeField(auto_now_add=True,verbose_name='创建时间')
class Meta:
verbose_name_plural = '05关联主机与账号密码'
unique_together = ('host', 'remote_user')#联合唯一
def __str__(self):
return "<%s：%s:%s>" %(self.host.id,self.host.hostname,self.remote_user.username)
#06主机分组
class HostGroups(models.Model):
name = models.CharField(max_length=64,unique=True,verbose_name="分组名称")
memo = models.CharField(max_length=128,blank=True,null=True,verbose_name="主机分组的备注")
bind_hosts = models.ManyToManyField('BindHost',blank=True,verbose_name="远程主机")
tag = models.ManyToManyField("Tag",blank=True,verbose_name="标签")#多对多关联到 标签表
date = models.DateTimeField(auto_now_add=True,verbose_name='创建时间')
class Meta:
verbose_name_plural='06主机分组'
def __str__(self):
return self.name
#————————————————02堡垒机设计后台交互程序 结束———————————————— """
# ————————————————Django中使用中国时区和中文 开始————————————————
到settings.py里修改#E:\堡垒机\eye\eye\settings.py
LANGUAGE_CODE = 'zh-Hans'
TIME_ZONE = 'Asia/Shanghai'
如果需要全局配置默认时区和语言,可在django下的conf模块下的global_settings.py中进行配置
所有可选的语言选项可通过python输入以下代码查看:
from django.conf.locale import LANG_INFO
{key: LANG_INFO[key]['name_local'] for key in LANG_INFO if 'name_local' in LANG_INFO[key]}
# ————————时间扩展 开始————————
USE_TZ = True
当设置了 USE_TZ 为 True 时，Django 与其他系统或服务的交流将强制使用 UTC 时间。
1.保证存储到数据库中的是 UTC 时间；
2.在函数之间传递时间参数时，确保时间已经转换成 UTC 时间；
# ——Django中跨时区 开始——
一般不跨时区的应用，可以不使用时区，即在settings.py设置
USE_TZ = False
这样存储的时间就是无时区的时间。
# ——Django中跨时区 结束——
# ————————时间扩展 结束————————
# ————————————————Django中使用中国时区和中文 结束————————————————
# ————————————————Django数据库生成 开始————————————————
运行 Terminal
生成 数据表
E:\堡垒机\eye>python manage.py makemigrations
数据表 迁移
E:\堡垒机\eye>python manage.py migrate
# ————————————————Django数据库生成 结束————————————————
# ————————————————重写Django账号表后的创建方法 开始————————————————
E:\堡垒机\eye>python manage.py createsuperuser
邮箱地址: admin@qq.com
昵称: admin
Password: admin123456
Password (again): admin123456
Superuser created successfully.
# ————————————————重写Django账号表后的创建方法 结束————————————————
#启动 DJANGO 服务
E:\堡垒机\eye>python manage.py runserver 127.0.0.1:8000
"""

#admin.py
#————————————————01堡垒机重写DJANGO账户表 开始————————————————
#————————同目录下可以用 . 开始————————
#from Fortress import models#数据库 #重写DJAGO账号表
from . import models#数据库 #重写DJAGO账号表
#————————同目录下可以用 . 结束————————
from django import forms #django的字段验证
from django.contrib.auth.admin import UserAdmin as BaseUserAdmin ##表单添加和更改用户实例
from django.contrib.auth.forms import ReadOnlyPasswordHashField#只读密码散列字段
from django.contrib import admin #注册类到admin
#创建新用户
class UserCreationForm(forms.ModelForm):
password1 = forms.CharField(label='设置密码', widget=forms.PasswordInput)#密码
password2 = forms.CharField(label='确认密码', widget=forms.PasswordInput)#确认密码
class Meta:
model = models.UserProfile
fields = ['email', 'name']
def clean_password2(self): #检查两个密码条目是否一样（匹配）
password1 = self.cleaned_data.get("password1")
password2 = self.cleaned_data.get("password2")
if password1 and password2 and password1 != password2:
raise forms.ValidationError("密码不匹配")
return password2
def save(self, commit=True): #保存密码（散列的格式提供）
user = super(UserCreationForm, self).save(commit=False)
user.set_password(self.cleaned_data["password1"])
if commit:
user.save()#保存
return user
# 修改用户密码
class UserChangeForm(forms.ModelForm):
password = ReadOnlyPasswordHashField(label="修改密码",
help_text=("原始密码不存储,所以没有办法看到"
"这个用户的密码,但是你可以改变密码 "
"使用 <a href=\"../password/\">修改密码</a>."))#哈值#只读密码散列字段
class Meta:
model = models.UserProfile
fields = ['email', 'password', 'name', 'is_active', 'is_admin']
def clean_password(self):#不管用户提供什么,返回初始值。
return self.initial["password"] #表单添加和更改用户实例
class UserProfileAdmin(BaseUserAdmin):
form = UserChangeForm #修改
add_form = UserCreationForm#创建
# 字段用于显示用户模型。
# 这些覆盖定义UserAdmin固定在底座上
# auth.User参考特定字段。
list_display = ['email', 'name','is_staff', 'is_admin']#后台显示的字段
list_filter = ['is_admin','is_staff']#后台过滤的字段
fieldsets = (('账号密码', {'fields': ('email', 'password')}),
('个人信息', {'fields': ('name',)}),
# ————————————————02堡垒机设计后台交互程序 开始————————————————
('堡垒机主机授权', {'fields': ('bind_hosts', 'host_groups')}),
# ————————————————02堡垒机设计后台交互程序 结束————————————————
('权限', {'fields': ('is_admin','is_staff','user_permissions','groups')}),)
#添加账户时显示要填写的字段
add_fieldsets = (('添加账户', {'classes': ('wide',),#wide（宽屏样式）
'fields': ('email', 'name', 'password1', 'password2')}),)
search_fields = ['email']#后台搜索的字段
ordering = ['email']#后台排序的字段
filter_horizontal =['user_permissions','groups',#必须覆盖的变量#后台复选框的字段
# ————————————————02堡垒机设计后台交互程序 开始————————————————
'bind_hosts', 'host_groups',
# ————————————————02堡垒机设计后台交互程序 结束————————————————
]
admin.site.register(models.UserProfile, UserProfileAdmin)#现在注册这个新UserProfileAdmin
# ——可选——使用Django的内置权限 开始————
#from django.contrib.auth.models import Group #权限组
#admin.site.unregister(Group)#权限组#因为我们不使用Django的内置权限,
# ——可选——使用Django的内置权限 结束————
#————————————————01堡垒机重写DJANGO账户表 结束———————————————— #————————————————02堡垒机设计后台交互程序 开始————————————————
#01标签
class TagAdmin(admin.ModelAdmin):
list_display = ['id','name'] #显示字段表头
list_editable = ['name'] #可编辑
search_fields = ['name'] #搜索
admin.site.register(models.Tag, TagAdmin)
#02机房
class IDCAdmin(admin.ModelAdmin):
list_display = ['id','name','address','memo','date'] #显示字段表头
list_editable = ['name','address', 'memo'] #可编辑
search_fields = ['name','tag'] #搜索
filter_horizontal = ['tag'] #复选框
admin.site.register(models.IDC, IDCAdmin)
#03主机
class HostAdmin(admin.ModelAdmin):
list_display = ['id','hostname','ip_addr','port','memo','enabled','date'] #显示字段表头
list_editable = ['hostname','ip_addr','port','memo','enabled'] #可编辑
search_fields = ['hostname','ip_addr','port','tag'] #搜索
filter_horizontal = ['tag'] #复选框
admin.site.register(models.Host, HostAdmin)
#04账号密码
class RemoteUserAdmin(admin.ModelAdmin):
list_display = ['id','userlevel','username','memo','date'] #显示字段表头
list_editable = ['memo'] #可编辑
search_fields = ['userlevel','username','tag'] #搜索
filter_horizontal = ['host_prompt','tag'] #复选框
admin.site.register(models.RemoteUser,RemoteUserAdmin)
#05关联主机与账号密码
class BindHostAdmin(admin.ModelAdmin):
list_display = ['id','host','remote_user','memo','date']#显示字段表头
list_editable = ['host','remote_user','memo'] #可编辑
search_fields = ['host','remote_user','tag'] #搜索
filter_horizontal = ['tag'] #复选框
ordering = ['host','remote_user'] #自定义排序
admin.site.register(models.BindHost,BindHostAdmin)
#06主机分组
class HostGroupsAdmin(admin.ModelAdmin):
list_display = ['id','name','memo','date']#显示字段表头
list_editable = ['name','memo'] #可编辑
search_fields = ['name','tag'] #搜索
filter_horizontal = ['bind_hosts','tag'] #复选框
admin.site.register(models.HostGroups, HostGroupsAdmin)
#————————————————02堡垒机设计后台交互程序 结束————————————————

#main.py
#本文件写所有的连接交互动作程序
#————————————————02堡垒机设计后台交互程序 开始————————————————
from django.contrib.auth import authenticate #验证用户的证书
#用户登陆堡垒机后的交互程序
class HostManager(object):
def __init__(self):
self.user = None #当前账号
def interactive(self):#交互脚本
# ————————————————登录堡垒机 开始————————————————
print("----开始 运行交互脚本----")
count = 0 #用户名密码试错的次数
while count < 6:#用户名密码试错的次数
username = input("Username:").strip()#用户名
password = input("Password:").strip()#密码
user = authenticate(username=username,password=password)#验证用户名密码
if user: #验证成功
print("欢迎 %s 登陆使用！".center(50,'-') % user.name )
self.user = user #验证成功,赋值
break #退出无限循环
else:
print("用户名或密码错误!")
count += 1#用户名密码试错的次数
else:
print("----结束 运行交互脚本----")
exit("用户名密码试错的次数到了,再见。")
"""
#————从外部调用django的方法 开始————
#注意写,从外部调用django的方法 E:\堡垒机\eye\fortess_manage.py
#注意,到admin后台配置主机给账户
# ————————————
#运行 Terminal
#E:\堡垒机\eye>python fortess_manage.py
#Username:admin@qq.com
#Password:admin123456
#————从外部调用django的方法 结束————
"""
# ————————————————登录堡垒机 结束————————————————
# ——————————————选择要连接的主机 开始————————————
if self.user: #验证登录成功
while True:
#————————打印当前账户所有的可以登录主机 开始——————
for index,host_group in enumerate(self.user.host_groups.all()): #select_related()
print("%s.\t%s[%s]" %(index,host_group.name, host_group.bind_hosts.count()))
print("z.\t未分组主机[%s]" %(self.user.bind_hosts.count()))
#————————打印当前账户所有的可以登录主机 结束——————
# ————————开始选择分组 开始——————
choice = input("%s>>:"% self.user).strip()#输入选择
if len(choice) == 0:#长度等于0
continue#跳过这次循环
selected_host_group = None #选择分组
if choice.isdigit():#检测字符串是否只由数字组成
choice = int(choice)#转成整数
if choice >=0 and choice <= index: #判断输入的选项是否合法
selected_host_group = self.user.host_groups.all()[choice]#迭代选择分组
elif choice == 'z':#未分组
selected_host_group = self.user
# ————————开始选择分组 结束——————
# ————————开始选择主机 开始——————
if selected_host_group: #选择分组后选择主机
while True:
print("当前选择的分组: %s ".center(50,'-') %selected_host_group)
# ————————打印分组里的所有主机 开始——————
for index, bind_host in enumerate(selected_host_group.bind_hosts.all()):
print("%s.\t%s" % (index, bind_host))
print("返回上级菜单，请按 b ")
# ————————打印分组里的所有主机 结束——————
# ————————选择分组里的主机 开始——————
choice = input("%s>>>:" % self.user).strip()#输入选择
if choice.isdigit():#检测字符串是否只由数字组成
choice = int(choice)#转成整数
if choice >= 0 and choice <= index:#判断输入的选项是否合法
bind_host = selected_host_group.bind_hosts.all()[choice]#迭代选择主机
print("准备登录 %s 主机".center(50,'-') %bind_host)
# ————————选择分组里的主机 结束——————
elif choice == 'b':#退出
break #终止无限循环
# ————————开始选择主机 结束——————
# ——————————————选择要连接的主机 结束————————————
#————————————————02堡垒机设计后台交互程序 结束————————————————