Clean up basic_types

yaboli/basic_types.py

@@ -33,17 +33,14 @@ class SessionView():
 		data - a euphoria SessionView
 		"""
 		
-		is_staff = data["is_staff"] if "is_staff" in data else None
-		is_manager = data["is_manager"] if "is_manager" in data else None
-		
 		return self(
-			data["id"],
+			data.get("id"),
-			data["name"],
+			data.get("name"),
-			data["server_id"],
+			data.get("server_id"),
-			data["server_era"],
+			data.get("server_era"),
-			data["session_id"],
+			data.get("session_id"),
-			is_staff,
+			data.get("is_staff"),
-			is_manager
+			data.get("is_manager")
 		)
 	
 	def session_type(self):
@@ -66,7 +63,7 @@ class Message():
 		"""
 		id                 - message id
 		time               - time the message was sent (epoch)
-		sender             - session of the sender
+		sender             - SessionView of the sender
 		content            - content of the message
 		parent             - id of the parent message, or None
 		edited             - time of last edit (epoch)
@@ -96,25 +93,19 @@ class Message():
 		data - a euphoria message: http://api.euphoria.io/#message
 		"""
 		
-		sender = SessionView.from_data(data["sender"])
+		sender = SessionView.from_data(data.get("sender"))
-		parent = data["parent"] if "parent" in data else None
-		edited = data["edited"] if "edited" in data else None
-		deleted = data["deleted"] if "deleted" in data else None
-		truncated = data["truncated"] if "truncated" in data else None
-		previous_edit_id = data["previous_edit_id"] if "previous_edit_id" in data else None
-		encryption_key_id = data["encryption_key_id"] if "encryption_key_id" in data else None
 		
 		return self(
-			data["id"],
+			data.get("id"),
-			data["time"],
+			data.get("time"),
 			sender,
-			data["content"],
+			data.get("content"),
-			parent=parent,
+			parent=data.get("parent"),
-			edited=edited,
+			edited=data.get("edited"),
-			deleted=deleted,
+			deleted=data.get("deleted"),
-			truncated=truncated,
+			truncated=data.get("truncated"),
-			previous_edit_id=previous_edit_id,
+			previous_edit_id=data.get("previous_edit_id"),
-			encryption_key_id=encryption_key_id
+			encryption_key_id=data.get("encryption_key_id")
 		)
 	
 	def time_formatted(self, date=True):