mcp-security-framework 0.1.0__py3-none-any.whl → 1.1.0__py3-none-any.whl

mcp_security_framework/examples/django_example.py

@@ -21,13 +21,46 @@ import os
 import json
 import logging
 from typing import Dict, List, Any, Optional
-from datetime import datetime, timedelta
+from datetime import datetime, timedelta, timezone
+
+# Configure Django settings before importing Django modules
+import django
+from django.conf import settings
+
+if not settings.configured:
+    settings.configure(
+        DEBUG=True,
+        SECRET_KEY='django-insecure-test-key-for-examples',
+        INSTALLED_APPS=[
+            'django.contrib.auth',
+            'django.contrib.contenttypes',
+            'django.contrib.sessions',
+        ],
+        DATABASES={
+            'default': {
+                'ENGINE': 'django.db.backends.sqlite3',
+                'NAME': ':memory:',
+            }
+        },
+        MIDDLEWARE=[
+            'django.middleware.security.SecurityMiddleware',
+            'django.contrib.sessions.middleware.SessionMiddleware',
+            'django.middleware.common.CommonMiddleware',
+            'django.middleware.csrf.CsrfViewMiddleware',
+            'django.contrib.auth.middleware.AuthenticationMiddleware',
+        ],
+        ROOT_URLCONF=None,
+    )
+    django.setup()
 
 from django.http import HttpRequest, HttpResponse, JsonResponse
 from django.views.decorators.csrf import csrf_exempt
 from django.views.decorators.http import require_http_methods
-from django.middleware.base import BaseMiddleware
-from django.conf import settings
+try:
+    from django.middleware.base import BaseMiddleware
+except ImportError:
+    # Fallback for Django 5.x
+    from django.utils.deprecation import MiddlewareMixin as BaseMiddleware
 from django.urls import path, include
 from django.contrib.auth.models import User
 from django.contrib.auth.decorators import login_required, permission_required
@@ -89,10 +122,10 @@ class DjangoSecurityMiddleware(BaseMiddleware):
                 security_headers=DEFAULT_SECURITY_HEADERS
             ),
             ssl=SSLConfig(
-                enabled=True,
-                cert_file="certs/server.crt",
-                key_file="certs/server.key",
-                ca_cert_file="certs/ca.crt",
+                enabled=False,  # Disable SSL for example
+                cert_file=None,
+                key_file=None,
+                ca_cert_file=None,
                 verify_mode="CERT_REQUIRED",
                 min_version="TLSv1.2"
             ),
@@ -113,7 +146,7 @@ class DjangoSecurityMiddleware(BaseMiddleware):
                 "exempt_roles": ["admin"]
             },
             permissions={
-                "enabled": True,
+                "enabled": False,  # Disable permissions for example
                 "roles_file": "config/roles.json",
                 "default_role": "user",
                 "hierarchy_enabled": True
@@ -376,7 +409,7 @@ class HealthCheckView(View):
         """Handle GET request."""
         return JsonResponse({
             "status": "healthy",
-            "timestamp": datetime.utcnow().isoformat(),
+            "timestamp": datetime.now(timezone.utc).isoformat(),
             "version": "1.0.0"
         })
 
@@ -407,7 +440,7 @@ class UserProfileView(View):
             "username": user_info.get("username"),
             "roles": user_info.get("roles", []),
             "permissions": user_info.get("permissions", []),
-            "last_login": datetime.utcnow().isoformat()
+            "last_login": datetime.now(timezone.utc).isoformat()
         })
 
 
@@ -461,7 +494,7 @@ class DataView(View):
             "id": "data_123",
             "created_by": user_info.get("username"),
             "data": data,
-            "created_at": datetime.utcnow().isoformat()
+            "created_at": datetime.now(timezone.utc).isoformat()
         })
 
 
@@ -535,7 +568,7 @@ class DjangoExample:
             "auth_enabled": True,
             "rate_limiting_enabled": True,
             "permissions_enabled": True,
-            "timestamp": datetime.utcnow().isoformat()
+            "timestamp": datetime.now(timezone.utc).isoformat()
         }