Models/server: Fetch builds belonging to same server

Signed-off-by: ZorEl212 <[email protected]>
ZorEl212 · Oct 26, 2024 · 807782a · 807782a
1 parent 64ead01
commit 807782a
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/models/server.py b/models/server.py
@@ -6,7 +6,15 @@ class Server(BaseModel):
     userId = ""
     serverName = ""
     ipAddress = ""
+    server_users = set()
 
     @property
     def availableBuilds(self):
-        pass
+        from models import storage
+        return storage.all('Build', 'server_id', self.id)
+
+    def add_user(self, userId):
+        if userId not in self.server_users:
+            self.server_users.add(userId)
+            return True
+        return False