mirror of
https://github.com/infinition/Bjorn.git
synced 2024-11-11 22:38:39 +03:00
72 lines
3.6 KiB
Python
72 lines
3.6 KiB
Python
# comment.py
|
|
# This module defines the `Commentaireia` class, which provides context-based random comments.
|
|
# The comments are based on various themes such as "IDLE", "SCANNER", and others, to simulate
|
|
# different states or actions within a network scanning and security context. The class uses a
|
|
# shared data object to determine delays between comments and switches themes based on the current
|
|
# state. The `get_commentaire` method returns a random comment from the specified theme, ensuring
|
|
# comments are not repeated too frequently.
|
|
|
|
import random
|
|
import time
|
|
import logging
|
|
import json
|
|
from init_shared import shared_data
|
|
from logger import Logger
|
|
import os
|
|
|
|
logger = Logger(name="comment.py", level=logging.DEBUG)
|
|
|
|
class Commentaireia:
|
|
"""Provides context-based random comments for bjorn."""
|
|
def __init__(self):
|
|
self.shared_data = shared_data
|
|
self.last_comment_time = 0 # Initialize last_comment_time
|
|
self.comment_delay = random.randint(self.shared_data.comment_delaymin, self.shared_data.comment_delaymax) # Initialize comment_delay
|
|
self.last_theme = None # Initialize last_theme
|
|
self.themes = self.load_comments(self.shared_data.commentsfile) # Load themes from JSON file
|
|
|
|
def load_comments(self, commentsfile):
|
|
"""Load comments from a JSON file."""
|
|
cache_file = commentsfile + '.cache'
|
|
|
|
# Check if a cached version exists and is newer than the original file
|
|
if os.path.exists(cache_file) and os.path.getmtime(cache_file) >= os.path.getmtime(commentsfile):
|
|
try:
|
|
with open(cache_file, 'r') as file:
|
|
comments_data = json.load(file)
|
|
logger.info("Comments loaded successfully from cache.")
|
|
return comments_data
|
|
except (FileNotFoundError, json.JSONDecodeError):
|
|
logger.warning("Cache file is corrupted or not found. Loading from the original file.")
|
|
|
|
# Load from the original file if cache is not used or corrupted
|
|
try:
|
|
with open(commentsfile, 'r') as file:
|
|
comments_data = json.load(file)
|
|
logger.info("Comments loaded successfully from JSON file.")
|
|
# Save to cache
|
|
with open(cache_file, 'w') as cache:
|
|
json.dump(comments_data, cache)
|
|
return comments_data
|
|
except FileNotFoundError:
|
|
logger.error(f"The file '{commentsfile}' was not found.")
|
|
return {"IDLE": ["Default comment, no comments file found."]} # Fallback to a default theme
|
|
except json.JSONDecodeError:
|
|
logger.error(f"The file '{commentsfile}' is not a valid JSON file.")
|
|
return {"IDLE": ["Default comment, invalid JSON format."]} # Fallback to a default theme
|
|
|
|
def get_commentaire(self, theme):
|
|
""" This method returns a random comment based on the specified theme."""
|
|
current_time = time.time() # Get the current time in seconds
|
|
if theme != self.last_theme or current_time - self.last_comment_time >= self.comment_delay: # Check if the theme has changed or if the delay has expired
|
|
self.last_comment_time = current_time # Update the last comment time
|
|
self.last_theme = theme # Update the last theme
|
|
|
|
if theme not in self.themes:
|
|
logger.warning(f"The theme '{theme}' is not defined, using the default theme IDLE.")
|
|
theme = "IDLE"
|
|
|
|
return random.choice(self.themes[theme]) # Return a random comment based on the specified theme
|
|
else:
|
|
return None
|