From: Guillaume Pellerin <yomguy@parisson.com>
Date: Tue, 10 Feb 2015 09:50:18 +0000 (+0100)
Subject: Auto reject script if file is too large
X-Git-Tag: 1.1~279
X-Git-Url: https://git.parisson.com/?a=commitdiff_plain;h=159a7cf066c57e6d564d62b5021769fbe89da8a1;p=teleforma.git

Auto reject script if file is too large
---

diff --git a/teleforma/exam/models.py b/teleforma/exam/models.py
index 73bf468a..b2781503 100644
--- a/teleforma/exam/models.py
+++ b/teleforma/exam/models.py
@@ -78,6 +78,8 @@ REJECT_REASON = (('unreadable', _('unreadable')),
 cache_path = settings.MEDIA_ROOT + 'cache/'
 script_path = settings.MEDIA_ROOT + 'scripts/'
 
+SCRIPT_MAX_SIZE = 50000000
+
 
 def sha1sum_file(filename):
     '''
@@ -379,6 +381,10 @@ class Script(BaseResource):
             self.auto_reject('wrong format')
             return
 
+        if os.stat(self.file.path).st_size > SCRIPT_MAX_SIZE:
+            self.auto_reject('file too large')
+            return
+
         if not self.status == 0 and self.file:
             if not self.box_uuid:
                 self.uuid_link()