Initial commit

sessions.py

@@ -0,0 +1,104 @@
+from . import session
+
+class Sessions():
+	"""
+	Keeps track of sessions.
+	"""
+	
+	def __init__(self):
+		"""
+		TODO
+		"""
+		self.sessions = {}
+	
+	def add_raw(self, raw_session):
+		"""
+		add_raw(raw_session) -> None
+		
+		Create a session from raw data and add it.
+		"""
+		
+		ses = session.Session(raw_session)
+		
+		self.sessions[ses.session_id()] = ses
+	
+	def add(self, ses):
+		"""
+		add(session) -> None
+		
+		Add a session.
+		"""
+		
+		self.sessions[ses.session_id()] = ses
+	
+	def remove(self, ses):
+		"""
+		remove(session_id) -> None
+		
+		Remove a session.
+		"""
+		
+		if ses.session_id() in self.sessions:
+			self.sessions.pop(ses.session_id())
+	
+	def remove_on_network_partition(self, server_id, server_era):
+		"""
+		remove_on_network_partition(server_id, server_era) -> None
+		
+		Removes all sessions matching the server_id/server_era combo.
+		http://api.euphoria.io/#network-event
+		"""
+		
+		for ses in self.sessions:
+			if ses.server_id() == server_id and ses.server_era() == server_era:
+				self.remove(ses)
+	
+	def get_people(self):
+		"""
+		get_people() -> list
+		
+		Returns a list of all non-bot and non-lurker sessions.
+		"""
+		
+		# not a list comprehension because that would span several lines too
+		people = []
+		for ses in self.sessions:
+			if ses.session_type() in ["agent", "account"] and ses.name():
+				people.append(ses)
+		return people
+	
+	def get_accounts(self):
+		"""
+		get_accounts() -> list
+		
+		Returns a list of all logged-in sessions.
+		"""
+		
+		return [ses for ses in self.sessions if ses.session_type() == "account" and ses.name()]
+	
+	def get_agents(self):
+		"""
+		get_agents() -> list
+		
+		Returns a list of all sessions who are not signed into an account and not bots or lurkers.
+		"""
+		
+		return [ses for ses in self.sessions if ses.session_type() == "agent" and ses.name()]
+	
+	def get_bots(self):
+		"""
+		get_bots() -> list
+		
+		Returns a list of all bot sessions.
+		"""
+		
+		return [ses for ses in self.sessions if ses.session_type() == "bot" and ses.name()]
+	
+	def get_lurkers(self):
+		"""
+		get_lurkers() -> list
+		
+		Returns a list of all lurker sessions.
+		"""
+		
+		return [ses for ses in self.sessions if not ses.name()]