Skip to content

Commit c0c982f

Browse files
committed
fix no policy issue
1 parent 05eb4ea commit c0c982f

File tree

1 file changed

+12
-11
lines changed

1 file changed

+12
-11
lines changed

src/api/handlers/job_api.py

Lines changed: 12 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -94,18 +94,19 @@ def generate_batch_token(self, service_token, ttl="1h"):
9494
:param ttl: Time-to-live for the batch token.
9595
:return: The generated batch token.
9696
"""
97-
if self.policies is None:
98-
self.policies = self.get_policies(service_token)
99-
if 'token-creator' in self.policies:
100-
self.policies.remove('token-creator')
101-
102-
batch_payload = {
103-
"type": "batch",
104-
"policies": self.policies or ['default'],
105-
"ttl": ttl
106-
}
107-
url = self.base_url + '/v1/' + self.namespace + '/auth/token/create' if self.namespace else self.base_url + '/v1/auth/token/create'
10897
try:
98+
if self.policies is None:
99+
self.policies = self.get_policies(service_token) or ['default']
100+
if self.policies and 'token-creator' in self.policies:
101+
self.policies.remove('token-creator')
102+
103+
batch_payload = {
104+
"type": "batch",
105+
"policies": self.policies or ['default'],
106+
"ttl": ttl
107+
}
108+
url = self.base_url + '/v1/' + self.namespace + '/auth/token/create' if self.namespace else self.base_url + '/v1/auth/token/create'
109+
109110
for i in range(0, 10):
110111
res = requests.post(url=url, json=batch_payload, headers={"X-Vault-Token": service_token}, verify=False)
111112
if res.status_code == 200:

0 commit comments

Comments
 (0)