import datetime
import logging
import os
import random
import time
from typing import List, Dict, Any
# Configure logging
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
# Dummy data generation functions (replace with actual database queries)
def generate_dummy_cost_data_by_provider() -> Dict[str, float]:
"""Generates dummy cost data by provider."""
providers = ["AWS", "Azure", "GCP", "Other"]
return {provider: round(random.uniform(100, 1000), 2) for provider in providers}
def generate_dummy_cost_data_by_day(days: int = 30) -> Dict[str, float]:
"""Generates dummy cost data by day for the last 'days' days."""
today = datetime.date.today()
data = {}
for i in range(days):
date = today - datetime.timedelta(days=i)
data[date.strftime("%Y-%m-%d")] = round(random.uniform(50, 300), 2)
return data
def generate_dummy_top_costs(num_costs: int = 5) -> List[Dict[str, Any]]:
"""Generates dummy top costs."""
costs = []
for i in range(num_costs):
costs.append({
"resource": f"Resource-{i+1}",
"cost": round(random.uniform(20, 500), 2),
"provider": random.choice(["AWS", "Azure", "GCP"])
})
return costs
class DashboardGenerator:
"""
Generates a static HTML dashboard showing cost breakdown using dummy data.
"""
def __init__(self, output_file: str = "/mnt/e/genesis-system/apps/dashboard/cost_dashboard.html"):
"""
Initializes the DashboardGenerator with the output file path.
Args:
output_file: The path to the HTML file to be generated.
"""
self.output_file = output_file
self.logger = logging.getLogger(__name__) # Use class-specific logger
def generate_html(self) -> None:
"""
Generates the HTML content for the cost dashboard using dummy data and saves it to the output file.
"""
try:
cost_data_by_provider = generate_dummy_cost_data_by_provider()
cost_data_by_day = generate_dummy_cost_data_by_day()
top_costs = generate_dummy_top_costs()
html_content = self._build_html(cost_data_by_provider, cost_data_by_day, top_costs)
with open(self.output_file, "w") as f:
f.write(html_content)
self.logger.info(f"Dashboard generated successfully at {self.output_file}")
except Exception as e:
self.logger.error(f"Error generating dashboard: {e}", exc_info=True)
raise
def _build_html(self, cost_data_by_provider: Dict[str, float], cost_data_by_day: Dict[str, float], top_costs: List[Dict[str, Any]]) -> str:
"""
Builds the HTML content for the dashboard.
Args:
cost_data_by_provider: Cost data by provider.
cost_data_by_day: Cost data by day.
top_costs: A list of top costs.
Returns:
The HTML content as a string.
"""
html = f"""
Cost Dashboard
Cost Dashboard
Cost by Provider
Cost by Day
Top Costs
| Resource |
Cost |
Provider |
{''.join(f'| {cost["resource"]} | {cost["cost"]} | {cost["provider"]} |
' for cost in top_costs)}
"""
return html
# Example usage (for testing)
if __name__ == "__main__":
generator = DashboardGenerator()
generator.generate_html()