Update models.py

pull/707/head
mjzhang95 2025-04-06 19:09:14 +08:00 committed by GitHub
parent 64e15935a2
commit 47bcaaa0b5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 0 deletions

View File

@ -19,6 +19,7 @@ class Host(models.Model, ModelMixin):
is_verified = models.BooleanField(default=False) is_verified = models.BooleanField(default=False)
created_at = models.CharField(max_length=20, default=human_datetime) created_at = models.CharField(max_length=20, default=human_datetime)
created_by = models.ForeignKey(User, models.PROTECT, related_name='+') created_by = models.ForeignKey(User, models.PROTECT, related_name='+')
connect_type = models.CharField(max_length=20, default='ssh')
@property @property
def private_key(self): def private_key(self):
@ -28,6 +29,12 @@ class Host(models.Model, ModelMixin):
pkey = pkey or self.private_key pkey = pkey or self.private_key
return SSH(self.hostname, self.port, self.username, pkey, default_env=default_env) return SSH(self.hostname, self.port, self.username, pkey, default_env=default_env)
def get_telnet(self, password=None):
"""获取telnet连接实例"""
from libs.telnet import Telnet
port = self.port or 23 # telnet默认端口23
return Telnet(self.hostname, port, self.username, password)
def to_view(self): def to_view(self):
tmp = self.to_dict() tmp = self.to_dict()
if hasattr(self, 'hostextend'): if hasattr(self, 'hostextend'):