%PDF-
%PDF-
Mini Shell
Mini Shell
"""
This program is free software: you can redistribute it and/or modify it under
the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License,
or (at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
See the GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <https://www.gnu.org/licenses/>.
Copyright © 2019 Cloud Linux Software Inc.
This software is also available under ImunifyAV commercial license,
see <https://www.imunify360.com/legal/eula>
"""
from peewee import CharField, Check, TextField
from defence360agent.feature_management.constants import (
AV,
AV_REPORT,
FULL,
NA,
PROACTIVE,
)
from defence360agent.model import Model, instance
class FeatureManagementPerms(Model):
"""Permissions state for Feature Management.
Each record/instance is a set of permissions of a single user.
"""
DEFAULT = ''
class Meta:
database = instance.db
db_table = 'feature_management_permissions'
#: The username of the end-user, or an empty string for the default value
#: for all new users.
user = CharField(unique=True)
#: How much the user can access and control Proactive Defense feature.
#: Must be either :obj:`.NA` or :obj:`.FULL`.
proactive = TextField(
null=False,
constraints=[Check("proactive in ('{}','{}')".format(NA, FULL))],
default=FULL
)
#: How much the user can access and control Proactive Defense feature.
#: Must be either :obj:`.NA` or :obj:`.AV_REPORT` or :obj:`.FULL`.
av = TextField(
null=False,
constraints=[Check("av in ('{}','{}','{}')".format(
NA, AV_REPORT, FULL))],
default=AV_REPORT
)
@classmethod
def get_perm(cls, user: str) -> 'FeatureManagementPerms':
"""
Get feature permissions by user name
:param user: user name
:return: :class:`FeatureManagementPerms` object for user
"""
default = cls.get_default()
if user is None:
return default
defaults = {
AV: default.av,
PROACTIVE: default.proactive,
}
perm, _ = cls.get_or_create(user=user, defaults=defaults)
return perm
@classmethod
def get_default(cls):
"""Get default permissions"""
return cls.get(user=cls.DEFAULT)
def is_default(self):
"""Check if current permission is default"""
return self.user == self.DEFAULT
def get_feature(self, key: str) -> str:
"""Get permission by feature name"""
return getattr(self, key)
def set_feature(self, key: str, value: str):
"""Set permission"""
setattr(self, key, value)
self.save()
def as_dict(self):
return {
AV: self.av,
PROACTIVE: self.proactive,
}
Zerion Mini Shell 1.0