Issue.__str__()   A
last analyzed

Complexity

Conditions 1

Size

Total Lines 2

Duplication

Lines 0
Ratio 0 %

Importance

Changes 0
Metric Value
cc 1
c 0
b 0
f 0
dl 0
loc 2
rs 10
1
import datetime
2
import json
3
import time
4
from enum import Enum
5
from time import mktime
6
7
from django.contrib.auth.models import BaseUserManager, AbstractBaseUser
8
from django.db import models
9
from django.db.models import CharField, DateField, FileField, BooleanField, URLField
10
11
from chat.log_filters import id_generator
12
from chat.settings import GENDERS, DEFAULT_PROFILE_ID, JS_CONSOLE_LOGS
13
14
15
def get_random_path(instance, filename):
16
	"""
17
	:param filename base string for generated name
18
	:return: a unique string filename
19
	"""
20
	return "{}_{}".format(id_generator(8), filename)
21
22
def myoverridenmeta(name, bases, adict):
23
	newClass = type(name, bases, adict)
24
	for field in newClass._meta.fields:
25
		if field.attname == 'password':
26
			field.blank = True
27
	return newClass
28
29
30
class User(AbstractBaseUser):
31
	def get_short_name(self):
32
		return self.username
33
34
	def get_full_name(self):
35
		return self.username
36
37
	@property
38
	def is_staff(self):
39
		# every registered user can edit database
40
		return self.pk == DEFAULT_PROFILE_ID
41
42
	def has_perm(self, perm, obj=None):
43
		return self.is_staff
44
45
	def has_perms(self, perm, obj=None):
46
		return True
47
48
	def has_module_perms(self, app_label):
49
		return self.is_staff
50
51
	USERNAME_FIELD = 'username'
52
	username = CharField(max_length=30, null=False, unique=True)
53
54
	# specifies auth, create email, etc methods
55
	objects = BaseUserManager()
56
57
	# ISO/IEC 5218 1 male, 2 - female
58
	sex = models.SmallIntegerField(null=False, default=0)
59
60
	@property
61
	def sex_str(self):
62
		return GENDERS[self.sex]
63
64
	@sex_str.setter
65
	def sex_str(self, sex):
66
		if sex == 'Male':
67
			self.sex = 1
68
		elif sex == 'Female':
69
			self.sex = 2
70
		else:
71
			self.sex = 0
72
73
	__metaclass__ = myoverridenmeta
74
75
76
class Subscription(models.Model):
77
	user = models.ForeignKey(User, null=False)
78
	inactive = BooleanField(default=False, null=False)
79
	registration_id = models.CharField(null=False, max_length=191, unique=True)
80
	created = models.DateTimeField(default=datetime.datetime.now)
81
	updated = models.DateTimeField(default=datetime.datetime.now)
82
	agent = models.CharField(max_length=64, null=True, blank=True)
83
	is_mobile = BooleanField(default=False, null=False, blank=True)
84
	ip = models.ForeignKey('IpAddress', null=True, blank=True)
85
86
	def __unicode__(self):
87
		return self.__str__()
88
89
	def __str__(self):
90
		return str(self.id)
91
92
93
class Verification(models.Model):
94
95
	class TypeChoices(Enum):
96
		register = 'r'
97
		password = 'p'
98
		email = 'e'
99
		confirm_email = 'c'
100
101
	# a - account activation, r - recover
102
	type = models.CharField(null=False, max_length=1)
103
	token = models.CharField(max_length=17, null=False, default=id_generator)
104
	user = models.ForeignKey(User, null=False)
105
	time = models.DateTimeField(default=datetime.datetime.now)
106
	verified = BooleanField(default=False)
107
	email = models.EmailField(null=True, unique=False, blank=True, max_length=190)
108
109
	def __unicode__(self):
110
		return self.__str__()
111
112
	def __str__(self):
113
		return str(self.id)
114
115
	@property
116
	def type_enum(self):
117
		return self.TypeChoices(self.type)
118
119
	@type_enum.setter
120
	def type_enum(self, p_type):
121
		"""
122
		:type p_type: Verification.TypeChoices
123
		"""
124
		self.type = p_type.value
