MediaTypeChoices - Code Metrics - Deathangel908/pychat - Measure and Improve Code Quality continuously with Scrutinizer

MediaTypeChoices A
last analyzed 2018-07-01 16:09 UTC

↳ Parent: Image

Complexity

Total Complexity

Size/Duplication

Total Lines	3
Duplicated Lines	0 %

Importance

Changes

Metric	Value
c	0
b	0
f	0
dl	0
loc	3
rs	10
wmc	0

import datetime
import json
import time
from enum import Enum
from time import mktime

from django.contrib.auth.models import BaseUserManager, AbstractBaseUser
from django.db import models
from django.db.models import CharField, DateField, FileField, BooleanField, URLField

from chat.log_filters import id_generator
from chat.settings import GENDERS, DEFAULT_PROFILE_ID, JS_CONSOLE_LOGS


def get_random_path(instance, filename):
	"""
	:param filename base string for generated name
	:return: a unique string filename
	"""
	return "{}_{}".format(id_generator(8), filename)

def myoverridenmeta(name, bases, adict):
	newClass = type(name, bases, adict)
	for field in newClass._meta.fields:
		if field.attname == 'password':
			field.blank = True
	return newClass


class User(AbstractBaseUser):
	def get_short_name(self):
		return self.username

	def get_full_name(self):
		return self.username

	@property
	def is_staff(self):
		# every registered user can edit database
		return self.pk == DEFAULT_PROFILE_ID

	def has_perm(self, perm, obj=None):
		return self.is_staff

	def has_perms(self, perm, obj=None):
		return True

	def has_module_perms(self, app_label):
		return self.is_staff

	USERNAME_FIELD = 'username'
	username = CharField(max_length=30, null=False, unique=True)

	# specifies auth, create email, etc methods
	objects = BaseUserManager()

	# ISO/IEC 5218 1 male, 2 - female
	sex = models.SmallIntegerField(null=False, default=0)

	@property
	def sex_str(self):
		return GENDERS[self.sex]

	@sex_str.setter
	def sex_str(self, sex):
		if sex == 'Male':
			self.sex = 1
		elif sex == 'Female':
			self.sex = 2
		else:
			self.sex = 0

	__metaclass__ = myoverridenmeta


class Subscription(models.Model):
	user = models.ForeignKey(User, null=False)
	inactive = BooleanField(default=False, null=False)
	registration_id = models.CharField(null=False, max_length=191, unique=True)
	created = models.DateTimeField(default=datetime.datetime.now)
	updated = models.DateTimeField(default=datetime.datetime.now)
	agent = models.CharField(max_length=64, null=True, blank=True)
	is_mobile = BooleanField(default=False, null=False, blank=True)
	ip = models.ForeignKey('IpAddress', null=True, blank=True)

	def __unicode__(self):
		return self.__str__()

	def __str__(self):
		return str(self.id)


class Verification(models.Model):

	class TypeChoices(Enum):
		register = 'r'
		password = 'p'
		email = 'e'
		confirm_email = 'c'

	# a - account activation, r - recover
	type = models.CharField(null=False, max_length=1)
	token = models.CharField(max_length=17, null=False, default=id_generator)
	user = models.ForeignKey(User, null=False)
	time = models.DateTimeField(default=datetime.datetime.now)
	verified = BooleanField(default=False)
	email = models.EmailField(null=True, unique=False, blank=True, max_length=190)

	def __unicode__(self):
		return self.__str__()

	def __str__(self):
		return str(self.id)

	@property
	def type_enum(self):
		return self.TypeChoices(self.type)

	@type_enum.setter
	def type_enum(self, p_type):
		"""
		:type p_type: Verification.TypeChoices
		"""
		self.type = p_type.value


class UserProfile(User):
	name = CharField(max_length=30, null=True, blank=True)
	surname = CharField(max_length=30, null=True, blank=True)
	# tho email max length is 254 characted mysql supports unique keys only 767 bytes long (utf8 4 bytes = 767/4 = 191)
	email = models.EmailField(null=True, unique=True, blank=True, max_length=190)
	city = CharField(max_length=50, null=True, blank=True)

	birthday = DateField(null=True, blank=True)
	contacts = CharField(max_length=100, null=True, blank=True)
	# fileField + <img instead of ImageField (removes preview link)
	photo = FileField(upload_to=get_random_path, null=True, blank=True)
	suggestions = BooleanField(null=False, default=True)
	embedded_youtube = BooleanField(null=False, default=True)
	highlight_code = BooleanField(null=False, default=False)
	logs = BooleanField(null=False, default=JS_CONSOLE_LOGS)
	theme = CharField(max_length=16, null=False, default='color-reg')
	online_change_sound = BooleanField(null=False, default=True)
	incoming_file_call_sound = BooleanField(null=False, default=True)
	message_sound = BooleanField(null=False, default=True)
	send_logs = BooleanField(null=False, default=True)

	email_verification = models.ForeignKey(Verification, null=True, blank=True)

	def save(self, *args, **kwargs):
		"""
		http://stackoverflow.com/questions/15422606/django-model-email-field-unique-if-not-null-blank
		"""
		if self.email is not None:
			self.email.lower().strip()  # Hopefully reduces junk to ""
			if self.email == "":
				self.email = None
		super(UserProfile, self).save(*args, **kwargs)


class Room(models.Model):
	name = CharField(max_length=16, null=True, blank=True)
	users = models.ManyToManyField(User, related_name='rooms', through='RoomUsers')
	disabled = BooleanField(default=False, null=False)

	@property
	def is_private(self):
		return self.name is None

	def __unicode__(self):
		return self.__str__()

	def __str__(self):
		return "{}/{}".format(self.id, self.name)


