From: Michael Tremer <michael.tremer@ipfire.org>
Date: Wed, 20 Sep 2023 14:14:13 +0000 (+0000)
Subject: users: Set default storage quota for users
X-Git-Url: http://git.ipfire.org/?a=commitdiff_plain;h=d24cf3ce3be714b758a384b140f03d05c96d6275;p=pbs.git

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 44a2b615..12b1c9c8 100644
--- 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)