MessagesCreator.prepare_img_video() - Code Metrics - Inspection of "Added ability to add video" - Deathangel908/pychat - Measure and Improve Code Quality continuously with Scrutinizer

Completed

Push — master ( cd5ef7...1da252 )

by Andrew

created 2018-05-01 08:03 UTC

MessagesCreator.prepare_img_video() A

↳ Parent: MessagesCreator

Complexity

Conditions

Size

Total Lines

Duplication

Lines	0
Ratio	0 %

Importance

Changes

Metric	Value
cc	4
dl	0
loc	8
rs	9.2
c	0
b	0
f	0

from chat.models import get_milliseconds
from chat.tornado.constants import VarNames, HandlerNames, Actions, RedisPrefix


class MessagesCreator(object):

	def default(self, content, event, handler):
		"""
		:return: {"action": event, "content": content, "time": "20:48:57"}
		"""
		return {
			VarNames.EVENT: event,
			VarNames.CONTENT: content,
			VarNames.USER_ID: self.user_id,
			VarNames.TIME: get_milliseconds(),
			VarNames.HANDLER_NAME: handler
		}

	@staticmethod
	def set_ws_id(random, self_id):
		return {
			VarNames.HANDLER_NAME: HandlerNames.WS,
			VarNames.EVENT: Actions.SET_WS_ID,
			VarNames.CONTENT: random,
			VarNames.WEBRTC_OPPONENT_ID: self_id
		}

	def room_online(self, online, event, channel):
		"""
		:return: {"action": event, "content": content, "time": "20:48:57"}
		"""
		room_less = self.default(online, event, HandlerNames.CHAT)
		room_less[VarNames.CHANNEL_NAME] = channel
		room_less[VarNames.USER] = self.sender_name
		room_less[VarNames.GENDER] = self.sex
		return room_less

	@classmethod
	def create_message(cls, message, files):
		res = {
			VarNames.USER_ID: message.sender_id,
			VarNames.CONTENT: message.content,
			VarNames.TIME: message.time,
			VarNames.MESSAGE_ID: message.id,
			VarNames.FILES: files,
			VarNames.GIPHY: message.giphy,
			VarNames.EDITED_TIMES: message.edited_times
		}
		return res

	@classmethod
	def create_send_message(cls, message, event, files, js_id):
		"""
		:type message: chat.models.Message
		:type imgs: dict
		:type event: Actions
		:return: "action": "joined", "content": {"v5bQwtWp": "alien", "tRD6emzs": "Alien"},
		"sex": "Alien", "user": "tRD6emzs", "time": "20:48:57"}
		"""
		res = cls.create_message(message, files)
		res[VarNames.EVENT] = event
		res[VarNames.JS_MESSAGE_ID] = js_id
		res[VarNames.CHANNEL] = message.room_id
		res[VarNames.SYMBOL] = message.symbol
		res[VarNames.HANDLER_NAME] = HandlerNames.CHAT
		return res

	@classmethod
	def append_images(cls, messages, files, prepare_img):
		"""
		:type messages: list[chat.models.Message] 
		:type files: list[chat.models.Image]
		"""
		res_mess = []
		for message in messages:
			res_mess.append(cls.create_message(message, prepare_img(files, message.id)))
		return res_mess

	@classmethod
	def get_messages(cls, messages, channel, files, prepare_img):
		"""
		:type images: list[chat.models.Image]
		:type messages: list[chat.models.Message]
		:type channel: str
		"""
		return {
			VarNames.CONTENT: cls.append_images(messages, files, prepare_img),
			VarNames.EVENT: Actions.GET_MESSAGES,
			VarNames.CHANNEL: channel,
			VarNames.HANDLER_NAME: HandlerNames.CHAT
		}

	@staticmethod
	def prepare_img_video(files, message_id):
		"""
		:type message_id: int
		:type files: list[chat.models.Image]
		"""
		if files:
			return {x.symbol: {VarNames.FILE_URL: x.img.url, VarNames.FILE_TYPE: x.type} for x in files if x.message_id == message_id}

	@property
	def channel(self):
		return RedisPrefix.generate_user(self.user_id)

	def subscribe_direct_channel_message(self, room_id, other_user_id, notifications):
		return {
			VarNames.EVENT: Actions.CREATE_DIRECT_CHANNEL,
			VarNames.VOLUME: 2,
			VarNames.NOTIFICATIONS: notifications,
			VarNames.ROOM_ID: room_id,
			VarNames.ROOM_USERS: [self.user_id, other_user_id],
			VarNames.HANDLER_NAME: HandlerNames.CHANNELS
		}

	def responde_pong(self):
		return {
			VarNames.EVENT: Actions.PING,
			VarNames.HANDLER_NAME: HandlerNames.WS
		}

	def subscribe_room_channel_message(self, room_id, room_name):
		return {
			VarNames.EVENT: Actions.CREATE_ROOM_CHANNEL,
			VarNames.ROOM_ID: room_id,
			VarNames.ROOM_USERS: [self.user_id],
			VarNames.HANDLER_NAME: HandlerNames.CHANNELS,
			VarNames.VOLUME: 2,
			VarNames.NOTIFICATIONS: True,
			VarNames.ROOM_NAME: room_name
		}

	@staticmethod
	def invite_room_channel_message(room_id, user_id, room_name, users):
		return {
			VarNames.EVENT: Actions.INVITE_USER,
			VarNames.ROOM_ID: room_id,
			VarNames.USER_ID: user_id,
			VarNames.HANDLER_NAME: HandlerNames.CHANNELS,
			VarNames.ROOM_NAME: room_name,
			VarNames.CONTENT: users
		}

	@staticmethod
	def add_user_to_room(channel, user_id, content):
		return {
			VarNames.EVENT: Actions.ADD_USER,
			VarNames.CHANNEL: channel,
			VarNames.USER_ID: user_id,
			VarNames.HANDLER_NAME: HandlerNames.CHAT,
			VarNames.GENDER: content[VarNames.GENDER],  # SEX: 'Alien', USER: 'Andrew'
			VarNames.USER: content[VarNames.USER]  # SEX: 'Alien', USER: 'Andrew'
		}

	def unsubscribe_direct_message(self, room_id):
		return {
			VarNames.EVENT: Actions.DELETE_ROOM,
			VarNames.ROOM_ID: room_id,
			VarNames.USER_ID: self.user_id,
			VarNames.HANDLER_NAME: HandlerNames.CHANNELS,
			VarNames.TIME: get_milliseconds()
		}

	def load_offline_message(self, offline_messages, history_messages, channel_key):
		res = self.default({
			VarNames.LOAD_MESSAGES_OFFLINE: offline_messages,
			VarNames.LOAD_MESSAGES_HISTORY: history_messages
		},
			Actions.OFFLINE_MESSAGES,
			HandlerNames.CHAT)
		res[VarNames.CHANNEL] = channel_key
		return res


