Skip to content

Commit 9d94822

Browse files
author
zhongzichao
authored
fix stop job failed (#976)
* fix stop job failed
1 parent ac9b38c commit 9d94822

File tree

1 file changed

+24
-15
lines changed

1 file changed

+24
-15
lines changed

uninstaller/pre-delete.py

+24-15
Original file line numberDiff line numberDiff line change
@@ -31,18 +31,7 @@
3131
import os
3232

3333

34-
def check_pfserver_status(service_name, namespace, port, user, password):
35-
print("check_pfserver_status service_name=[%s] namespace=[%s] port=[%s] user=[%s] password=[%s]" % (
36-
service_name, namespace, port, user, password))
37-
host = service_name
38-
39-
client = paddleflow.Client(host, user, password, port)
40-
ret, response = client.login(user, password)
41-
if not ret:
42-
print(response)
43-
err_msg = "client login failed, with host[%s], user[%s], password[%s], port[%s], response[%s]" % (
44-
host, user, password, port, response)
45-
raise Exception(err_msg)
34+
def check_pfserver_status(client):
4635
sum = 1
4736
while sum != 0:
4837
ppl_num = clean_pipelines(client)
@@ -139,8 +128,17 @@ def clean_jobs_with_status(client, status, next_marker=None):
139128
raise Exception(err_msg)
140129
job_list = response
141130
if len(job_list) != 0:
142-
err_msg = "there are [%s] active jobs" % len(job_list)
143-
print(err_msg)
131+
print("there are [%s] active jobs" % len(job_list))
132+
for job in job_list:
133+
try:
134+
print("job is ", job.__dict__)
135+
ret, response = client.stop_job(job.job_id)
136+
print("stop_job %s got %s, err is %s " % (job.job_id, ret, response) )
137+
if ret is True:
138+
print("stop job %s success" % job.job_id)
139+
140+
except Exception as e:
141+
print(e)
144142
else:
145143
err_msg = "no [%s] job, quit clean_jobs_with_status check" % status
146144
print(err_msg)
@@ -154,9 +152,20 @@ def clean_jobs_with_status(client, status, next_marker=None):
154152
port = int(os.getenv("port"))
155153
user = os.getenv("user")
156154
password = os.getenv("password")
155+
client = None
156+
try:
157+
print("check_pfserver_status service_name=[%s] namespace=[%s] port=[%s] user=[%s] password=[%s]" % (
158+
service_name, namespace, port, user, password))
159+
client = paddleflow.Client(service_name, user, password, port)
160+
ret, response = client.login(user, password)
161+
except Exception as e:
162+
print("client login failed, with service_name[%s], user[%s], password[%s], port[%s], exception %s" % (service_name, user, password, port, e))
163+
if e.__str__().__contains__("Connection Error"):
164+
exit(0)
165+
exit(1)
157166

158167
try:
159-
check_pfserver_status(service_name, namespace, port, user, password)
168+
check_pfserver_status(client)
160169
except Exception as e:
161170
print(e)
162171

0 commit comments

Comments
 (0)