def __init__(self, opt):
self.personas_idx_stack_path = os.path.join(opt['extract_personas_path'],
'./personas_idx_stack.pkl')
self.personas_path = '{}/personas-{}'.format(
opt['extract_personas_path'], opt['persona_type'] + 'Original'
)
if not os.path.exists(self.personas_path):
opt['personas_path'] = self.personas_path
main_extract(opt)
self.personas_name_list = []
# list of personas completed from a previous task
self.completed_personas = []
# mark which ones are done
self.done_personas = []
# list of recently popped personas
self.recently_popped = []
for f_name in os.listdir(self.personas_path):
if f_name.endswith('.pkl'):
self.personas_name_list.append(f_name)
if os.path.exists(self.personas_idx_stack_path):
with open(self.personas_idx_stack_path, 'rb') as handle:
self.idx_stack = pickle.load(handle)
else:
self.idx_stack = []
self.add_idx_stack()
self.save_idx_stack()
pass
def __init__(self, opt):
self.personas_idx_stack_path = os.path.join(
os.getcwd(), './personas_idx_stack.pkl'
)
self.personas_path = '{}/data/personas-{}'.format(
os.getcwd(),
opt['persona_type'] + 'Revised' if opt['revised'] else 'Original',
)
if not os.path.exists(self.personas_path):
opt['personas_path'] = self.personas_path
main_extract(opt)
self.personas_name_list = []
for f_name in os.listdir(self.personas_path):
if f_name.endswith('.pkl'):
self.personas_name_list.append(f_name)
if os.path.exists(self.personas_idx_stack_path):
with open(self.personas_idx_stack_path, 'rb') as handle:
self.idx_stack = pickle.load(handle)
else:
self.idx_stack = []
self.add_idx_stack()
self.save_idx_stack()
pass