1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
|
from models.Session import Session
from typing import Dict, Optional
class SessionStore:
def initialize(self) -> None:
pass
def get_session(self, session_id: str) -> Optional[Session]:
raise NotImplementedError()
def create_session(self) -> Session:
raise NotImplementedError()
class InMemorySessionStore(SessionStore):
def __init__(self):
SessionStore.__init__(self)
self.sessions: Dict[str, Session] = {}
def get_session(self, session_id: str) -> Optional[Session]:
if session_id in self.sessions:
return self.sessions[session_id]
else:
return None
def create_session(self):
session = Session()
self.sessions[session.session_id] = session
return session
|