diff --git a/sat/plugins/plugin_comp_file_sharing.py b/sat/plugins/plugin_comp_file_sharing.py index a6fc4b4d..1093d1c7 100644 --- a/sat/plugins/plugin_comp_file_sharing.py +++ b/sat/plugins/plugin_comp_file_sharing.py @@ -410,13 +410,14 @@ class FileSharing: def getQuota(self, client, entity): """Return maximum size allowed for all files for entity""" - # TODO: handle special entities like admins quotas = self.host.memory.getConfig("component file-sharing", "quotas_json", {}) - entity_bare_s = entity.userhost() - try: - quota = quotas["jids"][entity_bare_s] - except KeyError: - quota = quotas.get("users") + if self.host.memory.isAdminJID(entity): + quota = quotas.get("admins") + else: + try: + quota = quotas["jids"][entity.userhost()] + except KeyError: + quota = quotas.get("users") return None if quota is None else utils.parseSize(quota) async def generate_thumbnails(self, extra: dict, image_path: Path):