Retropex/custom-ocean.xyz-dashboard
1
0
Fork 0
mirror of https://github.com/Retropex/custom-ocean.xyz-dashboard.git synced 2025-05-12 19:20:45 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update notification_service.py

Browse Source
This commit is contained in:
DJObleezy 2025-04-02 13:39:33 -07:00 committed by GitHub
parent 353c9567bb
commit 1416cf2bf5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 70 additions and 90 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

160
notification_service.py
View File

@ -31,6 +31,8 @@ class NotificationService:
self.last_daily_stats = None self.last_daily_stats = None
self.max_notifications = 100 # Maximum number to store self.max_notifications = 100 # Maximum number to store
self.last_block_height = None # Track the last seen block height self.last_block_height = None # Track the last seen block height
self.last_payout_notification_time = None # Track the last payout notification time
self.last_estimated_payout_time = None # Track the last estimated payout time
# Load existing notifications from state # Load existing notifications from state
self._load_notifications() self._load_notifications()
@ -227,22 +229,15 @@ class NotificationService:
def check_and_generate_notifications(self, current_metrics, previous_metrics): def check_and_generate_notifications(self, current_metrics, previous_metrics):
""" """
Check metrics and generate notifications for significant events. Check metrics and generate notifications for significant events.
Args:
current_metrics (dict): Current mining metrics
previous_metrics (dict): Previous mining metrics
Returns:
list: Newly created notifications
""" """
new_notifications = [] new_notifications = []
try: try:
# Skip if no metrics # Skip if no metrics
if not current_metrics: if not current_metrics:
logging.warning("No current metrics available, skipping notification checks") logging.warning("No current metrics available, skipping notification checks")
return new_notifications return new_notifications
# Check for block updates (using persistent storage) # Check for block updates (using persistent storage)
last_block_height = current_metrics.get("last_block_height") last_block_height = current_metrics.get("last_block_height")
if last_block_height and last_block_height != "N/A": if last_block_height and last_block_height != "N/A":
@ -251,12 +246,12 @@ class NotificationService:
block_notification = self._generate_block_notification(current_metrics) block_notification = self._generate_block_notification(current_metrics)
if block_notification: if block_notification:
new_notifications.append(block_notification) new_notifications.append(block_notification)
# Always update the stored last block height when it changes # Always update the stored last block height when it changes
if self.last_block_height != last_block_height: if self.last_block_height != last_block_height:
self.last_block_height = last_block_height self.last_block_height = last_block_height
self._save_last_block_height() self._save_last_block_height()
# Regular comparison with previous metrics # Regular comparison with previous metrics
if previous_metrics: if previous_metrics:
# Check for daily stats # Check for daily stats
@ -264,24 +259,19 @@ class NotificationService:
stats_notification = self._generate_daily_stats(current_metrics) stats_notification = self._generate_daily_stats(current_metrics)
if stats_notification: if stats_notification:
new_notifications.append(stats_notification) new_notifications.append(stats_notification)
# Check for significant hashrate drop # Check for significant hashrate drop
hashrate_notification = self._check_hashrate_change(current_metrics, previous_metrics) hashrate_notification = self._check_hashrate_change(current_metrics, previous_metrics)
if hashrate_notification: if hashrate_notification:
new_notifications.append(hashrate_notification) new_notifications.append(hashrate_notification)
# Check for worker status changes
worker_notification = self._check_worker_status_change(current_metrics, previous_metrics)
if worker_notification:
new_notifications.append(worker_notification)
# Check for earnings and payout progress # Check for earnings and payout progress
earnings_notification = self._check_earnings_progress(current_metrics, previous_metrics) earnings_notification = self._check_earnings_progress(current_metrics, previous_metrics)
if earnings_notification: if earnings_notification:
new_notifications.append(earnings_notification) new_notifications.append(earnings_notification)
return new_notifications return new_notifications
except Exception as e: except Exception as e:
logging.error(f"Error generating notifications: {e}") logging.error(f"Error generating notifications: {e}")
error_notification = self.add_notification( error_notification = self.add_notification(
@ -292,23 +282,27 @@ class NotificationService:
return [error_notification] return [error_notification]
def _should_post_daily_stats(self): def _should_post_daily_stats(self):
"""Check if it's time to post daily stats every 6 hours.""" """Check if it's time to post daily stats based on flexible conditions."""
now = datetime.now() now = datetime.now()
# Define the times to post daily stats (every 6 hours) # Define the times to post daily stats (every 6 hours by default)
daily_stats_times = ["00:00:00", "06:00:00", "12:00:00", "18:00:00"] daily_stats_times = ["00:00:00", "06:00:00", "12:00:00", "18:00:00"]
# Customizable interval in hours (default to 6 hours)
interval_hours = 6
# If we have a last_daily_stats timestamp # If we have a last_daily_stats timestamp
if self.last_daily_stats: if self.last_daily_stats:
# Check time elapsed since last stats # Check time elapsed since last stats
time_since_last_stats = now - self.last_daily_stats time_since_last_stats = now - self.last_daily_stats
# If more than 6 hours have passed # If more than the interval has passed
if time_since_last_stats.total_seconds() >= 6 * 3600: if time_since_last_stats.total_seconds() >= interval_hours * 3600:
current_time_str = now.strftime("%H:%M:%S") current_time_str = now.strftime("%H:%M:%S")
# Check if current time is one of our scheduled times # Check if current time is one of our scheduled times
if current_time_str >= "00:00:00" and current_time_str in daily_stats_times: if current_time_str in daily_stats_times:
logging.info(f"Posting daily stats at {current_time_str}")
self.last_daily_stats = now self.last_daily_stats = now
return True return True
return False return False
@ -316,6 +310,7 @@ class NotificationService:
# First time - post if current time is one of our scheduled times # First time - post if current time is one of our scheduled times
current_time_str = now.strftime("%H:%M:%S") current_time_str = now.strftime("%H:%M:%S")
if current_time_str in daily_stats_times: if current_time_str in daily_stats_times:
logging.info(f"First time posting daily stats at {current_time_str}")
self.last_daily_stats = now self.last_daily_stats = now
return True return True
return False return False
@ -324,6 +319,7 @@ class NotificationService:
"""Generate daily stats notification.""" """Generate daily stats notification."""
try: try:
if not metrics: if not metrics:
logging.warning("No metrics available for daily stats")
return None return None
# Format hashrate with appropriate unit # Format hashrate with appropriate unit
@ -338,6 +334,7 @@ class NotificationService:
message = f"Daily Mining Summary: {hashrate_24hr} {hashrate_unit} average hashrate, {daily_mined_sats} sats mined (${daily_profit_usd:.2f})" message = f"Daily Mining Summary: {hashrate_24hr} {hashrate_unit} average hashrate, {daily_mined_sats} sats mined (${daily_profit_usd:.2f})"
# Add notification # Add notification
logging.info(f"Generating daily stats notification: {message}")
return self.add_notification( return self.add_notification(
message, message,
level=NotificationLevel.INFO, level=NotificationLevel.INFO,
@ -381,25 +378,41 @@ class NotificationService:
try: try:
current_3hr = current.get("hashrate_3hr", 0) current_3hr = current.get("hashrate_3hr", 0)
previous_3hr = previous.get("hashrate_3hr", 0) previous_3hr = previous.get("hashrate_3hr", 0)
# Log what we're comparing
logging.debug(f"Comparing hashrates - current: {current_3hr}, previous: {previous_3hr}")
# Skip if values are missing # Skip if values are missing
if not current_3hr or not previous_3hr: if not current_3hr or not previous_3hr:
logging.debug("Skipping hashrate check - missing values")
return None return None
# Convert to float # Handle strings with units (e.g., "10.5 TH/s")
current_3hr = float(current_3hr) if isinstance(current_3hr, str):
previous_3hr = float(previous_3hr) current_3hr = float(current_3hr.split()[0])
else:
current_3hr = float(current_3hr)
if isinstance(previous_3hr, str):
previous_3hr = float(previous_3hr.split()[0])
else:
previous_3hr = float(previous_3hr)
logging.debug(f"Converted hashrates - current: {current_3hr}, previous: {previous_3hr}")
# Skip if previous was zero (prevents division by zero) # Skip if previous was zero (prevents division by zero)
if previous_3hr == 0: if previous_3hr == 0:
logging.debug("Skipping hashrate check - previous was zero")
return None return None
# Calculate percentage change # Calculate percentage change
percent_change = ((current_3hr - previous_3hr) / previous_3hr) * 100 percent_change = ((current_3hr - previous_3hr) / previous_3hr) * 100
logging.debug(f"Hashrate change: {percent_change:.1f}%")
# Significant decrease (more than 15%) # Significant decrease (more than 15%)
if percent_change <= -15: if percent_change <= -15:
message = f"Significant hashrate drop detected: {abs(percent_change):.1f}% decrease" message = f"Significant hashrate drop detected: {abs(percent_change):.1f}% decrease"
logging.info(f"Generating hashrate notification: {message}")
return self.add_notification( return self.add_notification(
message, message,
level=NotificationLevel.WARNING, level=NotificationLevel.WARNING,
@ -410,10 +423,11 @@ class NotificationService:
"change": percent_change "change": percent_change
} }
) )
# Significant increase (more than 15%) # Significant increase (more than 15%)
elif percent_change >= 15: elif percent_change >= 15:
message = f"Hashrate increase detected: {percent_change:.1f}% increase" message = f"Hashrate increase detected: {percent_change:.1f}% increase"
logging.info(f"Generating hashrate notification: {message}")
return self.add_notification( return self.add_notification(
message, message,
level=NotificationLevel.SUCCESS, level=NotificationLevel.SUCCESS,
@ -424,57 +438,12 @@ class NotificationService:
"change": percent_change "change": percent_change
} }
) )
return None return None
except Exception as e: except Exception as e:
logging.error(f"Error checking hashrate change: {e}") logging.error(f"Error checking hashrate change: {e}")
return None return None
def _check_worker_status_change(self, current, previous):
"""Check for worker status changes."""
try:
current_workers = current.get("workers_hashing", 0)
previous_workers = previous.get("workers_hashing", 0)
# Skip if values are missing
if current_workers is None or previous_workers is None:
return None
# Worker(s) went offline
if current_workers < previous_workers:
diff = previous_workers - current_workers
message = f"{diff} worker{'s' if diff > 1 else ''} went offline"
return self.add_notification(
message,
level=NotificationLevel.WARNING,
category=NotificationCategory.WORKER,
data={
"previous_count": previous_workers,
"current_count": current_workers,
"difference": diff
}
)
# Worker(s) came online
elif current_workers > previous_workers:
diff = current_workers - previous_workers
message = f"{diff} new worker{'s' if diff > 1 else ''} came online"
return self.add_notification(
message,
level=NotificationLevel.SUCCESS,
category=NotificationCategory.WORKER,
data={
"previous_count": previous_workers,
"current_count": current_workers,
"difference": diff
}
)
return None
except Exception as e:
logging.error(f"Error checking worker status change: {e}")
return None
def _check_earnings_progress(self, current, previous): def _check_earnings_progress(self, current, previous):
"""Check for significant earnings progress or payout approach.""" """Check for significant earnings progress or payout approach."""
try: try:
@ -488,22 +457,26 @@ class NotificationService:
if est_time.isdigit() or (est_time[0] == '-' and est_time[1:].isdigit()): if est_time.isdigit() or (est_time[0] == '-' and est_time[1:].isdigit()):
days = int(est_time) days = int(est_time)
if 0 < days <= 1: if 0 < days <= 1:
message = f"Payout approaching! Estimated within 1 day" if self._should_send_payout_notification():
message = f"Payout approaching! Estimated within 1 day"
self.last_payout_notification_time = datetime.now()
return self.add_notification(
message,
level=NotificationLevel.SUCCESS,
category=NotificationCategory.EARNINGS,
data={"days_to_payout": days}
)
# If it says "next block"
elif "next block" in est_time.lower():
if self._should_send_payout_notification():
message = f"Payout expected with next block!"
self.last_payout_notification_time = datetime.now()
return self.add_notification( return self.add_notification(
message, message,
level=NotificationLevel.SUCCESS, level=NotificationLevel.SUCCESS,
category=NotificationCategory.EARNINGS, category=NotificationCategory.EARNINGS,
data={"days_to_payout": days} data={"payout_imminent": True}
) )
# If it says "next block"
elif "next block" in est_time.lower():
message = f"Payout expected with next block!"
return self.add_notification(
message,
level=NotificationLevel.SUCCESS,
category=NotificationCategory.EARNINGS,
data={"payout_imminent": True}
)
# Check for payout (unpaid balance reset) # Check for payout (unpaid balance reset)
if previous.get("unpaid_earnings"): if previous.get("unpaid_earnings"):
@ -527,3 +500,10 @@ class NotificationService:
except Exception as e: except Exception as e:
logging.error(f"Error checking earnings progress: {e}") logging.error(f"Error checking earnings progress: {e}")
return None return None
def _should_send_payout_notification(self):
"""Check if enough time has passed since the last payout notification."""
if self.last_payout_notification_time is None:
return True
time_since_last_notification = datetime.now() - self.last_payout_notification_time
return time_since_last_notification.total_seconds() > 86400 # 1 Day