From 4d0b9f929961d1b9fc1750400ba0a511aa446119 Mon Sep 17 00:00:00 2001
From: Fabio Rauber <fabiorauber@gmail.com>
Date: Mon, 17 Apr 2017 12:09:48 -0300
Subject: [PATCH] Add setlimits and setmanager commands to manage.py

---
 admin/manage.py | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/admin/manage.py b/admin/manage.py
index a277d719..4097811f 100644
--- a/admin/manage.py
+++ b/admin/manage.py
@@ -69,6 +69,26 @@ def alias(localpart, domain_name, destination):
     db.session.add(alias)
     db.session.commit()
 
+# Set limits to a domain
+@manager.command
+def setlimits(domain_name, max_users, max_aliases, max_quota_bytes):
+  domain = models.Domain.query.get(domain_name)
+  domain.max_users = max_users
+  domain.max_aliases = max_aliases
+  domain.max_quota_bytes = max_quota_bytes
+
+  db.session.add(domain)
+  db.session.commit()
+
+# Make the user manager of a domain
+@manager.command
+def setmanager(domain_name, user_name='manager'):
+  domain = models.Domain.query.get(domain_name)
+  manageruser = models.User.query.get(user_name + '@' + domain_name)
+  domain.managers.append(manageruser)
+  db.session.add(domain)
+  db.session.commit()
+
 
 if __name__ == "__main__":
     manager.run()