users: Set default storage quota for users

Signed-off-by: Michael Tremer <michael.tremer@ipfire.org>

diff --git a/src/buildservice/users.py b/src/buildservice/users.py
index 44a2b615d329f11a012f1dbc0e7923e60c045fc1..12b1c9c8392018e63eee2254314fd3ea62399979 100644 (file)
--- a/src/buildservice/users.py
+++ b/src/buildservice/users.py
@@ -31,6 +31,8 @@ from . import httpclient
 
 from .decorators import *
 
+DEFAULT_STORAGE_QUOTA = 256 * 1024 * 1024 # 256 MiB
+
 # Setup logging
 log = logging.getLogger("pbs.users")
 
@@ -170,7 +172,7 @@ class Users(base.Object):
 
                return results[0]
 
-       def create(self, name, notify=False, _attrs=None):
+       def create(self, name, notify=False, storage_quota=None, _attrs=None):
                """
                        Creates a new user
                """
@@ -178,17 +180,25 @@ class Users(base.Object):
                if _attrs:
                        _attrs = pickle.dumps(_attrs)
 
+               # Set default for storage quota
+               if storage_quota is None:
+                       storage_quota = DEFAULT_STORAGE_QUOTA
+
                user = self._get_user("""
                        INSERT INTO
-                               users(
-                                       name,
-                                       _attrs
-                               )
+                               users
+                       (
+                               name,
+                               storage_quota,
+                               _attrs
+                       )
                        VALUES
-                               (%s, %s)
+                       (
+                               %s, %s, %s
+                       )
                        RETURNING
                                *
-                       """, name, _attrs,
+                       """, name, storage_quota, _attrs,
                )
 
                log.debug("Created user %s" % user)