12
12
13
13
log = logging .getLogger (__name__ )
14
14
15
- PRODUCTION_DOMAIN = settings .PRODUCTION_DOMAIN
16
- API_HOST = settings .SLUMBER_API_HOST
17
- USER = settings .SLUMBER_USERNAME
18
- PASS = settings .SLUMBER_PASSWORD
19
-
20
15
21
16
class DrfJsonSerializer (serialize .JsonSerializer ):
22
17
@@ -31,7 +26,7 @@ def dumps(self, data):
31
26
32
27
def setup_api ():
33
28
session = requests .Session ()
34
- if API_HOST .startswith ('https' ):
29
+ if settings . SLUMBER_API_HOST .startswith ('https' ):
35
30
# Only use the HostHeaderSSLAdapter for HTTPS connections
36
31
adapter_class = host_header_ssl .HostHeaderSSLAdapter
37
32
else :
@@ -50,12 +45,12 @@ def setup_api():
50
45
)
51
46
52
47
session .mount (
53
- API_HOST ,
48
+ settings . SLUMBER_API_HOST ,
54
49
adapter_class (max_retries = retry ),
55
50
)
56
- session .headers .update ({'Host' : PRODUCTION_DOMAIN })
51
+ session .headers .update ({'Host' : settings . PRODUCTION_DOMAIN })
57
52
api_config = {
58
- 'base_url' : '%s/api/v2/' % API_HOST ,
53
+ 'base_url' : '%s/api/v2/' % settings . SLUMBER_API_HOST ,
59
54
'serializer' : serialize .Serializer (
60
55
default = 'json-drf' ,
61
56
serializers = [
@@ -65,13 +60,13 @@ def setup_api():
65
60
),
66
61
'session' : session ,
67
62
}
68
- if USER and PASS :
63
+ if settings . SLUMBER_USERNAME and settings . SLUMBER_PASSWORD :
69
64
log .debug (
70
65
'Using slumber v2 with user %s, pointed at %s' ,
71
- USER ,
72
- API_HOST ,
66
+ settings . SLUMBER_USERNAME ,
67
+ settings . SLUMBER_API_HOST ,
73
68
)
74
- session .auth = (USER , PASS )
69
+ session .auth = (settings . SLUMBER_USERNAME , settings . SLUMBER_PASSWORD )
75
70
else :
76
71
log .warning ('SLUMBER_USERNAME/PASSWORD settings are not set' )
77
72
return API (** api_config )
0 commit comments