REDROOM

Navigation

Breadcrumbs

Create

Edit File Size: 1.16 KB

/proc/thread-self/root/opt/imunify360/venv/lib64/python3.11/site-packages/defence360agent/model/icontact.py

Text Base64

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()

Exit & Reset Text mode: syntax highlighting auto-detects file type.

Name	Size	Perms	Modified	Actions
__pycache__ DIR	-	drwxr-xr-x	2026-06-08 20:23:14	Edit Download recursive recursive
analyst_cleanup.py	3.40 KB	lrw-r--r--	2026-05-26 21:20:44	Edit Download recursive recursive
event_hook.py	1.74 KB	lrw-r--r--	2026-05-26 21:20:44	Edit Download recursive recursive
icontact.py	1.16 KB	lrw-r--r--	2026-05-26 21:20:44	Edit Download recursive recursive
infected_domain.py	4.24 KB	lrw-r--r--	2026-05-26 21:20:44	Edit Download recursive recursive
instance.py	537 B	lrw-r--r--	2026-05-26 21:20:44	Edit Download recursive recursive
messages_to_send.py	1.37 KB	lrw-r--r--	2026-05-26 21:20:44	Edit Download recursive recursive
simplification.py	5.01 KB	lrw-r--r--	2026-05-26 21:20:44	Edit Download recursive recursive
tls_check.py	2.23 KB	lrw-r--r--	2026-05-26 21:20:44	Edit Download recursive recursive
wordpress.py	1.66 KB	lrw-r--r--	2026-05-26 21:20:44	Edit Download recursive recursive
wordpress_incident.py	13.02 KB	lrw-r--r--	2026-05-26 21:20:44	Edit Download recursive recursive
wp_disabled_rule.py	13.21 KB	lrw-r--r--	2026-05-26 21:20:44	Edit Download recursive recursive
__init__.py	963 B	lrw-r--r--	2026-05-26 21:20:44	Edit Download recursive recursive

Directory Contents