mirror of https://github.com/jumpserver/jumpserver
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
100 lines
3.0 KiB
100 lines
3.0 KiB
import datetime
|
|
from django.db import models
|
|
# from juser.models import User, UserGroup
|
|
|
|
|
|
class AssetGroup(models.Model):
|
|
GROUP_TYPE = (
|
|
('P', 'PRIVATE'),
|
|
('A', 'ASSET'),
|
|
)
|
|
name = models.CharField(max_length=80, unique=True)
|
|
comment = models.CharField(max_length=160, blank=True, null=True)
|
|
|
|
def __unicode__(self):
|
|
return self.name
|
|
|
|
# def get_asset(self):
|
|
# return self.asset_set.all()
|
|
#
|
|
# def get_asset_info(self, printable=False):
|
|
# assets = self.get_asset()
|
|
# ip_comment = {}
|
|
# for asset in assets:
|
|
# ip_comment[asset.ip] = asset.comment
|
|
#
|
|
# for ip in sorted(ip_comment):
|
|
# if ip_comment[ip]:
|
|
# print '%-15s -- %s' % (ip, ip_comment[ip])
|
|
# else:
|
|
# print '%-15s' % ip
|
|
# print ''
|
|
#
|
|
# def get_asset_num(self):
|
|
# return len(self.get_asset())
|
|
#
|
|
# def get_user_group(self):
|
|
# perm_list = self.perm_set.all()
|
|
# user_group_list = []
|
|
# for perm in perm_list:
|
|
# user_group_list.append(perm.user_group)
|
|
# return user_group_list
|
|
#
|
|
# def get_user(self):
|
|
# user_list = []
|
|
# user_group_list = self.get_user_group()
|
|
# for user_group in user_group_list:
|
|
# user_list.extend(user_group.user_set.all())
|
|
# return user_list
|
|
#
|
|
# def is_permed(self, user=None, user_group=None):
|
|
# if user:
|
|
# if user in self.get_user():
|
|
# return True
|
|
#
|
|
# if user_group:
|
|
# if user_group in self.get_user_group():
|
|
# return True
|
|
# return False
|
|
|
|
|
|
class Asset(models.Model):
|
|
ip = models.GenericIPAddressField(unique=True)
|
|
port = models.IntegerField()
|
|
group = models.ManyToManyField(AssetGroup)
|
|
username = models.CharField(max_length=20, blank=True, null=True)
|
|
password = models.CharField(max_length=80, blank=True, null=True)
|
|
use_default_auth = models.BooleanField(default=True)
|
|
date_added = models.DateTimeField(auto_now_add=True)
|
|
is_active = models.BooleanField(default=True)
|
|
comment = models.CharField(max_length=100, blank=True, null=True)
|
|
|
|
def __unicode__(self):
|
|
return self.ip
|
|
|
|
# def get_user(self):
|
|
# perm_list = []
|
|
# asset_group_all = self.bis_group.all()
|
|
# for asset_group in asset_group_all:
|
|
# perm_list.extend(asset_group.perm_set.all())
|
|
#
|
|
# user_group_list = []
|
|
# for perm in perm_list:
|
|
# user_group_list.append(perm.user_group)
|
|
#
|
|
# user_permed_list = []
|
|
# for user_group in user_group_list:
|
|
# user_permed_list.extend(user_group.user_set.all())
|
|
# user_permed_list = list(set(user_permed_list))
|
|
# return user_permed_list
|
|
|
|
|
|
class AssetAlias(models.Model):
|
|
pass
|
|
# user = models.ForeignKey(User)
|
|
# asset = models.ForeignKey(Asset)
|
|
# alias = models.CharField(max_length=100, blank=True, null=True)
|
|
#
|
|
# def __unicode__(self):
|
|
# return self.alias
|