roughed in run(with_message)

This commit is contained in:
Paul Gauthier 2023-06-22 18:55:23 -07:00
parent 208b3c2022
commit cf3637dd5c

View file

@ -93,6 +93,7 @@ class Coder:
self.abs_fnames = set() self.abs_fnames = set()
self.cur_messages = [] self.cur_messages = []
self.done_messages = [] self.done_messages = []
self.num_control_c = 0
self.io = io self.io = io
@ -263,18 +264,20 @@ class Coder:
return files_messages return files_messages
def run(self): def run(self, with_message=None):
self.done_messages = []
self.cur_messages = []
self.num_control_c = 0
while True: while True:
try: try:
new_user_message = self.run_loop() if with_message:
new_user_message = with_message
else:
new_user_message = self.run_loop()
while new_user_message: while new_user_message:
new_user_message = self.send_new_user_message(new_user_message) new_user_message = self.send_new_user_message(new_user_message)
if with_message:
return
except KeyboardInterrupt: except KeyboardInterrupt:
self.num_control_c += 1 self.num_control_c += 1
if self.num_control_c >= 2: if self.num_control_c >= 2: