feat: migrate user management from json to mongodb

core/scripts/db/database.py

@@ -0,0 +1,41 @@
+import pymongo
+from bson.objectid import ObjectId
+
+class Database:
+    def __init__(self, db_name="blitz_panel", collection_name="users"):
+        try:
+            self.client = pymongo.MongoClient("mongodb://localhost:27017/")
+            self.db = self.client[db_name]
+            self.collection = self.db[collection_name]
+            self.client.server_info()
+        except pymongo.errors.ConnectionFailure as e:
+            print(f"Could not connect to MongoDB: {e}")
+            raise
+
+    def add_user(self, user_data):
+        username = user_data.pop('username', None)
+        if not username:
+            raise ValueError("Username is required")
+
+        if self.collection.find_one({"_id": username.lower()}):
+            return None
+
+        user_data['_id'] = username.lower()
+        return self.collection.insert_one(user_data)
+
+    def get_user(self, username):
+        return self.collection.find_one({"_id": username.lower()})
+
+    def get_all_users(self):
+        return list(self.collection.find({}))
+
+    def update_user(self, username, updates):
+        return self.collection.update_one({"_id": username.lower()}, {"$set": updates})
+
+    def delete_user(self, username):
+        return self.collection.delete_one({"_id": username.lower()})
+
+try:
+    db = Database()
+except pymongo.errors.ConnectionFailure:
+    db = None