models
from django.contrib.auth.models import AbstractUser
from django.db import models
# Create your models here.
class RemoteUser(models.Model):
"""主机账号"""
auth_type = models.SmallIntegerField(choices=((0, 'sshpass'), (1, 'sshkey')), default=0)
username = models.CharField(max_length=30)
password = models.CharField(max_length=60, blank=True, null=True)
class Meta:
unique_together = ("auth_type", "username", "password")
verbose_name = "主机账号"
verbose_name_plural = verbose_name
def __str__(self):
return "%s/%s" % (self.username, self.password)
class HostGroup(models.Model):
"""主机组"""
name = models.CharField(max_length=30)
# hosts = models.ForeignKey(Hosts, on_delete=models.CASCADE)
class Meta:
verbose_name = "主机组"
verbose_name_plural = verbose_name
def __str__(self):
return "%s" % (self.name)
class Hosts(models.Model):
"""主机"""
name = models.CharField(max_length=30, unique=True)
ip = models.GenericIPAddressField(unique=True)
port = models.SmallIntegerField(default=22)
idc = models.ForeignKey("IDC", on_delete=models.CASCADE)
# remote_user = models.ManyToManyField(RemoteUser)
host_group = models.ForeignKey(HostGroup, on_delete=models.CASCADE)
class Meta:
verbose_name = "主机"
verbose_name_plural = verbose_name
def __str__(self):
return self.name
class HostToRemoteUser(models.Model):
"""主机和账号绑定"""
host = models.ForeignKey(Hosts, on_delete=models.CASCADE)
remote_user = models.ForeignKey(RemoteUser, on_delete=models.CASCADE)
class IDC(models.Model):
name = models.CharField(max_length=30, unique=True)
class Meta:
verbose_name = "IDC机房"
verbose_name_plural = verbose_name
def __str__(self):
return self.name
# class UserProfile(AbstractUser):
# # host_to_remote_users = models.ManyToManyField("HostToRemoteUser", blank=True, null=True)
# # host_groups = models.ManyToManyField("HostGroup", blank=True, null=True)
# tel = models.CharField(max_length=30)
#
# class Meta:
# verbose_name = "用户信息"
# verbose_name_plural = verbose_name
#
# def __str__(self):
# return self.username
admin
from django.contrib import admin
from .models import RemoteUser, Hosts, HostGroup, IDC, HostToRemoteUser
# Register your models here.
class RemoteUserAdmin(admin.ModelAdmin):
"""主机账号"""
list_display = ("username", "password")
class HostsAdmin(admin.ModelAdmin):
"""主机"""
list_display = ("name", "ip", "port","idc","host_group")
class HostGroupAdmin(admin.ModelAdmin):
"""主机组"""
list_display = ("name",)
class IDCAdmin(admin.ModelAdmin):
"""主机组"""
list_display = ("name",)
class HostToRemoteUserAdmin(admin.ModelAdmin):
"""主机组"""
list_display = ("host", "remote_user")
admin.site.register(RemoteUser, RemoteUserAdmin)
admin.site.register(Hosts, HostsAdmin)
admin.site.register(HostGroup, HostGroupAdmin)
admin.site.register(HostToRemoteUser, HostToRemoteUserAdmin)
admin.site.register(IDC, IDCAdmin)