ZeroCommand commited on
Commit
e631fcc
1 Parent(s): f04482d

fix pipe io bug

Browse files
Files changed (1) hide show
  1. io_utils.py +9 -5
io_utils.py CHANGED
@@ -95,13 +95,17 @@ def save_job_to_pipe(id, job, lock):
95
  def pop_job_from_pipe():
96
  if not os.path.exists(PIPE_PATH):
97
  return
98
- with open(PIPE_PATH, "r+") as f:
99
- jobs = f.readlines()
100
- f.write("\n".join(jobs[1:]))
101
  f.close()
102
- if len(jobs) == 0:
 
 
 
 
103
  return
104
- job_info = jobs[0].split('\n')[0].split("@")
105
  if len(job_info) != 2:
106
  raise ValueError("Invalid job info: ", job_info)
107
  print(f"Running job {job_info}")
 
95
  def pop_job_from_pipe():
96
  if not os.path.exists(PIPE_PATH):
97
  return
98
+ with open(PIPE_PATH, "r") as f:
99
+ job = f.readline().strip()
100
+ remaining = f.readlines()
101
  f.close()
102
+ print(job, remaining, ">>>>")
103
+ with open(PIPE_PATH, "w") as f:
104
+ f.write("\n".join(remaining))
105
+ f.close()
106
+ if len(job) == 0:
107
  return
108
+ job_info = job.split('\n')[0].split("@")
109
  if len(job_info) != 2:
110
  raise ValueError("Invalid job info: ", job_info)
111
  print(f"Running job {job_info}")