Limit moderation by category: subcategoryuserlimit model - modify admin for config

1 files changed, 17 insertions, 0 deletions

diff --git a/chimere/models.py b/chimere/models.py
index 8be874c..f333589 100644
--- a/chimere/models.py
+++ b/chimere/models.py
@@ -310,6 +310,23 @@ class SubCategory(models.Model):
     def item_nb(self):
         return Marker.objects.filter(categories=self).count()
 
+
+class SubCategoryUserLimit(models.Model):
+    """
+    Moderation limit for user by category
+    """
+    subcategory = models.ForeignKey(
+        SubCategory, related_name='limited_for_user')
+    user = models.ForeignKey(User, related_name='subcategory_limit')
+
+    def __unicode__(self):
+        return u"{} / {}".format(self.user, self.subcategory)
+
+    class Meta:
+        verbose_name = _(u"Sub-category limit for user")
+        verbose_name_plural = _(u"Sub-category limits for users")
+
+
 STATUS = (('S', _(u'Submited')),
           ('A', _(u'Available')),
           ('M', _(u'Modified')),