From e1d017a92b87733f6f436b97cf6178f630e8ccff Mon Sep 17 00:00:00 2001 From: freeram Date: Sat, 16 Mar 2024 17:20:12 -0600 Subject: [PATCH] Add basic plots --- pomodorodiscord/src/graphs.py | 37 +++++++++++++++++++++++++++++++++++ 1 file changed, 37 insertions(+) create mode 100644 pomodorodiscord/src/graphs.py diff --git a/pomodorodiscord/src/graphs.py b/pomodorodiscord/src/graphs.py new file mode 100644 index 0000000..20ccca3 --- /dev/null +++ b/pomodorodiscord/src/graphs.py @@ -0,0 +1,37 @@ +import matplotlib.pyplot as plt +from utils import load_data + + +def graph_pomodoro_sessions(data): + dates = list(data.get('sessions_by_date', {}).keys()) + sessions = list(data.get('sessions_by_date', {}).values()) + print(sessions) + + plt.figure(figsize=(10, 6)) + plt.plot(dates, sessions, marker='o') + plt.title('Pomodoro Sessions per Day') + plt.xlabel('Days') + plt.ylabel('Pomodoro Sessions') + plt.xticks(rotation=45) + plt.yticks(sessions) + plt.tight_layout() + plt.show() + +def graph_hours_studied(data): + dates = list(data.get('seconds_by_date', {}).keys()) + seconds = list(data.get('seconds_by_date', {}).values()) + hours = [s / 3600 for s in seconds] + + plt.figure(figsize=(10, 6)) + plt.plot(dates, hours, marker='o', color='red') + plt.title('Hours Studied per Day') + plt.xlabel('Days') + plt.ylabel('Hours Studied') + plt.xticks(rotation=45) + plt.tight_layout() + plt.show() + +if __name__ == '__main__': + data = load_data() + graph_pomodoro_sessions(data) + graph_hours_studied(data)