|
8 | 8 | from bravado.requests_client import Authenticator |
9 | 9 |
|
10 | 10 |
|
11 | | -QUOTATION_PARAMS = ['uuids', 'txids', 'identifiers', 'states'] |
| 11 | +EXCHANGE_PARAMS = ['uuids', 'txids', 'identifiers', 'states'] |
| 12 | +QUOTATION_PARAMS = ['uuids', 'txids', 'identifiers',] |
| 13 | +QUERY_PARAMS = set(EXCHANGE_PARAMS) |
12 | 14 | MAPPER = 'swg_mapper.json' |
13 | 15 |
|
14 | 16 |
|
| 17 | +QUERY_PARAMS.update(QUOTATION_PARAMS) |
| 18 | + |
| 19 | + |
15 | 20 | class APIKeyAuthenticator(Authenticator): |
16 | 21 |
|
17 | 22 | def __init__( |
@@ -68,15 +73,15 @@ def generate_query(self, params): |
68 | 73 | query = urlencode({ |
69 | 74 | k: v |
70 | 75 | for k, v in params.items() |
71 | | - if k not in QUOTATION_PARAMS |
| 76 | + if k not in QUERY_PARAMS |
72 | 77 | }) |
73 | | - for quotation in QUOTATION_PARAMS: |
74 | | - if params.get(quotation): |
75 | | - param = params.pop(quotation) |
76 | | - params[f"{quotation}[]"] = param |
| 78 | + for query_param in QUERY_PARAMS: |
| 79 | + if params.get(query_param): |
| 80 | + param = params.pop(query_param) |
| 81 | + params[f"{query_param}[]"] = param |
77 | 82 | query_params = '&'.join([ |
78 | | - f"{quotation}[]={q}" |
79 | | - for q in quotation |
| 83 | + f"{query_param}[]={q}" |
| 84 | + for q in query_param |
80 | 85 | ]) |
81 | 86 | query = f"{query}&{query_params}" if query else query_params |
82 | 87 | return query |
0 commit comments