add rescuetime stuff

Author: azegas

.env_example

@@ -1 +1,2 @@
-BASE_DIR=/home/username/GIT/mm2/
+BASE_DIR=/home/username/GIT/mm2/
+AG_RESCUETIME_API_KEY=key

app.py

@@ -59,6 +59,13 @@ def read_random_quote():
             return data
     return {"error": "Data not found"}
 
+def read_rescuetime_data():
+    file_path = os.path.join(base_dir, "data/rescuetime_data.json")
+    if os.path.exists(file_path):
+        with open(file_path, "r") as file:
+            data = json.load(file)
+            return data
+    return {"error": "Data not found or invalid format"}
 
 # SocketIO Event Handlers (BACKEND, SERVER)
 # Listening for incoming connections such as `request_from_client_to_server_for_stock_data`
@@ -74,6 +81,7 @@ def event_handler_connect():
     emit('response_from_server_to_client_with_stock_data', read_stock_data())
     emit('response_from_server_to_client_with_cvbankas_data', read_cvbankas_data())
     emit('response_from_server_to_client_with_sensor_data', read_sensor_data())
+    emit('response_from_server_to_client_with_rescuetime_data', read_rescuetime_data())
 
 @socketio.on('request_from_client_to_server_for_stock_data')
 def event_handler_stock():
@@ -95,6 +103,10 @@ def event_handler_system_info():
 def event_handler_random_quote():
     emit('response_from_server_to_client_with_quote_data', read_random_quote())
 
+@socketio.on('request_from_client_to_server_for_rescuetime_data')
+def event_handler_rescuetime():
+    emit('response_from_server_to_client_with_rescuetime_data', read_rescuetime_data())
+
 @app.route("/")
 def home():
     return render_template("index.html")

cron.txt

@@ -23,6 +23,9 @@
 # Run every day at 01:00 AM
 0 1 * * * /home/arvypi/venvs/venv-mm2/bin/python /home/arvypi/GIT/mm2/fetches/random_quote.py
 
+# Run every day at 02:00 AM
+0 2 * * * /home/arvypi/venvs/venv-mm2/bin/python /home/arvypi/GIT/mm2/fetches/rescuetime.py
+
 # Run every minute from 5:00 AM to 6:59 AM
 * 5-6 * * * /home/arvypi/venvs/venv-mm2/bin/python /home/arvypi/GIT/mm2/fetches/invest_fetch_stock_data.py
 * 5-6 * * * /home/arvypi/venvs/venv-mm2/bin/python /home/arvypi/GIT/mm2/fetches/cvbankas.py

fetches/rescuetime.py

@@ -0,0 +1,72 @@
+import requests
+import os
+import sys
+import json
+from datetime import datetime, timedelta
+from dotenv import load_dotenv
+
+# Load environment variables
+load_dotenv()
+
+sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
+from log_config import logger
+
+# Your RescueTime API keys
+api_key_ag = os.getenv("AG_RESCUETIME_API_KEY")
+
+# Pick yesterday's date
+yesterday_full = datetime.now() - timedelta(days=1)
+yesterday_formatted = yesterday_full.strftime("%Y-%m-%d")
+
+def fetch_for_user(api_key, yesterday_formatted):
+    # Define the API endpoint URL
+    api_url = f"https://www.rescuetime.com/anapi/daily_summary_feed?key={api_key}"
+
+    # Make the API request
+    response = requests.get(api_url)
+
+    if response.status_code == 200:
+        data = response.json()
+        
+        # Filter the data for the specific date
+        filtered_data = [entry for entry in data if entry['date'] == yesterday_formatted]
+        
+        if filtered_data:
+            return filtered_data[0]
+        else:
+            logger.warning(f"No data found for {yesterday_formatted}")
+            return None
+    else:
+        logger.error(f"Failed to retrieve data. Status code: {response.status_code}")
+        return None
+
+def save_rescuetime_data(data):
+    data_to_save = {
+        "fetch_date": datetime.now().strftime("%Y-%m-%d %H:%M"),
+        "rescuetime_data": data
+    }
+
+    base_dir = os.getenv("BASE_DIR")
+    file_path = os.path.join(base_dir, "data/rescuetime_data.json")
+
+    os.makedirs(os.path.dirname(file_path), exist_ok=True)
+    
+    try:
+        with open(file_path, 'w', encoding='utf-8') as f:
+            json.dump(data_to_save, f, ensure_ascii=False, indent=4)
+        logger.info(f"RescueTime data successfully saved to {file_path}")
+    except IOError as e:
+        logger.error(f"Failed to save RescueTime data to {file_path}: {e}")
+
+def main():
+    ag_data = fetch_for_user(api_key_ag, yesterday_formatted)
+
+    combined_data = {
+        "ag": ag_data,
+    }
+
+    save_rescuetime_data(combined_data)
+
+if __name__ == "__main__":
+    main()