class WebRtcMessageCreator(object):

	def offer_webrtc(self, content, connection_id, room_id, action):
		"""
		:return: {"action": "call", "content": content, "time": "20:48:57"}
		"""
		return {
			VarNames.EVENT: action,
			VarNames.CONTENT: content,
			VarNames.USER_ID: self.user_id,
			VarNames.HANDLER_NAME: HandlerNames.WEBRTC,
			VarNames.USER: self.sender_name,
			VarNames.CONNECTION_ID: connection_id,
			VarNames.WEBRTC_OPPONENT_ID: self.id,
			VarNames.CHANNEL: room_id
		}

	def set_webrtc_error(self, error, connection_id, qued_id=None):
		return {
			VarNames.EVENT: Actions.SET_WEBRTC_ERROR,
			VarNames.CONTENT: error,
			VarNames.USER_ID: self.user_id,
			VarNames.HANDLER_NAME: HandlerNames.PEER_CONNECTION,
			VarNames.CONNECTION_ID: connection_id,
			VarNames.WEBRTC_QUED_ID: qued_id
		}

	@staticmethod
	def set_connection_id(qued_id, connection_id):
		return {
			VarNames.EVENT: Actions.SET_WEBRTC_ID,
			VarNames.HANDLER_NAME: HandlerNames.WEBRTC,
			VarNames.CONNECTION_ID: connection_id,
			VarNames.WEBRTC_QUED_ID: qued_id
		}

	def get_close_file_sender_message(self, connection_id):
		return {
			VarNames.EVENT: Actions.CLOSE_FILE_CONNECTION,
			VarNames.CONNECTION_ID: connection_id,
			VarNames.WEBRTC_OPPONENT_ID: self.id,
			VarNames.HANDLER_NAME: HandlerNames.WEBRTC_TRANSFER,
		}

	def get_accept_file_message(self, connection_id, content):
		return {
			VarNames.EVENT: Actions.ACCEPT_FILE,
			VarNames.CONNECTION_ID: connection_id,
			VarNames.WEBRTC_OPPONENT_ID: self.id,
			VarNames.HANDLER_NAME: HandlerNames.PEER_CONNECTION,
			VarNames.CONTENT: content,
		}

	def reply_webrtc(self, event, connection_id, handler, content):
		return {
			VarNames.EVENT: event,
			VarNames.CONNECTION_ID: connection_id,
			VarNames.USER_ID: self.user_id,
			VarNames.USER: self.sender_name,
			VarNames.CONTENT: content,
			VarNames.WEBRTC_OPPONENT_ID: self.id,
			VarNames.HANDLER_NAME: handler,
		}

	def retry_file(self, connection_id):
		return {
			VarNames.EVENT: Actions.RETRY_FILE_CONNECTION,
			VarNames.CONNECTION_ID: connection_id,
			VarNames.WEBRTC_OPPONENT_ID: self.id,
			VarNames.HANDLER_NAME: HandlerNames.PEER_CONNECTION,
		}


