gmakstutis/openrag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Update session_ownership_service.py

Browse source
This commit is contained in:
Edwin Jose 2025-09-10 12:42:09 -04:00
parent 8390653861
commit 4212c1025c
1 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
src/services/session_ownership_service.py
View file

@ -7,7 +7,9 @@ import json
import os import os
from typing import Dict, List, Optional from typing import Dict, List, Optional
from datetime import datetime from datetime import datetime
from utils.logging_config import get_logger
logger = get_logger(__name__)
class SessionOwnershipService: class SessionOwnershipService:
"""Simple service to track which user owns which session""" """Simple service to track which user owns which session"""
@ -23,7 +25,7 @@ class SessionOwnershipService:
with open(self.ownership_file, 'r') as f: with open(self.ownership_file, 'r') as f:
return json.load(f) return json.load(f)
except Exception as e: except Exception as e:
print(f"Error loading session ownership data: {e}") logger.error(f"Error loading session ownership data: {e}")
return {} return {}
return {} return {}
@ -32,9 +34,9 @@ class SessionOwnershipService:
try: try:
with open(self.ownership_file, 'w') as f: with open(self.ownership_file, 'w') as f:
json.dump(self.ownership_data, f, indent=2) json.dump(self.ownership_data, f, indent=2)
print(f"Saved session ownership data to {self.ownership_file}") logger.debug(f"Saved session ownership data to {self.ownership_file}")
except Exception as e: except Exception as e:
print(f"Error saving session ownership data: {e}") logger.error(f"Error saving session ownership data: {e}")
def claim_session(self, user_id: str, session_id: str): def claim_session(self, user_id: str, session_id: str):
"""Claim a session for a user""" """Claim a session for a user"""
@ -45,7 +47,7 @@ class SessionOwnershipService:
"last_accessed": datetime.now().isoformat() "last_accessed": datetime.now().isoformat()
} }
self._save_ownership_data() self._save_ownership_data()
print(f"Claimed session {session_id} for user {user_id}") logger.debug(f"Claimed session {session_id} for user {user_id}")
else: else:
# Update last accessed time # Update last accessed time
self.ownership_data[session_id]["last_accessed"] = datetime.now().isoformat() self.ownership_data[session_id]["last_accessed"] = datetime.now().isoformat()