gen_fun()

Complexity

Conditions

3

Size

Total Lines

8

Duplication

Lines	0
Ratio	0 %

Importance

Changes	1
Bugs	0	Features	0

1 Method

Rating	Name	Duplication	Size	Complexity
A	col_name()	0	6	2

import inspect
import json
import sys

from django.conf import settings
from django.contrib import admin
from django.db.models import ForeignKey
from django.utils.html import format_html

from chat import models

exclude_auto = ()
model_classes = (class_name[1] for class_name in inspect.getmembers(sys.modules[models.__name__], inspect.isclass)
					if class_name[1].__module__ == models.__name__ and class_name[0] not in exclude_auto)

def gen_fun(field):
	def col_name(o):
		attr = getattr(o, field)
		v = json.dumps(attr)
		if len(v) > 50:
			v = v[:50]
		return v
	return col_name


main_fields = {
	'user': gen_fun('username'),
	'room': gen_fun('name'),
	'issue': gen_fun('content'),
	'ip address': gen_fun('ip'),
	'message': gen_fun('content'),
	'subscription': gen_fun('id'),
}


def country(instance):
	iso2 = instance.country_code if instance.country_code else "None"
	return format_html("<span style='white-space:nowrap'><img src='{}/flags/{}.png' /> {}</span>",
							 settings.STATIC_URL,
							 iso2.lower(),
							 instance.country)
extra_fields = {
	'ip address': (country,)
}

exclude_fields = {
	'ip address': ('country',)
}

for model in model_classes:
	fields = []
	vname = model._meta.verbose_name
	class_struct = {'fields': fields, 'list_display': fields}
	for field in model._meta.fields:
		if exclude_fields.get(vname) is not None and field.name in extra_fields.get(vname):
			break
		if isinstance(field, ForeignKey):
			def gen_link(field):
				def link(obj):
					print(field)
					another = getattr(obj, field.name)
					if another is None:
						return "Null"
					else:
						another_name = another._meta.verbose_name
						link = '/admin/chat/{}/{}/change'.format(another_name, another.id)
						return u'<a href="%s">%s</a>' % (link, main_fields[another_name](another))
				link.allow_tags = True
				link.__name__ = str(field.name)
				return link
			fields.append(gen_link(field))
		else:
			fields.append(field.name)

	if extra_fields.get(vname) is not None:
		fields.extend(extra_fields.get(vname))
	admin.site.register(model, type(
		'SubClass',
		(admin.ModelAdmin,),
		class_struct
	))


# class CountryFilter(SimpleListFilter):
# 	title = 'country'
# 	parameter_name = 'country'
#
# 	def lookups(self, request, model_admin):
# 		query_set = model_admin.model.objects.values('ip__country').annotate(count=Count('ip__country'))
# 		return [(c['ip__country'], '%s(%s)' % (c['ip__country'], c['count'])) for c in query_set]
#
# 	def queryset(self, request, queryset):
# 		if self.value():
# 			return queryset.filter(ip__country=self.value())
# 		else:
# 			return queryset
#
#
# @admin.register(UserJoinedInfo)
# class UserLocation(admin.ModelAdmin):
# 	list_display = ["time", "link_to_B"]
#
# 	def link_to_B(self, obj):
# 		link = urlresolvers.reverse("admin:chat_user_change", args=[obj.Object.id])  # model name has to be lowercase
# 		return u'<a href="%s">%s</a>' % (link, obj.B.name)
#
# 	link_to_B.allow_tags = True