jiayong commited on
Commit
b2f0a65
1 Parent(s): 003a395

Update gen_client.py

Browse files

update get_ranking_location

Files changed (1) hide show
  1. gen_client.py +8 -2
gen_client.py CHANGED
@@ -632,7 +632,8 @@ class HumanGenService:
632
  if user_id is None or user_id == '':
633
  user_id = 'test_version_phone'
634
  process_status = ''
635
-
 
636
  if len(self.all_requests) > 0:
637
  for i in range(min(num_instance_dashone, len(self.all_requests))):
638
  req = self.all_requests[i]
@@ -646,7 +647,8 @@ class HumanGenService:
646
  for uid in self.all_user_requests:
647
  if req in self.all_user_requests[uid]:
648
  uuid = uid
649
- break
 
650
  data = '{"header":{"request_id":"","service_id":"","task_id":""},"payload":{"input": {"ref_image_path": "", "ref_video_path": "", "ref_video_name": "", "input_prompt": "", "prompt_template": "", "scale_depth": 0.7, "scale_pose": 0.5},"parameters":{}}}'
651
  data = json.loads(data) # string to dict
652
  data['header']['service_id'] = DASHONE_SERVICE_ID
@@ -659,16 +661,20 @@ class HumanGenService:
659
  ret_status, ret_json = query_video_generation(request_id=req, data=data)
660
  # print(f'ret_json = {ret_json}')
661
  if ret_status == "SUCCESS" or ret_status == "FAILED":
 
662
  if req in self.all_requests:
663
  self.all_requests.remove(req) # delete request_id
664
  if req in self.all_requests_time:
665
  del self.all_requests_time[req]
666
  else:
 
667
  break
668
  else:
 
669
  break
670
  else:
671
  print(f'size of all_requests is empty.')
 
672
 
673
  if user_id not in self.all_user_requests:
674
  return f'You have not request a video generation task.', ''
 
632
  if user_id is None or user_id == '':
633
  user_id = 'test_version_phone'
634
  process_status = ''
635
+
636
+ print(f'[get_ranking_location] ------ clean timeout and process over request start ------ ')
637
  if len(self.all_requests) > 0:
638
  for i in range(min(num_instance_dashone, len(self.all_requests))):
639
  req = self.all_requests[i]
 
647
  for uid in self.all_user_requests:
648
  if req in self.all_user_requests[uid]:
649
  uuid = uid
650
+ break
651
+ print(f'[get_ranking_location] find timeout request: {req}, uuid: {uuid}')
652
  data = '{"header":{"request_id":"","service_id":"","task_id":""},"payload":{"input": {"ref_image_path": "", "ref_video_path": "", "ref_video_name": "", "input_prompt": "", "prompt_template": "", "scale_depth": 0.7, "scale_pose": 0.5},"parameters":{}}}'
653
  data = json.loads(data) # string to dict
654
  data['header']['service_id'] = DASHONE_SERVICE_ID
 
661
  ret_status, ret_json = query_video_generation(request_id=req, data=data)
662
  # print(f'ret_json = {ret_json}')
663
  if ret_status == "SUCCESS" or ret_status == "FAILED":
664
+ print(f'[get_ranking_location] query timeout request process over: {req}, uuid: {uuid}')
665
  if req in self.all_requests:
666
  self.all_requests.remove(req) # delete request_id
667
  if req in self.all_requests_time:
668
  del self.all_requests_time[req]
669
  else:
670
+ print(f'[get_ranking_location] query timeout request process running: {req}, uuid: {uuid}')
671
  break
672
  else:
673
+ print(f'[get_ranking_location] no timeout request.')
674
  break
675
  else:
676
  print(f'size of all_requests is empty.')
677
+ print(f'[get_ranking_location] ------ clean timeout and process over request end ------ ')
678
 
679
  if user_id not in self.all_user_requests:
680
  return f'You have not request a video generation task.', ''