jumpserver/apps/assets/backends/db.py

# -*- coding: utf-8 -*-
#

from ..models import AuthBook
from .base import BaseBackend


class AuthBookBackend(BaseBackend):
    @classmethod
    def filter(cls, username=None, assets=None, latest=True, **kwargs):
        queryset = AuthBook.objects.all()
        if username is not None:
            queryset = queryset.filter(username=username)
        if assets:
            queryset = queryset.filter(asset__in=assets)
        if latest:
            queryset = queryset.latest_version()
        return queryset

    @classmethod
    def create(cls, **kwargs):
        auth_info = {
            'password': kwargs.pop('password', ''),
            'public_key': kwargs.pop('public_key', ''),
            'private_key': kwargs.pop('private_key', '')
        }
        obj = AuthBook.objects.create(**kwargs)
        obj.set_auth(**auth_info)
        return obj