From 02dcfac930f0b45dd97f6d4851b161dce5217bd4 Mon Sep 17 00:00:00 2001 From: yugasun Date: Tue, 2 Feb 2021 17:25:52 +0800 Subject: [PATCH] fix: multi cookie bug --- serverless.component.yml | 2 +- src/_shims/severless_wsgi.py | 6 ++++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/serverless.component.yml b/serverless.component.yml index c2b404f..8a37f6c 100644 --- a/serverless.component.yml +++ b/serverless.component.yml @@ -1,5 +1,5 @@ name: flask -version: 0.0.9 +version: 0.0.10 author: 'Tencent Cloud, Inc' org: 'Tencent Cloud, Inc' description: Deploy a serverless Flask application onto Tencent SCF and API Gateway. diff --git a/src/_shims/severless_wsgi.py b/src/_shims/severless_wsgi.py index 2de91a3..71e0a94 100644 --- a/src/_shims/severless_wsgi.py +++ b/src/_shims/severless_wsgi.py @@ -55,11 +55,13 @@ def split_headers(headers): """ new_headers = {} - for key in headers.keys(): + for key in set(headers.keys()): values = headers.get_all(key) - if len(values) > 1: + if len(values) > 1 and key.lower() != 'set-cookie': for value, casing in zip(values, all_casings(key)): new_headers[casing] = value + elif key.lower() == 'set-cookie': + new_headers[key] = values elif len(values) == 1: new_headers[key] = values[0]