|
1 | 1 | from gql import gql
|
2 | 2 | from ..graphql import graphql_client
|
| 3 | +from .common import ListLogs |
| 4 | +from .. import logger as sdk_logger |
3 | 5 |
|
4 | 6 |
|
5 | 7 | def create_deployment(name, project_id, spec, cluster_id=None, api_key=None):
|
@@ -215,3 +217,29 @@ def delete_deployment(id, api_key=None):
|
215 | 217 | }
|
216 | 218 | }
|
217 | 219 | return client.execute(query, variable_values=params)['deleteDeployment']
|
| 220 | + |
| 221 | + |
| 222 | +# my disappointment is immeasurable |
| 223 | +# and my day is ruined |
| 224 | +class ListDeploymentV3Logs(ListLogs): |
| 225 | + def _get_request_params(self, kwargs): |
| 226 | + params = { |
| 227 | + 'gradientDeploymentId': kwargs['id'], |
| 228 | + 'line': kwargs['line'], |
| 229 | + 'limit': kwargs['limit'] |
| 230 | + } |
| 231 | + return params |
| 232 | + |
| 233 | + |
| 234 | +def get_deployment_logs(deployment_id, line=1, limit=10000, api_key=None): |
| 235 | + DeploymentLogs = ListDeploymentV3Logs( |
| 236 | + api_key=api_key, |
| 237 | + logger=sdk_logger.MuteLogger()) |
| 238 | + return DeploymentLogs.list(id=deployment_id, line=line, limit=limit) |
| 239 | + |
| 240 | + |
| 241 | +def yield_deployment_logs(deployment_id, line=1, limit=10000, api_key=None): |
| 242 | + DeploymentLogs = ListDeploymentV3Logs( |
| 243 | + api_key=api_key, |
| 244 | + logger=sdk_logger.MuteLogger()) |
| 245 | + return DeploymentLogs.yield_logs(id=deployment_id, line=line, limit=limit) |
0 commit comments