static/js/domUpdates.js

@@ -139,3 +139,19 @@ export function updateRandomQuote(data) {
         quoteDiv.textContent = `${data.quote.content} - ${data.quote.author}`;
     }
 }
+
+export function updateRescueTimeData(data) {
+    const rescueTimeDiv = document.getElementById('rescuetime_data');
+    if (rescueTimeDiv && data.rescuetime_data && data.rescuetime_data.ag) {
+        const agData = data.rescuetime_data.ag;
+        rescueTimeDiv.innerHTML = `
+            <span>Arvy's RescueTime for ${agData.date}:</span><br>
+            <span>Productive: ${agData.all_productive_duration_formatted}</span><br>
+            <span>Distracting: ${agData.all_distracting_duration_formatted}</span><br>
+            <span>Total Time: ${agData.total_duration_formatted}</span><br>
+            <span class="timestamp">Updated: ${data.fetch_date}</span>
+        `;
+    } else {
+        rescueTimeDiv.innerHTML = '<span>RescueTime data not available</span>';
+    }
+}

static/js/main.js

@@ -5,7 +5,7 @@
  * and schedules periodic updates for various components of the application.
  */
 
-import { updateStockData, updateTextColors, updateSensorData, updateCvbankasData, updateCurrentTime, updateSystemInfo, updateRandomQuote } from './domUpdates.js';
+import { updateStockData, updateTextColors, updateSensorData, updateCvbankasData, updateCurrentTime, updateSystemInfo, updateRandomQuote, updateRescueTimeData } from './domUpdates.js';
 
 
 // Schedule periodic update requests
@@ -31,6 +31,10 @@ setInterval(() => {
     socket.emit('request_from_client_to_server_for_random_quote_data');
 }, 3600000); //1 hour
 
+setInterval(() => {
+    socket.emit('request_from_client_to_server_for_rescuetime_data');
+}, 36000000); // 10 hours
+
 setInterval(updateCurrentTime, 1000);
 
 
@@ -50,6 +54,7 @@ socket.on('connect', () => {
     socket.emit('request_from_client_to_server_for_stock_data');
     socket.emit('request_from_client_to_server_for_system_info_data');
     socket.emit('request_from_client_to_server_for_random_quote_data');
+    socket.emit('request_from_client_to_server_for_rescuetime_data');
     updateCurrentTime();
 });
 
@@ -77,3 +82,6 @@ socket.on('response_from_server_to_client_with_quote_data', (data) => {
     updateRandomQuote(data);
 });
 
+socket.on('response_from_server_to_client_with_rescuetime_data', (data) => {
+    updateRescueTimeData(data);
+});

templates/index.html

@@ -30,6 +30,7 @@
         <div class="column column-2">
             <div class="row" id="current-time"></div>
             <div class="row" id="sensor_data"></div>
+            <div class="row" id="rescuetime_data"></div>
             <div id="cvbankas_jobs" class="row"></div>
         </div>
     </div>