Create user_management.py

auth/user_management.py

@@ -0,0 +1,143 @@
+from datetime import datetime
+from .authentication import auth_manager
+import json
+import os
+
+class UserManagement:
+    def __init__(self, users_file='data/users.json'):
+        self.users_file = users_file
+        self.users = self.load_users()
+    
+    def load_users(self):
+        """بارگذاری کاربران از فایل"""
+        try:
+            if os.path.exists(self.users_file):
+                with open(self.users_file, 'r', encoding='utf-8') as f:
+                    return json.load(f)
+            return {}
+        except Exception as e:
+            print(f"Error loading users: {e}")
+            return {}
+    
+    def save_users(self):
+        """ذخیره کاربران در فایل"""
+        try:
+            os.makedirs(os.path.dirname(self.users_file), exist_ok=True)
+            with open(self.users_file, 'w', encoding='utf-8') as f:
+                json.dump(self.users, f, ensure_ascii=False, indent=2)
+            return True
+        except Exception as e:
+            print(f"Error saving users: {e}")
+            return False
+    
+    def register_user(self, username, email, password, role='student', **extra_data):
+        """ثبت نام کاربر جدید"""
+        # بررسی تکراری نبودن username و email
+        for user_id, user_data in self.users.items():
+            if user_data['username'] == username:
+                return False, "Username already exists"
+            if user_data['email'] == email:
+                return False, "Email already exists"
+        
+        # ایجاد کاربر جدید
+        user_id = str(len(self.users) + 1)
+        hashed_password = auth_manager.hash_password(password)
+        
+        self.users[user_id] = {
+            'user_id': user_id,
+            'username': username,
+            'email': email,
+            'password': hashed_password,
+            'role': role,
+            'created_at': datetime.now().isoformat(),
+            'last_login': None,
+            'profile': {
+                'full_name': extra_data.get('full_name', ''),
+                'bio': extra_data.get('bio', ''),
+                'avatar': extra_data.get('avatar', '')
+            },
+            'progress': {
+                'total_score': 0,
+                'completed_lessons': [],
+                'current_lesson': 1,
+                'achievements': []
+            },
+            'is_active': True
+        }
+        
+        if self.save_users():
+            return True, user_id
+        return False, "Error saving user data"
+    
+    def authenticate_user(self, username, password):
+        """احراز هویت کاربر"""
+        for user_id, user_data in self.users.items():
+            if user_data['username'] == username and user_data['is_active']:
+                if auth_manager.check_password(user_data['password'], password):
+                    # به‌روزرسانی آخرین زمان ورود
+                    user_data['last_login'] = datetime.now().isoformat()
+                    self.save_users()
+                    
+                    return True, {
+                        'user_id': user_id,
+                        'username': user_data['username'],
+                        'role': user_data['role'],
+                        'profile': user_data['profile']
+                    }
+                else:
+                    return False, "Invalid password"
+        
+        return False, "User not found"
+    
+    def get_user(self, user_id):
+        """دریافت اطلاعات کاربر"""
+        return self.users.get(user_id)
+    
+    def update_user_profile(self, user_id, **updates):
+        """به‌روزرسانی پروفایل کاربر"""
+        if user_id not in self.users:
+            return False, "User not found"
+        
+        user = self.users[user_id]
+        
+        # به‌روزرسانی فیلدهای اصلی
+        if 'email' in updates:
+            user['email'] = updates['email']
+        
+        # به‌روزرسانی پروفایل
+        if 'profile' in updates:
+            for key, value in updates['profile'].items():
+                if value is not None:
+                    user['profile'][key] = value
+        
+        # به‌روزرسانی progress
+        if 'progress' in updates:
+            for key, value in updates['progress'].items():
+                if key == 'total_score':
+                    user['progress'][key] += value
+                elif key == 'completed_lessons':
+                    if value not in user['progress'][key]:
+                        user['progress'][key].append(value)
+                else:
+                    user['progress'][key] = value
+        
+        if self.save_users():
+            return True, "Profile updated successfully"
+        return False, "Error updating profile"
+    
+    def deactivate_user(self, user_id):
+        """غیرفعال کردن کاربر"""
+        if user_id in self.users:
+            self.users[user_id]['is_active'] = False
+            return self.save_users()
+        return False
+    
+    def get_user_progress(self, user_id):
+        """دریافت پیشرفت کاربر"""
+        user = self.get_user(user_id)
+        if user:
+            return user.get('progress', {})
+        return None
+
+# نمونه singleton
+user_manager = UserManagement()