Coverage for /builds/BuildGrid/buildgrid/buildgrid/browser/utils.py: 90.91%

1# Copyright (C) 2021 Bloomberg LP 

2# 

3# Licensed under the Apache License, Version 2.0 (the "License"); 

4# you may not use this file except in compliance with the License. 

5# You may obtain a copy of the License at 

6# 

7# <http://www.apache.org/licenses/LICENSE-2.0> 

8# 

9# Unless required by applicable law or agreed to in writing, software 

10# distributed under the License is distributed on an "AS IS" BASIS, 

11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 

12# See the License for the specific language governing permissions and 

13# limitations under the License. 

14 

15from aiohttp_middlewares.cors import ACCESS_CONTROL_ALLOW_CREDENTIALS, ACCESS_CONTROL_ALLOW_ORIGIN 

16 

17 

18TARBALL_DIRECTORY_PREFIX = 'bgd-browser-tarball-' 

19 

20 

21def get_cors_headers(origin, origins, allow_all): 

22 headers = {} 

23 if origin: 

24 headers[ACCESS_CONTROL_ALLOW_CREDENTIALS] = "true" 

25 if origin in origins: 

26 headers[ACCESS_CONTROL_ALLOW_ORIGIN] = origin 

27 elif allow_all: 

28 headers[ACCESS_CONTROL_ALLOW_ORIGIN] = "*" 

29 return headers