mirror of https://github.com/jumpserver/jumpserver
46 lines
1.1 KiB
Python
46 lines
1.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
from common.db.fields import BitChoices
|
|
from common.utils.integer import bit
|
|
|
|
__all__ = ["ActionChoices"]
|
|
|
|
|
|
class ActionChoices(BitChoices):
|
|
connect = bit(1), _("Connect")
|
|
upload = bit(2), _("Upload")
|
|
download = bit(3), _("Download")
|
|
copy = bit(4), _("Copy")
|
|
paste = bit(5), _("Paste")
|
|
|
|
@classmethod
|
|
def is_tree(cls):
|
|
return True
|
|
|
|
@classmethod
|
|
def branches(cls):
|
|
return (
|
|
cls.connect,
|
|
(_("Transfer"), [cls.upload, cls.download]),
|
|
(_("Clipboard"), [cls.copy, cls.paste]),
|
|
)
|
|
|
|
@classmethod
|
|
def transfer(cls):
|
|
return cls.upload | cls.download
|
|
|
|
@classmethod
|
|
def clipboard(cls):
|
|
return cls.copy | cls.paste
|
|
|
|
@classmethod
|
|
def contains(cls, total, action):
|
|
action_value = getattr(cls, action)
|
|
return action_value & total == action_value
|
|
|
|
@classmethod
|
|
def display(cls, value):
|
|
return ', '.join([str(c.label) for c in cls if c.value & value == c.value])
|