diff --git a/markov/models.py b/markov/models.py index 8a568f6..1e38219 100644 --- a/markov/models.py +++ b/markov/models.py @@ -49,6 +49,9 @@ class MarkovState(models.Model): context = models.ForeignKey(MarkovContext) class Meta: + permissions = { + ('import_log_file', "Can import states from a log file"), + } unique_together = ('context', 'k1', 'k2', 'v') def __unicode__(self): diff --git a/markov/views.py b/markov/views.py index 715cdb5..662a186 100644 --- a/markov/views.py +++ b/markov/views.py @@ -5,6 +5,7 @@ markov/views.py --- manipulate markov data import logging +from django.contrib.auth.decorators import permission_required from django.http import HttpResponse from django.shortcuts import render @@ -21,6 +22,7 @@ def index(request): return HttpResponse() +@permission_required('import_log_file', raise_exception=True) def import_file(request): """Accept a file upload and turn it into markov stuff.