From 1f69f9d27ebf64d2464b7a1cd4c8a6a17a8d6cf4 Mon Sep 17 00:00:00 2001
From: Usman Nasir <usman@cyberpersons.com>
Date: Mon, 5 Apr 2021 13:50:57 +0500
Subject: [PATCH] enforce disk limits check column

---
 packages/models.py  | 1 +
 plogical/upgrade.py | 6 ++++++
 2 files changed, 7 insertions(+)

diff --git a/packages/models.py b/packages/models.py
index ea0403ff5..431c2b866 100755
--- a/packages/models.py
+++ b/packages/models.py
@@ -16,3 +16,4 @@ class Package(models.Model):
     ftpAccounts = models.IntegerField(default=0)
     allowedDomains = models.IntegerField(default=0)
     allowFullDomain = models.IntegerField(default=1)
+    enforceDiskLimits = models.IntegerField(default=0)
diff --git a/plogical/upgrade.py b/plogical/upgrade.py
index 1fe554c6e..55a2ff9b1 100755
--- a/plogical/upgrade.py
+++ b/plogical/upgrade.py
@@ -1372,6 +1372,12 @@ imap_folder_list_limit = 0
             except:
                 pass
 
+            query = "ALTER TABLE packages_package ADD COLUMN enforceDiskLimits INT DEFAULT 0;"
+            try:
+                cursor.execute(query)
+            except:
+                pass
+
             try:
                 connection.close()
             except: