X7ROOT File Manager
Current Path:
/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/model
opt
/
imunify360
/
venv
/
lib
/
python3.11
/
site-packages
/
defence360agent
/
model
/
📁
..
📄
__init__.py
(963 B)
📁
__pycache__
📄
analyst_cleanup.py
(3.4 KB)
📄
event_hook.py
(1.74 KB)
📄
icontact.py
(1.16 KB)
📄
infected_domain.py
(4.24 KB)
📄
instance.py
(481 B)
📄
messages_to_send.py
(1.37 KB)
📄
simplification.py
(5.01 KB)
📄
tls_check.py
(1.29 KB)
Editing: icontact.py
import time from peewee import CharField, IntegerField, CompositeKey from defence360agent.contracts.config import IContactMessageType from defence360agent.model import Model, instance from defence360agent.utils.common import DAY, WEEK THROTTLING_PERIOD = { IContactMessageType.MALWARE_FOUND: DAY, IContactMessageType.SCAN_NOT_SCHEDULED: WEEK, } class IContactThrottle(Model): class Meta: database = instance.db db_table = "icontact_throttle" primary_key = CompositeKey("message_type", "user") message_type = CharField() user = CharField(null=True) #: The last time we sent a notification about :attr:`message_type` timestamp = IntegerField(default=0) @classmethod def may_be_notified(cls, message_type, period_limit, user=None): obj, _ = cls.get_or_create(message_type=message_type, user=user) return (time.time() - obj.timestamp) > period_limit @classmethod def refresh(cls, message_type, user=None): cls.update(timestamp=time.time()).where( cls.message_type == message_type, cls.user.is_null(True) if user is None else cls.user == user, ).execute()
Upload File
Create Folder