def test_function(self):
"""register_view works as a function."""
site = AdminSitePlus()
def foo(request):
return 'foo'
site.register_view('foo', view=foo)
urls = site.get_urls()
assert any(u.resolve('foo') for u in urls)
def test_function(self):
"""register_view works as a function."""
site = AdminSitePlus()
def foo(request):
return 'foo'
site.register_view('foo', view=foo)
urls = site.get_urls()
assert any(u.resolve('foo') for u in urls)
def test_visibility(self):
"""Make sure visibility works."""
site = AdminSitePlus()
req_factory = RequestFactory()
def always_visible(request):
return 'i am here'
site.register_view('always-visible', view=always_visible, visible=True)
def always_hidden(request):
return 'i am here, but not shown'
site.register_view('always-hidden', view=always_visible, visible=False)
cond = lambda req: req.user.pk == 1
b = lambda s: s.encode('ascii') if hasattr(s, 'encode') else s
@site.register_view(r'conditional-view', visible=cond)
class ConditionallyVisible(View):
def get(self, request):
return 'hi there'
urls = site.get_urls()
assert any(u.resolve('always-visible') for u in urls)
assert any(u.resolve('always-hidden') for u in urls)
assert any(u.resolve('conditional-view') for u in urls)
class MockUser(object):
is_active = True
is_staff = True
def __init__(self, pk):
self.pk = pk
self.id = pk
req_show = req_factory.get('/admin/')
req_show.user = MockUser(1)
result = site.index(req_show).render().content
assert b('always-visible') in result
assert b('always-hidden') not in result
assert b('conditional-view') in result
req_hide = req_factory.get('/admin/')
req_hide.user = MockUser(2)
result = site.index(req_hide).render().content
assert b('always-visible') in result
assert b('always-hidden') not in result
assert b('conditional-view') not in result
def test_path(self):
"""Setting the path works correctly."""
site = AdminSitePlus()
def foo(request):
return 'foo'
site.register_view('foo', view=foo)
site.register_view('bar/baz', view=foo)
site.register_view('baz-qux', view=foo)
urls = site.get_urls()
foo_urls = [u for u in urls if u.resolve('foo')]
self.assertEqual(1, len(foo_urls))
bar_urls = [u for u in urls if u.resolve('bar/baz')]
self.assertEqual(1, len(bar_urls))
qux_urls = [u for u in urls if u.resolve('baz-qux')]
self.assertEqual(1, len(qux_urls))
def test_path(self):
"""Setting the path works correctly."""
site = AdminSitePlus()
def foo(request):
return 'foo'
site.register_view('foo', view=foo)
site.register_view('bar/baz', view=foo)
site.register_view('baz-qux', view=foo)
urls = site.get_urls()
foo_urls = [u for u in urls if u.resolve('foo')]
self.assertEqual(1, len(foo_urls))
bar_urls = [u for u in urls if u.resolve('bar/baz')]
self.assertEqual(1, len(bar_urls))
qux_urls = [u for u in urls if u.resolve('baz-qux')]
self.assertEqual(1, len(qux_urls))
def test_path(self):
"""Setting the path works correctly."""
site = AdminSitePlus()
def foo(request):
return 'foo'
site.register_view('foo', view=foo)
site.register_view('bar/baz', view=foo)
site.register_view('baz-qux', view=foo)
urls = site.get_urls()
matches = lambda u: lambda p: p.resolve(u)
foo_urls = filter(matches('foo'), urls)
self.assertEqual(1, len(foo_urls))
bar_urls = filter(matches('bar/baz'), urls)
self.assertEqual(1, len(bar_urls))
qux_urls = filter(matches('baz-qux'), urls)
self.assertEqual(1, len(qux_urls))
actions = [
export_to_app_excel, export_all_info_excel, email_guests_initial,
email_guests_reminder, statistics_admin_action, export_rides_action,
set_to_default_action
]
def save_model(self, request, obj, form, change):
"""Add and remove guest person using the checkbox"""
obj.save()
has_guest = obj.has_guest_person()
person_name = request.POST["person_set-0-name"]
is_english = str_is_english(person_name)
if not has_guest and obj.with_guest:
obj.create_guest(is_english)
if has_guest and not obj.with_guest:
people = Person.objects.filter(invitation=obj.id)
for person in people:
if person.is_guest():
person.delete()
obj.save()
def statistics(request):
stats = Statistics()
return render(request, 'admin/statistics.html', {'stats': stats})
site.register_view('statistics.html', view=statistics)
site.register(Invitation, InvitationAdmin)
