adds data analysis

This commit is contained in:
Michael Beck
2025-02-08 19:12:38 +01:00
parent afd9ec1c5c
commit 2615972566
5 changed files with 178 additions and 3 deletions

60
app/analysis.py Normal file
View File

@@ -0,0 +1,60 @@
import os
import pandas as pd
import matplotlib
matplotlib.use("Agg") # Prevents GUI-related issues in Flask
import matplotlib.pyplot as plt
import seaborn as sns
def load_data(file_path: str) -> pd.DataFrame:
"""Loads the scraped data from a CSV file into a Pandas DataFrame."""
if not os.path.exists(file_path):
raise FileNotFoundError(f"File {file_path} not found.")
df = pd.read_csv(file_path)
df["timestamp"] = pd.to_datetime(df["timestamp"], errors="coerce")
df["last_action"] = pd.to_datetime(df["last_action"], errors="coerce")
return df
def generate_statistics(df: pd.DataFrame):
"""Generates activity statistics grouped by hour."""
df["hour"] = df["timestamp"].dt.hour
return df.groupby("hour").size()
def plot_activity_distribution(df: pd.DataFrame, output_path="activity_distribution.png"):
"""Plots user activity distribution and saves the figure."""
# Ensure the directory exists
static_dir = os.path.join("app", "static", "plots")
output_path = os.path.join(static_dir, output_path)
os.makedirs(static_dir, exist_ok=True)
# Convert timestamp column to datetime (if not already)
if not pd.api.types.is_datetime64_any_dtype(df["timestamp"]):
df["timestamp"] = pd.to_datetime(df["timestamp"], errors="coerce")
df["hour"] = df["timestamp"].dt.hour
activity_counts = df.groupby("hour").size().reset_index(name="count")
# Use non-GUI backend for Matplotlib
plt.figure(figsize=(10, 5))
# Fix Seaborn Warning: Assign `hue` explicitly
sns.barplot(x="hour", y="count", data=activity_counts, hue="hour", palette="Blues", legend=False)
plt.xlabel("Hour of the Day")
plt.ylabel("Activity Count")
plt.title("User Activity Distribution")
plt.xticks(range(0, 24))
# Save the plot file safely
plt.savefig(output_path, bbox_inches="tight")
plt.close()
# Verify the file exists after saving
if not os.path.exists(output_path):
raise FileNotFoundError(f"Plot could not be saved to {output_path}")
return output_path

View File

@@ -6,6 +6,7 @@ from app.views import register_views
from app.api import register_api
from app.config import load_config
from app.filters import register_filters
from app.analysis import generate_statistics
def init_app():
config = load_config()

View File

@@ -0,0 +1,68 @@
{% extends 'base.html' %}
{% block content %}
<section class="container-fluid d-flex justify-content-center">
<div class="container-md my-5 mx-2 shadow-lg p-4 ">
<div class="container-sm">
<div class="row">
<div class="col">
<h2>User Activity Distribution</h2>
</div>
<div class="col text-end">
<!-- Dropdown for selecting data file -->
<form method="POST" action="{{ url_for('views.data_visualization') }}">
<label for="data_file" class="form-label">Choose Data File:</label>
<select name="data_file" id="data_file" class="form-select" onchange="this.form.submit()">
{% for file in data_files %}
<option value="{{ file }}" {% if file == selected_file %}selected{% endif %}>
{{ file.split('/')[-1] }}
</option>
{% endfor %}
</select>
</form>
</div>
</div>
{% if error %}
<div class="alert alert-danger mt-3" role="alert">
{{ error }}
</div>
{% endif %}
{% if plot_url %}
<div class="row mt-4">
<div class="col">
<h4>Selected File: {{ selected_file.split('/')[-1] }}</h4>
<img src="{{ plot_url }}" class="img-fluid rounded shadow" alt="User Activity Distribution">
</div>
</div>
{% endif %}
{% if statistics %}
<div class="row mt-4">
<div class="col">
<h2>Activity Statistics</h2>
<table class="table table-bordered table-hover">
<thead class="table-dark">
<tr>
<th>Hour</th>
<th>Activity Count</th>
</tr>
</thead>
<tbody>
{% for hour, count in statistics.items() %}
<tr>
<td>{{ hour }}</td>
<td>{{ count }}</td>
</tr>
{% endfor %}
</tbody>
</table>
</div>
</div>
{% endif %}
</div>
</div>
</section>
{% endblock content %}

View File

@@ -3,7 +3,7 @@
<div class="container-fluid">
<a class="navbar-brand" href="/">Torn User Activity Scraper</a>
{% from 'bootstrap4/nav.html' import render_nav_item %}
{{ render_nav_item('analyze', 'Analyze') }}
{{ render_nav_item('views.data_visualization', 'Data Visualization') }}
{{ render_nav_item('download_results', 'Files') }}
{{ render_nav_item('log_viewer', 'Logs') }}
<div class="d-flex" id="color-mode-toggle">

View File

@@ -1,12 +1,14 @@
import os
import glob
from flask import render_template
from flask import render_template, Blueprint, current_app, request
from app.forms import ScrapingForm
from app.util import get_size
from app.config import load_config
from app.api import scraper as scraper# Import the scraper instance
from app.logging_config import get_logger
from app.analysis import load_data, generate_statistics, plot_activity_distribution
from app.state import log_file_name
@@ -15,6 +17,8 @@ print(f"A imported log_file_name: {log_file_name}")
config = load_config()
logger = get_logger()
views_bp = Blueprint("views", __name__)
def register_views(app):
@app.route('/')
def index():
@@ -79,4 +83,46 @@ def register_views(app):
files = {"data": data_files_info, "log": log_files_info}
return render_template('download_results.html', files=files)
return render_template('download_results.html', files=files)
views_bp = Blueprint("views", __name__)
@views_bp.route("/data-visualization", methods=["GET", "POST"])
def data_visualization():
"""Route to display activity statistics with a visualization."""
data_dir = current_app.config["DATA"]["DATA_DIR"]
# Find all available CSV files
data_files = sorted(
glob.glob(os.path.join(data_dir, "*.csv")),
key=os.path.getmtime,
reverse=True
)
if not data_files:
return render_template("data_visualization.html", error="No data files found.", data_files=[])
# Get the selected file from the dropdown (default to the latest file)
selected_file = request.form.get("data_file", data_files[0] if data_files else None)
if selected_file and os.path.exists(selected_file):
df = load_data(selected_file)
statistics = generate_statistics(df)
# ✅ Generate the plot and get the correct URL path
# remove app/ from the base URL
plot_url = plot_activity_distribution(df).replace("app/", "")
else:
return render_template("data_visualization.html", error="Invalid file selection.", data_files=data_files)
return render_template(
"data_visualization.html",
plot_url=plot_url,
statistics=statistics.to_dict(),
data_files=data_files,
selected_file=selected_file
)
app.register_blueprint(views_bp)