31
31
import os
32
32
33
33
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 ):
46
35
sum = 1
47
36
while sum != 0 :
48
37
ppl_num = clean_pipelines (client )
@@ -139,8 +128,17 @@ def clean_jobs_with_status(client, status, next_marker=None):
139
128
raise Exception (err_msg )
140
129
job_list = response
141
130
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 )
144
142
else :
145
143
err_msg = "no [%s] job, quit clean_jobs_with_status check" % status
146
144
print (err_msg )
@@ -154,9 +152,20 @@ def clean_jobs_with_status(client, status, next_marker=None):
154
152
port = int (os .getenv ("port" ))
155
153
user = os .getenv ("user" )
156
154
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 )
157
166
158
167
try :
159
- check_pfserver_status (service_name , namespace , port , user , password )
168
+ check_pfserver_status (client )
160
169
except Exception as e :
161
170
print (e )
162
171
0 commit comments