1			from chat.models import get_milliseconds
2			from chat.tornado.constants import VarNames, HandlerNames, Actions, RedisPrefix
3
4
5			class MessagesCreator(object):
6
7			def default(self, content, event, handler):
8			"""
9			:return: {"action": event, "content": content, "time": "20:48:57"}
10			"""
11			return {
12			VarNames.EVENT: event,
13			VarNames.CONTENT: content,
14			VarNames.USER_ID: self.user_id,
15			VarNames.TIME: get_milliseconds(),
16			VarNames.HANDLER_NAME: handler
17			}
18
19			@staticmethod
20			def set_ws_id(random, self_id):
21			return {
22			VarNames.HANDLER_NAME: HandlerNames.WS,
23			VarNames.EVENT: Actions.SET_WS_ID,
24			VarNames.CONTENT: random,
25			VarNames.WEBRTC_OPPONENT_ID: self_id
26			}
27
28			def room_online(self, online, event, channel):
29			"""
30			:return: {"action": event, "content": content, "time": "20:48:57"}
31			"""
32			room_less = self.default(online, event, HandlerNames.CHAT)
33			room_less[VarNames.CHANNEL_NAME] = channel
34			room_less[VarNames.USER] = self.sender_name
35			room_less[VarNames.GENDER] = self.sex
36			return room_less
37
38			@classmethod
39			def create_message(cls, message, files):
40			res = {
41			VarNames.USER_ID: message.sender_id,
42			VarNames.CONTENT: message.content,
43			VarNames.TIME: message.time,
44			VarNames.MESSAGE_ID: message.id,
45			VarNames.FILES: files,
46			VarNames.GIPHY: message.giphy,
47			VarNames.EDITED_TIMES: message.edited_times
48			}
49			return res
50
51			@classmethod
52			def create_send_message(cls, message, event, files, js_id):
53			"""
54			:type message: chat.models.Message
55			:type imgs: dict
56			:type event: Actions
57			:return: "action": "joined", "content": {"v5bQwtWp": "alien", "tRD6emzs": "Alien"},
58			"sex": "Alien", "user": "tRD6emzs", "time": "20:48:57"}
59			"""
60			res = cls.create_message(message, files)
61			res[VarNames.EVENT] = event
62			res[VarNames.JS_MESSAGE_ID] = js_id
63			res[VarNames.CHANNEL] = message.room_id
64			res[VarNames.SYMBOL] = message.symbol
65			res[VarNames.HANDLER_NAME] = HandlerNames.CHAT
66			return res
67
68			@classmethod
69			def append_images(cls, messages, files, prepare_img):
70			"""
71			:type messages: list[chat.models.Message]
72			:type files: list[chat.models.Image]
73			"""
74			res_mess = []
75			for message in messages:
76			res_mess.append(cls.create_message(message, prepare_img(files, message.id)))
77			return res_mess
78
79			@classmethod
80			def get_messages(cls, messages, channel, files, prepare_img):
81			"""
82			:type images: list[chat.models.Image]
83			:type messages: list[chat.models.Message]
84			:type channel: str
85			"""
86			return {
87			VarNames.CONTENT: cls.append_images(messages, files, prepare_img),
88			VarNames.EVENT: Actions.GET_MESSAGES,
89			VarNames.CHANNEL: channel,
90			VarNames.HANDLER_NAME: HandlerNames.CHAT
91			}
92
93			@staticmethod
94			def prepare_img_video(files, message_id):
95			"""
96			:type message_id: int
97			:type files: list[chat.models.Image]
98			"""
99			if files:
100			return {x.symbol: {VarNames.FILE_URL: x.img.url, VarNames.FILE_TYPE: x.type} for x in files if x.message_id == message_id}
101
102			@property
103			def channel(self):
104			return RedisPrefix.generate_user(self.user_id)
105
106			def subscribe_direct_channel_message(self, room_id, other_user_id, notifications):
107			return {
108			VarNames.EVENT: Actions.CREATE_DIRECT_CHANNEL,
109			VarNames.VOLUME: 2,
110			VarNames.NOTIFICATIONS: notifications,
111			VarNames.ROOM_ID: room_id,
112			VarNames.ROOM_USERS: [self.user_id, other_user_id],
113			VarNames.HANDLER_NAME: HandlerNames.CHANNELS
114			}
115
116			def responde_pong(self):
117			return {
118			VarNames.EVENT: Actions.PING,
119			VarNames.HANDLER_NAME: HandlerNames.WS
120			}
121
122			def subscribe_room_channel_message(self, room_id, room_name):
123			return {
124			VarNames.EVENT: Actions.CREATE_ROOM_CHANNEL,
125			VarNames.ROOM_ID: room_id,
126			VarNames.ROOM_USERS: [self.user_id],
127			VarNames.HANDLER_NAME: HandlerNames.CHANNELS,
128			VarNames.VOLUME: 2,
129			VarNames.NOTIFICATIONS: True,
130			VarNames.ROOM_NAME: room_name
131			}
132
133			@staticmethod
134			def invite_room_channel_message(room_id, user_id, room_name, users):
135			return {
136			VarNames.EVENT: Actions.INVITE_USER,
137			VarNames.ROOM_ID: room_id,
138			VarNames.USER_ID: user_id,
139			VarNames.HANDLER_NAME: HandlerNames.CHANNELS,
140			VarNames.ROOM_NAME: room_name,
141			VarNames.CONTENT: users
142			}
143
144			@staticmethod
145			def add_user_to_room(channel, user_id, content):
146			return {
147			VarNames.EVENT: Actions.ADD_USER,
148			VarNames.CHANNEL: channel,
149			VarNames.USER_ID: user_id,
150			VarNames.HANDLER_NAME: HandlerNames.CHAT,
151			VarNames.GENDER: content[VarNames.GENDER], # SEX: 'Alien', USER: 'Andrew'
152			VarNames.USER: content[VarNames.USER] # SEX: 'Alien', USER: 'Andrew'
153			}
154
155			def unsubscribe_direct_message(self, room_id):
156			return {
157			VarNames.EVENT: Actions.DELETE_ROOM,
158			VarNames.ROOM_ID: room_id,
159			VarNames.USER_ID: self.user_id,
160			VarNames.HANDLER_NAME: HandlerNames.CHANNELS,
161			VarNames.TIME: get_milliseconds()
162			}
163
164			def load_offline_message(self, offline_messages, history_messages, channel_key):
165			res = self.default({
166			VarNames.LOAD_MESSAGES_OFFLINE: offline_messages,
167			VarNames.LOAD_MESSAGES_HISTORY: history_messages
168			},
169			Actions.OFFLINE_MESSAGES,
170			HandlerNames.CHAT)
171			res[VarNames.CHANNEL] = channel_key
172			return res
173
174
175			class WebRtcMessageCreator(object):
176
177			def offer_webrtc(self, content, connection_id, room_id, action):
178			"""
179			:return: {"action": "call", "content": content, "time": "20:48:57"}
180			"""
181			return {
182			VarNames.EVENT: action,
183			VarNames.CONTENT: content,
184			VarNames.USER_ID: self.user_id,
185			VarNames.HANDLER_NAME: HandlerNames.WEBRTC,
186			VarNames.USER: self.sender_name,
187			VarNames.CONNECTION_ID: connection_id,
188			VarNames.WEBRTC_OPPONENT_ID: self.id,
189			VarNames.CHANNEL: room_id
190			}
191
192			def set_webrtc_error(self, error, connection_id, qued_id=None):
193			return {
194			VarNames.EVENT: Actions.SET_WEBRTC_ERROR,
195			VarNames.CONTENT: error,
196			VarNames.USER_ID: self.user_id,
197			VarNames.HANDLER_NAME: HandlerNames.PEER_CONNECTION,
198			VarNames.CONNECTION_ID: connection_id,
199			VarNames.WEBRTC_QUED_ID: qued_id
200			}
201
202			@staticmethod
203			def set_connection_id(qued_id, connection_id):
204			return {
205			VarNames.EVENT: Actions.SET_WEBRTC_ID,
206			VarNames.HANDLER_NAME: HandlerNames.WEBRTC,
207			VarNames.CONNECTION_ID: connection_id,
208			VarNames.WEBRTC_QUED_ID: qued_id
209			}
210
211			def get_close_file_sender_message(self, connection_id):
212			return {
213			VarNames.EVENT: Actions.CLOSE_FILE_CONNECTION,
214			VarNames.CONNECTION_ID: connection_id,
215			VarNames.WEBRTC_OPPONENT_ID: self.id,
216			VarNames.HANDLER_NAME: HandlerNames.WEBRTC_TRANSFER,
217			}
218
219			def get_accept_file_message(self, connection_id, content):
220			return {
221			VarNames.EVENT: Actions.ACCEPT_FILE,
222			VarNames.CONNECTION_ID: connection_id,
223			VarNames.WEBRTC_OPPONENT_ID: self.id,
224			VarNames.HANDLER_NAME: HandlerNames.PEER_CONNECTION,
225			VarNames.CONTENT: content,
226			}
227
228			def reply_webrtc(self, event, connection_id, handler, content):
229			return {
230			VarNames.EVENT: event,
231			VarNames.CONNECTION_ID: connection_id,
232			VarNames.USER_ID: self.user_id,
233			VarNames.USER: self.sender_name,
234			VarNames.CONTENT: content,
235			VarNames.WEBRTC_OPPONENT_ID: self.id,
236			VarNames.HANDLER_NAME: handler,
237			}
238
239			def retry_file(self, connection_id):
240			return {
241			VarNames.EVENT: Actions.RETRY_FILE_CONNECTION,
242			VarNames.CONNECTION_ID: connection_id,
243			VarNames.WEBRTC_OPPONENT_ID: self.id,
244			VarNames.HANDLER_NAME: HandlerNames.PEER_CONNECTION,
245			}
246

Deathangel908 / pychat

Push — master ( cd5ef7...1da252 )

MessagesCreator.prepare_img_video() A

Complexity

Size

Duplication

Importance

Duplication Side-by-Side

Filter issues like