mirror of
https://github.com/jumpserver/jumpserver.git
synced 2026-04-09 13:53:20 +00:00
3b615719fe
* feat: 账号密钥用vault储存 * perf: 优化 Vault * perf: 重构 Vault Backend 设计架构 (未完成) * perf: 重构 Vault Backend 设计架构 (未完成2) * perf: 重构 Vault Backend 设计架构 (未完成3) * perf: 重构 Vault Backend 设计架构 (未完成4) * perf: 重构 Vault Backend 设计架构 (未完成5) * perf: 重构 Vault Backend 设计架构 (已完成) * perf: 重构 Vault Backend 设计架构 (已完成) * perf: 重构 Vault Backend 设计架构 (已完成) * perf: 小优化 * perf: 优化 --------- Co-authored-by: feng <1304903146@qq.com> Co-authored-by: Bai <baijiangjie@gmail.com> Co-authored-by: feng626 <57284900+feng626@users.noreply.github.com>
78 lines
2.0 KiB
Python
78 lines
2.0 KiB
Python
from abc import ABC, abstractmethod
|
|
|
|
from django.forms.models import model_to_dict
|
|
|
|
__all__ = ['BaseVault']
|
|
|
|
|
|
class BaseVault(ABC):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
self.type = kwargs.get('VAULT_TYPE')
|
|
|
|
def is_type(self, tp):
|
|
return self.type == tp
|
|
|
|
def get(self, instance):
|
|
""" 返回 secret 值 """
|
|
return self._get(instance)
|
|
|
|
def create(self, instance):
|
|
if not instance.secret_has_save_to_vault:
|
|
self._create(instance)
|
|
self._clean_db_secret(instance)
|
|
self.save_metadata(instance)
|
|
|
|
if instance.is_sync_metadata:
|
|
self.save_metadata(instance)
|
|
|
|
def update(self, instance):
|
|
if not instance.secret_has_save_to_vault:
|
|
self._update(instance)
|
|
self._clean_db_secret(instance)
|
|
self.save_metadata(instance)
|
|
|
|
if instance.is_sync_metadata:
|
|
self.save_metadata(instance)
|
|
|
|
def delete(self, instance):
|
|
self._delete(instance)
|
|
|
|
def save_metadata(self, instance):
|
|
metadata = model_to_dict(instance, fields=[
|
|
'name', 'username', 'secret_type',
|
|
'connectivity', 'su_from', 'privileged'
|
|
])
|
|
metadata = {field: str(value) for field, value in metadata.items()}
|
|
return self._save_metadata(instance, metadata)
|
|
|
|
# -------- abstractmethod -------- #
|
|
|
|
@abstractmethod
|
|
def _get(self, instance):
|
|
raise NotImplementedError
|
|
|
|
@abstractmethod
|
|
def _create(self, instance):
|
|
raise NotImplementedError
|
|
|
|
@abstractmethod
|
|
def _update(self, instance):
|
|
raise NotImplementedError
|
|
|
|
@abstractmethod
|
|
def _delete(self, instance):
|
|
raise NotImplementedError
|
|
|
|
@abstractmethod
|
|
def _clean_db_secret(self, instance):
|
|
raise NotImplementedError
|
|
|
|
@abstractmethod
|
|
def _save_metadata(self, instance, metadata):
|
|
raise NotImplementedError
|
|
|
|
@abstractmethod
|
|
def is_active(self, *args, **kwargs) -> (bool, str):
|
|
raise NotImplementedError
|