def get_milliseconds(dt=None):
	if dt is None:
		return int(time.time()*1000)
	if dt.time.timestamp:
		return int(dt.time.timestamp()*1000)
	else:
		return mktime(dt.time.timetuple()) * 1000 + int(dt.time.microsecond / 1000)


class MessageHistory(models.Model):
	time = models.BigIntegerField(default=get_milliseconds)
	message = models.ForeignKey('Message', null=False)
	content = models.TextField(null=True, blank=True)
	giphy = URLField(null=True, blank=True)


class Message(models.Model):
	"""
	Contains all public messages
	"""
	sender = models.ForeignKey(User, related_name='sender')
	room = models.ForeignKey(Room, null=True)
	# DateField.auto_now
	time = models.BigIntegerField(default=get_milliseconds)
	content = models.TextField(null=True, blank=True)
	# if symbol = null - no images refers this row
	# symbol is the same as "select max(symbol) from images where message_id = message.id
	# we store symbol in this table in case if user edits message
	# - images that refers same message always have unique symbols
	symbol = models.CharField(null=True, max_length=1, blank=True)
	deleted = BooleanField(default=False)
	giphy = URLField(null=True, blank=True)
	edited_times = models.IntegerField(default=0, null=False)

	def __unicode__(self):
		return self.__str__()

	def __str__(self):

		if self.content is None:
			v = ''
		elif len(self.content) > 50:
			v = self.content[:50]
		else:
			v = self.content
		v = json.dumps(v)
		return "{}/{}".format(self.id, v)


class UploadedFile(models.Model):
	class UploadedFileChoices(Enum):
		video = 'v'
		image = 'i'
		preview = 'p'
		issue = 's'
	symbol = models.CharField(null=False, max_length=1)
	file = FileField(upload_to=get_random_path, null=True)
	user = models.ForeignKey(User, null=False)
	type = models.CharField(null=False, max_length=1)

	@property
	def type_enum(self):
		return self.UploadedFileChoices(self.type)

	@type_enum.setter
	def type_enum(self, p_type):
		self.type = p_type.value


class Image(models.Model):

	class MediaTypeChoices(Enum):
		video = 'v'
		image = 'i'

	# character in Message.content that will be replaced with this image
	symbol = models.CharField(null=False, max_length=1)
	message = models.ForeignKey(Message, related_name='message', null=False)
	img = FileField(upload_to=get_random_path, null=True)
	preview = FileField(upload_to=get_random_path, null=True)
	type = models.CharField(null=False, max_length=1, default=MediaTypeChoices.image.value)

	@property
	def type_enum(self):
		return self.MediaTypeChoices(self.type)

	@type_enum.setter
	def type_enum(self, p_type):
		self.type = p_type.value

	class Meta:
		unique_together = ('symbol', 'message')


class RoomUsers(models.Model):
	room = models.ForeignKey(Room, null=False)
	user = models.ForeignKey(User, null=False)
	last_read_message = models.ForeignKey(Message, null=True)
	volume = models.IntegerField(default=2, null=False)
	notifications = BooleanField(null=False, default=True)

	class Meta:  # pylint: disable=C1001
		unique_together = ("user", "room")
		db_table = ''.join((User._meta.app_label, '_room_users'))


class SubscriptionMessages(models.Model):
	message = models.ForeignKey(Message, null=False)
	subscription = models.ForeignKey(Subscription, null=False)
	received = BooleanField(null=False, default=False)

	class Meta:  # pylint: disable=C1001
		unique_together = ("message", "subscription")
		db_table = ''.join((User._meta.app_label, '_subscription_message'))


class Issue(models.Model):
	content = models.TextField(null=False)  # unique = true, but mysql doesnt allow unique fields for unspecified size

	def __str__(self):
		return self.content


class IssueDetails(models.Model):
	sender = models.ForeignKey(User, null=True)
	browser = models.CharField(null=True, max_length=32, blank=True)
	time = models.DateField(default=datetime.datetime.now, blank=True)
	issue = models.ForeignKey(Issue, related_name='issue')
	log = models.TextField(null=True, blank=True)

	class Meta:  # pylint: disable=C1001
		db_table = ''.join((User._meta.app_label, '_issue_detail'))


class IpAddress(models.Model):
	ip = models.CharField(null=False, max_length=32, unique=True)
	isp = models.CharField(null=True, max_length=32, blank=True)
	country_code = models.CharField(null=True, max_length=16, blank=True)
	country = models.CharField(null=True, max_length=32, blank=True)
	region = models.CharField(null=True, max_length=32, blank=True)
	city = models.CharField(null=True, max_length=32, blank=True)

	def __str__(self):
		return self.ip

	@property
	def info(self):
		if self.country is not None:
			return "{} {} ({})".format(self.country, self.city, self.isp)
		else:
			return ""


	class Meta:  # pylint: disable=C1001
		db_table = ''.join((User._meta.app_label, '_ip_address'))


class UserJoinedInfo(models.Model):
	ip = models.ForeignKey(IpAddress, null=True)
	user = models.ForeignKey(User, null=True)
	time = models.DateField(default=datetime.datetime.now)

	class Meta:  # pylint: disable=C1001
		db_table = ''.join((User._meta.app_label, '_user_joined_info'))
		unique_together = ("user", "ip")


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

Deathangel908 / pychat

MediaTypeChoices A last analyzed 2018-07-01 16:09 UTC

Complexity

Size/Duplication

Importance

Duplication Side-by-Side

Filter issues like

MediaTypeChoices A
last analyzed 2018-07-01 16:09 UTC