|
1 | 1 | from typing import Optional, Dict |
2 | 2 |
|
3 | | -from mypy.semanal import names_modified_in_lvalue |
4 | 3 | from requests import Response |
5 | 4 |
|
6 | 5 | from scaleway_core.bridge import Zone as ScwZone |
@@ -78,36 +77,54 @@ def set_server_user_data( |
78 | 77 | self._throw_on_error(res) |
79 | 78 | return res |
80 | 79 |
|
81 | | - def get_all_server_user_data(self, server_id: str, zone: Optional[ScwZone] = None) -> GetAllServerUserDataResponse: |
| 80 | + def get_all_server_user_data( |
| 81 | + self, server_id: str, zone: Optional[ScwZone] = None |
| 82 | + ) -> GetAllServerUserDataResponse: |
82 | 83 | param_zone = validate_path_param("zone", zone or self.client.default_zone) |
83 | 84 | param_server_id = validate_path_param("server_id", server_id) |
84 | 85 |
|
85 | | - all_user_data_res = InstanceUtilsV1API.list_server_user_data(self, server_id=param_server_id, zone=param_zone) |
| 86 | + all_user_data_res = InstanceUtilsV1API.list_server_user_data( |
| 87 | + self, server_id=param_server_id, zone=param_zone |
| 88 | + ) |
86 | 89 |
|
87 | | - user_data: Dict[str,bytes] = {} |
| 90 | + user_data: Dict[str, bytes] = {} |
88 | 91 | for key in all_user_data_res.user_data: |
89 | | - value = InstanceUtilsV1API.get_server_user_data(self, server_id=param_server_id, key=key) |
| 92 | + value = InstanceUtilsV1API.get_server_user_data( |
| 93 | + self, server_id=param_server_id, key=key |
| 94 | + ) |
90 | 95 | print("value: ", value) |
91 | 96 | user_data[key] = value.content |
92 | 97 |
|
93 | 98 | res = GetAllServerUserDataResponse(user_data=user_data) |
94 | 99 |
|
95 | 100 | return res |
96 | 101 |
|
97 | | - def set_all_server_user_data(self, server_id: str, user_data: Dict[str, bytes], zone: Optional[ScwZone] = None): |
| 102 | + def set_all_server_user_data( |
| 103 | + self, |
| 104 | + server_id: str, |
| 105 | + user_data: Dict[str, bytes], |
| 106 | + zone: Optional[ScwZone] = None, |
| 107 | + ): |
98 | 108 | param_zone = validate_path_param("zone", zone or self.client.default_zone) |
99 | 109 | param_server_id = validate_path_param("server_id", server_id) |
100 | 110 |
|
101 | | - all_user_data_res = InstanceUtilsV1API.list_server_user_data(self, server_id=param_server_id, zone=param_zone) |
| 111 | + all_user_data_res = InstanceUtilsV1API.list_server_user_data( |
| 112 | + self, server_id=param_server_id, zone=param_zone |
| 113 | + ) |
102 | 114 | for key in all_user_data_res.user_data: |
103 | 115 | if user_data.get(key) is not None: |
104 | 116 | continue |
105 | | - InstanceUtilsV1API.delete_server_user_data(self, server_id=param_server_id, key=key) |
| 117 | + InstanceUtilsV1API.delete_server_user_data( |
| 118 | + self, server_id=param_server_id, key=key |
| 119 | + ) |
106 | 120 |
|
107 | 121 | for key in user_data: |
108 | | - InstanceUtilsV1API.set_server_user_data(self, server_id=param_server_id, zone=param_zone, key=key, content=user_data[key]) |
109 | | - |
110 | | - |
111 | | - |
112 | | - |
| 122 | + InstanceUtilsV1API.set_server_user_data( |
| 123 | + self, |
| 124 | + server_id=param_server_id, |
| 125 | + zone=param_zone, |
| 126 | + key=key, |
| 127 | + content=user_data[key], |
| 128 | + ) |
113 | 129 |
|
| 130 | + return None |
0 commit comments