125
126
127
class UserProfile(User):
128
	name = CharField(max_length=30, null=True, blank=True)
129
	surname = CharField(max_length=30, null=True, blank=True)
130
	# tho email max length is 254 characted mysql supports unique keys only 767 bytes long (utf8 4 bytes = 767/4 = 191)
131
	email = models.EmailField(null=True, unique=True, blank=True, max_length=190)
132
	city = CharField(max_length=50, null=True, blank=True)
133
134
	birthday = DateField(null=True, blank=True)
135
	contacts = CharField(max_length=100, null=True, blank=True)
136
	# fileField + <img instead of ImageField (removes preview link)
137
	photo = FileField(upload_to=get_random_path, null=True, blank=True)
138
	suggestions = BooleanField(null=False, default=True)
139
	embedded_youtube = BooleanField(null=False, default=True)
140
	highlight_code = BooleanField(null=False, default=False)
141
	logs = BooleanField(null=False, default=JS_CONSOLE_LOGS)
142
	theme = CharField(max_length=16, null=False, default='color-reg')
143
	online_change_sound = BooleanField(null=False, default=True)
144
	incoming_file_call_sound = BooleanField(null=False, default=True)
145
	message_sound = BooleanField(null=False, default=True)
146
	send_logs = BooleanField(null=False, default=True)
147
148
	email_verification = models.ForeignKey(Verification, null=True, blank=True)
149
150
	def save(self, *args, **kwargs):
151
		"""
152
		http://stackoverflow.com/questions/15422606/django-model-email-field-unique-if-not-null-blank
153
		"""
154
		if self.email is not None:
155
			self.email.lower().strip()  # Hopefully reduces junk to ""
156
			if self.email == "":
157
				self.email = None
158
		super(UserProfile, self).save(*args, **kwargs)
159
160
161
class Room(models.Model):
162
	name = CharField(max_length=16, null=True, blank=True)
163
	users = models.ManyToManyField(User, related_name='rooms', through='RoomUsers')
164
	disabled = BooleanField(default=False, null=False)
165
166
	@property
167
	def is_private(self):
168
		return self.name is None
169
170
	def __unicode__(self):
171
		return self.__str__()
172
173
	def __str__(self):
174
		return "{}/{}".format(self.id, self.name)
175
176
177
def get_milliseconds(dt=None):
178
	if dt is None:
179
		return int(time.time()*1000)
180
	if dt.time.timestamp:
181
		return int(dt.time.timestamp()*1000)
182
	else:
183
		return mktime(dt.time.timetuple()) * 1000 + int(dt.time.microsecond / 1000)
184
185
186
class MessageHistory(models.Model):
187
	time = models.BigIntegerField(default=get_milliseconds)
188
	message = models.ForeignKey('Message', null=False)
189
	content = models.TextField(null=True, blank=True)
190
	giphy = URLField(null=True, blank=True)
191
192
193
class Message(models.Model):
194
	"""
195
	Contains all public messages
196
	"""
197
	sender = models.ForeignKey(User, related_name='sender')
198
	room = models.ForeignKey(Room, null=True)
199
	# DateField.auto_now
200
	time = models.BigIntegerField(default=get_milliseconds)
201
	content = models.TextField(null=True, blank=True)
202
	# if symbol = null - no images refers this row
203
	# symbol is the same as "select max(symbol) from images where message_id = message.id
204
	# we store symbol in this table in case if user edits message
205
	# - images that refers same message always have unique symbols
206
	symbol = models.CharField(null=True, max_length=1, blank=True)
207
	deleted = BooleanField(default=False)
208
	giphy = URLField(null=True, blank=True)
209
	edited_times = models.IntegerField(default=0, null=False)
210
211
	def __unicode__(self):
212
		return self.__str__()
213
214
	def __str__(self):
215
216
		if self.content is None:
217
			v = ''
218
		elif len(self.content) > 50:
219
			v = self.content[:50]
220
		else:
221
			v = self.content
222
		v = json.dumps(v)
223
		return "{}/{}".format(self.id, v)
224
225
226
class UploadedFile(models.Model):
227
	class UploadedFileChoices(Enum):
228
		video = 'v'
229
		image = 'i'
230
		preview = 'p'
