fix: 修复swagger接口文档中schema不正确的问题

timeflies99 · Nov 13, 2024 · 6f84239 · 6f84239
1 parent 4a043cf
commit 6f84239
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/apps/common/config/swagger_conf.py b/apps/common/config/swagger_conf.py
@@ -20,3 +20,10 @@ def get_tags(self, operation_keys=None):
         if "api" in tags and operation_keys:
             return [tags_dict.get(operation_keys[1]) if operation_keys[1] in tags_dict else operation_keys[1]]
         return tags
+    def get_schema(self, request=None, public=False):
+        schema = super().get_schema(request, public)
+        if request.is_secure():
+            schema.schemes = ['https']
+        else:
+            schema.schemes = ['http']
+        return schema
diff --git a/apps/smartdoc/settings/base.py b/apps/smartdoc/settings/base.py
@@ -26,6 +26,8 @@
     'default': CONFIG.get_db_setting()
 }
 
+SECURE_PROXY_SSL_HEADER = ('HTTP_X_FORWARDED_PROTO', 'https')
+
 # Application definition
 
 INSTALLED_APPS = [