ewandor/portainer_updater
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Adding generic requests method

Browse Source
This commit is contained in:
Gentile G
2024-04-25 10:49:47 +02:00
parent 5b3b02980c
commit f234ef2015
1 changed files with 17 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
main.py
View File

@@ -25,16 +25,26 @@ class PortainerClient:
"X-API-Key": self.access_token "X-API-Key": self.access_token
} }
def list_stacks(self): def _get(self, uri):
response = requests.get(f"{self.api_url}/stacks", headers=self._get_headers()) response = requests.get(f"{self.api_url}/{uri}", headers=self._get_headers())
return response.json() return response.json()
def list_stacks(self):
return self._get('stacks')
def get_stack(self, stack_id): def get_stack(self, stack_id):
response = requests.get(f"{self.api_url}/stacks/{stack_id}", headers=self._get_headers()) return self._get(f'stacks/{stack_id}')
return response.json()
def get_stack_file(self, stack_id): def get_stack_file(self, stack_id):
response = requests.get(f"{self.api_url}/stacks/{stack_id}/file", headers=self._get_headers()) return self._get(f'stacks/{stack_id}/file')
def _put(self, uri, payload):
response = requests.put(
f"{self.api_url}/uri",
headers=self._get_headers(),
json=payload
)
return response.json() return response.json()
def upgrade_stack(self, stack): def upgrade_stack(self, stack):
@@ -45,22 +55,12 @@ class PortainerClient:
"StackFileContent": file_response["StackFileContent"], "StackFileContent": file_response["StackFileContent"],
"Env": stack["Env"] "Env": stack["Env"]
} }
response = requests.put(
f"{self.api_url}/stacks/{stack['Id']}?endpointId={stack['EndpointId']}",
headers=self._get_headers(),
json=payload
)
return response.json() return self._put(f"stacks/{stack['Id']}?endpointId={stack['EndpointId']}", payload)
def list_stack_containers(self, endpoint, stack_name): def list_stack_containers(self, endpoint, stack_name):
filters = '{"label":["com.docker.compose.project=' + stack_name + '"]}' filters = '{"label":["com.docker.compose.project=' + stack_name + '"]}'
response = requests.get( return self._get(f'endpoints/{endpoint}/docker/containers/json?all=1&filters={filters}')
f"{self.api_url}/endpoints/{endpoint}/docker/containers/json?all=1&filters={filters}",
headers=self._get_headers()
)
return response.json()
def update_image(self, endpoint, image): def update_image(self, endpoint, image):
response = requests.post( response = requests.post(