MessagesCreator.prepare_img_video() - Code Metrics - Inspection of "Fixed multiple youtube in 1 message" - Deathangel908/pychat - Measure and Improve Code Quality continuously with Scrutinizer

Completed

Push — master ( 51d889...c8777d )

by Andrew

created 2018-05-01 16:02 UTC

MessagesCreator.prepare_img_video() B

↳ Parent: MessagesCreator

Complexity

Conditions

Size

Total Lines

Duplication

Lines	0
Ratio	0 %

Importance

Changes

Metric	Value
cc	5
c	0
b	0
f	0
dl	0
loc	12
rs	8.5454

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,
				VarNames.PREVIEW: x.preview.url if x.preview else None
			} 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: {
101			VarNames.FILE_URL: x.img.url,
102			VarNames.FILE_TYPE: x.type,
103			VarNames.PREVIEW: x.preview.url if x.preview else None
104			} for x in files if x.message_id == message_id}
105
106			@property
107			def channel(self):
108			return RedisPrefix.generate_user(self.user_id)
109
110			def subscribe_direct_channel_message(self, room_id, other_user_id, notifications):
111			return {
112			VarNames.EVENT: Actions.CREATE_DIRECT_CHANNEL,
113			VarNames.VOLUME: 2,
114			VarNames.NOTIFICATIONS: notifications,
115			VarNames.ROOM_ID: room_id,
116			VarNames.ROOM_USERS: [self.user_id, other_user_id],
117			VarNames.HANDLER_NAME: HandlerNames.CHANNELS
118			}
119
120			def responde_pong(self):
121			return {
122			VarNames.EVENT: Actions.PING,
123			VarNames.HANDLER_NAME: HandlerNames.WS
124			}
125
126			def subscribe_room_channel_message(self, room_id, room_name):
127			return {
128			VarNames.EVENT: Actions.CREATE_ROOM_CHANNEL,
129			VarNames.ROOM_ID: room_id,
130			VarNames.ROOM_USERS: [self.user_id],
131			VarNames.HANDLER_NAME: HandlerNames.CHANNELS,
132			VarNames.VOLUME: 2,
133			VarNames.NOTIFICATIONS: True,
134			VarNames.ROOM_NAME: room_name
135			}
136
137			@staticmethod
138			def invite_room_channel_message(room_id, user_id, room_name, users):
139			return {
140			VarNames.EVENT: Actions.INVITE_USER,
141			VarNames.ROOM_ID: room_id,
142			VarNames.USER_ID: user_id,
143			VarNames.HANDLER_NAME: HandlerNames.CHANNELS,
144			VarNames.ROOM_NAME: room_name,
145			VarNames.CONTENT: users
146			}
147
148			@staticmethod
149			def add_user_to_room(channel, user_id, content):
150			return {
151			VarNames.EVENT: Actions.ADD_USER,
152			VarNames.CHANNEL: channel,
153			VarNames.USER_ID: user_id,
154			VarNames.HANDLER_NAME: HandlerNames.CHAT,
155			VarNames.GENDER: content[VarNames.GENDER], # SEX: 'Alien', USER: 'Andrew'
156			VarNames.USER: content[VarNames.USER] # SEX: 'Alien', USER: 'Andrew'
157			}
158
159			def unsubscribe_direct_message(self, room_id):
160			return {
161			VarNames.EVENT: Actions.DELETE_ROOM,
162			VarNames.ROOM_ID: room_id,
163			VarNames.USER_ID: self.user_id,
164			VarNames.HANDLER_NAME: HandlerNames.CHANNELS,
165			VarNames.TIME: get_milliseconds()
166			}
167
168			def load_offline_message(self, offline_messages, history_messages, channel_key):
169			res = self.default({
170			VarNames.LOAD_MESSAGES_OFFLINE: offline_messages,
171			VarNames.LOAD_MESSAGES_HISTORY: history_messages
172			},
173			Actions.OFFLINE_MESSAGES,
174			HandlerNames.CHAT)
175			res[VarNames.CHANNEL] = channel_key
176			return res
177
178
179			class WebRtcMessageCreator(object):
180
181			def offer_webrtc(self, content, connection_id, room_id, action):
182			"""
183			:return: {"action": "call", "content": content, "time": "20:48:57"}
184			"""
185			return {
186			VarNames.EVENT: action,
187			VarNames.CONTENT: content,
188			VarNames.USER_ID: self.user_id,
189			VarNames.HANDLER_NAME: HandlerNames.WEBRTC,
190			VarNames.USER: self.sender_name,
191			VarNames.CONNECTION_ID: connection_id,
192			VarNames.WEBRTC_OPPONENT_ID: self.id,
193			VarNames.CHANNEL: room_id
194			}
195
196			def set_webrtc_error(self, error, connection_id, qued_id=None):
197			return {
198			VarNames.EVENT: Actions.SET_WEBRTC_ERROR,
199			VarNames.CONTENT: error,
200			VarNames.USER_ID: self.user_id,
201			VarNames.HANDLER_NAME: HandlerNames.PEER_CONNECTION,
202			VarNames.CONNECTION_ID: connection_id,
203			VarNames.WEBRTC_QUED_ID: qued_id
204			}
205
206			@staticmethod
207			def set_connection_id(qued_id, connection_id):
208			return {
209			VarNames.EVENT: Actions.SET_WEBRTC_ID,
210			VarNames.HANDLER_NAME: HandlerNames.WEBRTC,
211			VarNames.CONNECTION_ID: connection_id,
212			VarNames.WEBRTC_QUED_ID: qued_id
213			}
214
215			def get_close_file_sender_message(self, connection_id):
216			return {
217			VarNames.EVENT: Actions.CLOSE_FILE_CONNECTION,
218			VarNames.CONNECTION_ID: connection_id,
219			VarNames.WEBRTC_OPPONENT_ID: self.id,
220			VarNames.HANDLER_NAME: HandlerNames.WEBRTC_TRANSFER,
221			}
222
223			def get_accept_file_message(self, connection_id, content):
224			return {
225			VarNames.EVENT: Actions.ACCEPT_FILE,
226			VarNames.CONNECTION_ID: connection_id,
227			VarNames.WEBRTC_OPPONENT_ID: self.id,
228			VarNames.HANDLER_NAME: HandlerNames.PEER_CONNECTION,
229			VarNames.CONTENT: content,
230			}
231
232			def reply_webrtc(self, event, connection_id, handler, content):
233			return {
234			VarNames.EVENT: event,
235			VarNames.CONNECTION_ID: connection_id,
236			VarNames.USER_ID: self.user_id,
237			VarNames.USER: self.sender_name,
238			VarNames.CONTENT: content,
239			VarNames.WEBRTC_OPPONENT_ID: self.id,
240			VarNames.HANDLER_NAME: handler,
241			}
242
243			def retry_file(self, connection_id):
244			return {
245			VarNames.EVENT: Actions.RETRY_FILE_CONNECTION,
246			VarNames.CONNECTION_ID: connection_id,
247			VarNames.WEBRTC_OPPONENT_ID: self.id,
248			VarNames.HANDLER_NAME: HandlerNames.PEER_CONNECTION,
249			}
250

Deathangel908 / pychat

Push — master ( 51d889...c8777d )

MessagesCreator.prepare_img_video() B

Complexity

Size

Duplication

Importance

Duplication Side-by-Side

Filter issues like