1.1 --- a/imiptools/stores/database/common.py Thu Mar 10 01:43:31 2016 +0100
1.2 +++ b/imiptools/stores/database/common.py Fri Mar 11 01:33:45 2016 +0100
1.3 @@ -809,7 +809,7 @@
1.4 "select store_user, user_group from user_groups :condition",
1.5 columns, values)
1.6
1.7 - self.cursor.execute(query)
1.8 + self.cursor.execute(query, values)
1.9 return dict(self.cursor.fetchall())
1.10
1.11 def get_limits(self, quota):
1.12 @@ -826,9 +826,40 @@
1.13 "select user_group, quota_limit from quota_limits :condition",
1.14 columns, values)
1.15
1.16 - self.cursor.execute(query)
1.17 + self.cursor.execute(query, values)
1.18 return dict(self.cursor.fetchall())
1.19
1.20 + def set_limit(self, quota, group, limit):
1.21 +
1.22 + """
1.23 + For the given 'quota', set for a user 'group' the given 'limit' on
1.24 + resource usage.
1.25 + """
1.26 +
1.27 + columns = ["quota", "user_group"]
1.28 + values = [quota, group]
1.29 + setcolumns = ["quota_limit"]
1.30 + setvalues = [limit]
1.31 +
1.32 + query, values = self.get_query(
1.33 + "update quota_limits :set :condition",
1.34 + columns, values, setcolumns, setvalues)
1.35 +
1.36 + self.cursor.execute(query, values)
1.37 +
1.38 + if self.cursor.rowcount > 0:
1.39 + return True
1.40 +
1.41 + columns = ["quota", "user_group", "quota_limit"]
1.42 + values = [quota, group, limit]
1.43 +
1.44 + query, values = self.get_query(
1.45 + "insert into quota_limits (:columns) values (:values)",
1.46 + columns, values)
1.47 +
1.48 + self.cursor.execute(query, values)
1.49 + return True
1.50 +
1.51 # Free/busy period access for users within quota groups.
1.52
1.53 def get_freebusy(self, quota, user, mutable=False):