watch history update + some general stuff

1 files changed, 12 insertions, 0 deletions

diff --git a/watch/utils.py b/watch/utils.py
index 83888f6..857601a 100644
--- a/watch/utils.py
+++ b/watch/utils.py
@@ -1,6 +1,7 @@
 import redis
 import os
 import dotenv
+from user_profile.models import UserHistory
 
 dotenv.load_dotenv()
 
@@ -11,6 +12,17 @@ r = redis.Redis(
     ssl=True,
 )
 
+def update_anime_user_history(user, anime_id, episode, time_watched):
+    # per episode history
+    history, created = UserHistory.objects.get_or_create(user=user, anime_id=anime_id, episode=episode)
+    history.time_watched = time_watched
+    history.save()
+
+def get_anime_user_history(user, anime_id):
+    history = UserHistory.objects.filter(user=user, anime_id=anime_id).order_by("-episode")
+    return history
+
+
 def store_in_redis_cache(anime_id, data):
     r.set(anime_id, data, ex=60*60*24*30) # 30 days
     print("data cached", anime_id)