231
		issue = 's'
232
	symbol = models.CharField(null=False, max_length=1)
233
	file = FileField(upload_to=get_random_path, null=True)
234
	user = models.ForeignKey(User, null=False)
235
	type = models.CharField(null=False, max_length=1)
236
237
	@property
238
	def type_enum(self):
239
		return self.UploadedFileChoices(self.type)
240
241
	@type_enum.setter
242
	def type_enum(self, p_type):
243
		self.type = p_type.value
244
245
246
class Image(models.Model):
247
248
	class MediaTypeChoices(Enum):
249
		video = 'v'
250
		image = 'i'
251
252
	# character in Message.content that will be replaced with this image
253
	symbol = models.CharField(null=False, max_length=1)
254
	message = models.ForeignKey(Message, related_name='message', null=False)
255
	img = FileField(upload_to=get_random_path, null=True)
256
	preview = FileField(upload_to=get_random_path, null=True)
257
	type = models.CharField(null=False, max_length=1, default=MediaTypeChoices.image.value)
258
259
	@property
260
	def type_enum(self):
261
		return self.MediaTypeChoices(self.type)
262
263
	@type_enum.setter
264
	def type_enum(self, p_type):
265
		self.type = p_type.value
266
267
	class Meta:
268
		unique_together = ('symbol', 'message')
269
270
271
class RoomUsers(models.Model):
272
	room = models.ForeignKey(Room, null=False)
273
	user = models.ForeignKey(User, null=False)
274
	last_read_message = models.ForeignKey(Message, null=True)
275
	volume = models.IntegerField(default=2, null=False)
276
	notifications = BooleanField(null=False, default=True)
277
278
	class Meta:  # pylint: disable=C1001
279
		unique_together = ("user", "room")
280
		db_table = ''.join((User._meta.app_label, '_room_users'))
281
282
283
class SubscriptionMessages(models.Model):
284
	message = models.ForeignKey(Message, null=False)
285
	subscription = models.ForeignKey(Subscription, null=False)
286
	received = BooleanField(null=False, default=False)
287
288
	class Meta:  # pylint: disable=C1001
289
		unique_together = ("message", "subscription")
290
		db_table = ''.join((User._meta.app_label, '_subscription_message'))
291
292
293
class Issue(models.Model):
294
	content = models.TextField(null=False)  # unique = true, but mysql doesnt allow unique fields for unspecified size
295
296
	def __str__(self):
297
		return self.content
298
299
300
class IssueDetails(models.Model):
301
	sender = models.ForeignKey(User, null=True)
302
	browser = models.CharField(null=True, max_length=32, blank=True)
303
	time = models.DateField(default=datetime.datetime.now, blank=True)
304
	issue = models.ForeignKey(Issue, related_name='issue')
305
	log = models.TextField(null=True, blank=True)
306
307
	class Meta:  # pylint: disable=C1001
308
		db_table = ''.join((User._meta.app_label, '_issue_detail'))
309
310
311
class IpAddress(models.Model):
312
	ip = models.CharField(null=False, max_length=32, unique=True)
313
	isp = models.CharField(null=True, max_length=32, blank=True)
314
	country_code = models.CharField(null=True, max_length=16, blank=True)
315
	country = models.CharField(null=True, max_length=32, blank=True)
316
	region = models.CharField(null=True, max_length=32, blank=True)
317
	city = models.CharField(null=True, max_length=32, blank=True)
318
319
	def __str__(self):
320
		return self.ip
321
322
	@property
323
	def info(self):
324
		if self.country is not None:
325
			return "{} {} ({})".format(self.country, self.city, self.isp)
326
		else:
327
			return ""
328
329
330
	class Meta:  # pylint: disable=C1001
331
		db_table = ''.join((User._meta.app_label, '_ip_address'))
332
333
334
class UserJoinedInfo(models.Model):
335
	ip = models.ForeignKey(IpAddress, null=True)
336
	user = models.ForeignKey(User, null=True)
337
	time = models.DateField(default=datetime.datetime.now)
338
339
	class Meta:  # pylint: disable=C1001
340
		db_table = ''.join((User._meta.app_label, '_user_joined_info'))
341
		unique_together = ("user", "ip")
342