Current File : /opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/contracts//eula.py
import asyncio
import os.path
from typing import Optional
from defence360agent import files
from defence360agent.contracts.config import ANTIVIRUS_MODE
from defence360agent.model.simplification import Eula, run_in_executor
_MESSAGE_TEMPLATE = "message{}.txt"
_SUFFIX = "-av" if ANTIVIRUS_MODE else ""
_TEXT_TEMPLATE = "eula{}.txt"
_UPDATED_TEMPLATE = "updated{}.txt"
def _readfile(path: str, errors: Optional[str] = None) -> str:
with open(path, errors=errors) as f:
return f.read().strip()
def _get_path(template: str) -> str:
return os.path.join(
files.Index.files_path(files.EULA), template.format(_SUFFIX)
)
async def is_accepted() -> bool:
"""Return True if latest EULA was accepted, False otherwise."""
return await run_in_executor(asyncio.get_event_loop(), Eula.is_accepted)
async def accept() -> None:
"""Accepts EULA."""
await run_in_executor(asyncio.get_event_loop(), Eula.accept)
async def update() -> None:
"""Updates latest EULA date from files."""
await run_in_executor(
asyncio.get_event_loop(), lambda: Eula.get_or_create(updated=updated())
)
def text() -> str:
"""Return main text of the EULA."""
return _readfile(_get_path(_TEXT_TEMPLATE), errors="ignore")
def message() -> str:
"""Return a message inviting to accept EULA."""
return _readfile(_get_path(_MESSAGE_TEMPLATE))
def updated() -> str:
"""Return last EULA's update time."""
return _readfile(_get_path(_UPDATED_TEMPLATE))
