Exemplo n.º 1
0
 def get_slot_from_sub_goal(self, slot):
     if not self.sub_goal_exists():
         return None
     sub_goal = self.next_sub_goal()
     if sub_goal['name'] in ('Copy_file', 'Move_file', 'Search_file',
                             'Open_file'):
         if slot == 'file_name':
             return sub_goal['file']
         if slot == 'parent_directory' or slot == 'origin':
             if 'parent_directory' in sub_goal:
                 return sub_goal['parent_directory']
             if 'origin' in sub_goal:
                 return FileTreeSimulator.last_dir_in_path(
                     sub_goal['origin'])
         if slot == 'dest' and 'dest' in sub_goal:
             return FileTreeSimulator.last_dir_in_path(sub_goal['dest'])
     elif sub_goal['name'] == 'Rename_file':
         if slot == 'old_name':
             return sub_goal['old_name']
         if slot == 'new_name':
             return sub_goal['new_name']
         if slot == 'parent_directory':
             return sub_goal['parent_directory']
     elif sub_goal['name'] == 'Change_directory':
         if slot == 'directory':
             return FileTreeSimulator.last_dir_in_path(sub_goal['dirs'][0])
     return None
Exemplo n.º 2
0
 def generate_sub_goal_intent(self):
     sub_goal = self.next_sub_goal()
     pF = 0.5
     if sub_goal['name'] == 'Change_directory':
         assert self.state['current_directory'] != sub_goal['dirs'][
             -1], 'sub goal already reached'
         dirs = sub_goal['dirs']
         index = int(random.uniform(0, len(dirs) - 0.01))
         directory = dirs[index].split('/')[-1]
         return self.create_change_dir_desire(directory)
     elif sub_goal['name'] == 'Search_file':
         if random.random() < pF:
             return {
                 'intent': self.u_request,
                 'slot': 'directory',
                 'file_name': sub_goal['file']
             }
         return {'intent': self.u_request, 'slot': 'directory'}
     elif sub_goal['name'] == 'Open_file':
         action = {'intent': self.Open_file_desire}
         if random.random() < pF:
             action['file_name'] = sub_goal['file']
         if random.random() < pF:
             action['parent_directory'] = sub_goal['parent_directory']
         return action
     elif sub_goal['name'] == 'Rename_file':
         action = {'intent': self.Rename_file_desire}
         if random.random() < pF:
             action['old_name'] = sub_goal['old_name']
         if random.random() < pF:
             action['new_name'] = sub_goal['new_name']
         if random.random() < pF:
             action['parent_directory'] = sub_goal['parent_directory']
         return action
     elif sub_goal['name'] == 'Move_file' or sub_goal['name'] == 'Copy_file':
         pO, pD, pF = 0.2, 0.4, 0.8
         action = {
             'intent': (self.Move_file_desire if sub_goal['name']
                        == 'Move_file' else self.Copy_file_desire)
         }
         if random.uniform(0, 1) < pO:
             action['origin'] = FileTreeSimulator.last_dir_in_path(
                 sub_goal['origin'])
         if random.uniform(0, 1) < pD:
             action['dest'] = FileTreeSimulator.last_dir_in_path(
                 sub_goal['dest'])
         if random.uniform(0, 1) < pF:
             action['file_name'] = sub_goal['file']
         return action
     return None
Exemplo n.º 3
0
 def create_tree_sim(self, root_inf=None):
     if root_inf is None:
         root = self.root
         sim = FileTreeSimulator([], name=self.name_by_node[root])
     else:
         root, sim = root_inf
     if root not in self.children:
         return sim
     for n in self.children[root]:
         if n not in self.file_exists:
             continue
         t = 0 if self.file_type[n] == fbrowser.Directory else 1
         _, m = sim.add_file(self.name_by_node[n], t)
         if not t:
             self.create_tree_sim((n, m['tree_sim']))
     return sim
Exemplo n.º 4
0
 def add_known_files_to_graph(self, tree_sim=None):
     """
     adds root directory for now
     :return: root's node and root's name
     """
     if tree_sim is None:
         tree_sim = FileTreeSimulator()
     root = self.add_root_file(tree_sim.name)
     self.add_tree_to_graph(tree_sim, root)
Exemplo n.º 5
0
    def _ask_sub_goal(self, agent_action):
        if not self.sub_goal_exists():
            return None
        asked_action = agent_action['action']
        sub_goal = self.next_sub_goal()
        if asked_action['intent'] == 'Change_directory' and sub_goal['name'] == 'Change_directory' \
                and asked_action['new_directory'] in sub_goal['dirs']:
            return {'intent': self.confirm}
        if asked_action['intent'] in (
                'Move_file',
                'Copy_file') and sub_goal['name'] == asked_action['intent']:
            keys = ['origin', 'dest']
            asked_action['file'] = asked_action['file_name']
            response = {'intent': self.confirm}
            for key in keys:
                if not FileTreeSimulator.equal_paths(sub_goal[key],
                                                     asked_action[key]):
                    return {'intent': self.deny}
            if asked_action['file'] != sub_goal['file']:
                return {'intent': self.deny}
            return response
        if asked_action['intent'] == 'Open_file' and sub_goal[
                'name'] == asked_action['intent']:
            p_dir = FileTreeSimulator.last_dir_in_path(asked_action['path'])
            if p_dir == sub_goal['parent_directory'] and asked_action[
                    'file_name'] == sub_goal['file']:
                return {'intent': self.confirm}
            else:
                return {'intent': self.deny}
        if sub_goal['name'] == 'Rename_file' and asked_action[
                'intent'] == sub_goal['name']:
            p_dir = FileTreeSimulator.last_dir_in_path(asked_action['path'])
            asked_action['parent_directory'] = p_dir
            keys = ['old_name', 'new_name', 'parent_directory']
            response = {'intent': self.confirm}
            for key in keys:
                if asked_action[key] != sub_goal[key]:
                    return {'intent': self.deny}
            return response

        return None
Exemplo n.º 6
0
 def reset(self, first_action):
     tree = FileTreeSimulator.read_existing_dirs(directory=self.directory)
     data = {'current_tree_sim': tree, 'tree_sim': tree}
     self.dqn_agent.eps = 0
     self.dqn_agent.reset(first_action, data)
Exemplo n.º 7
0
    def update_sub_goals(self, agent_action):
        for sub_goal in self.goal['sub_goal']:
            if sub_goal['name'] == 'Change_directory':
                last_dir = sub_goal['dirs'][-1]
                current_dir = self.state['current_directory']
                if current_dir[-1] == '/': current_dir = current_dir[:-1]
                if last_dir[-1] == '/': last_dir = last_dir[:-1]
                if current_dir == last_dir:
                    self.goal['sub_goal'].remove(sub_goal)
                    self.subgoal_reward = len(sub_goal['dirs'])
                elif current_dir in sub_goal['dirs']:
                    self.subgoal_reward = sub_goal['dirs'].index(
                        current_dir) + 1
                    del sub_goal['dirs'][:self.subgoal_reward]
            if sub_goal['name'] == 'Search_file':
                if agent_action[
                        'intent'] == 'inform' and 'paths' in agent_action:
                    self.subgoal_reward = 0
                    paths = agent_action['paths']
                    r = self.state['current_file_tree'].lookup_file_name(
                        sub_goal['file'])
                    self.subgoal_reward = -2
                    if r is not None:
                        f, m = r
                        for path in paths:
                            if FileTreeSimulator.equal_paths(
                                    m['tree_sim'].path(True), path):
                                self.subgoal_reward = 2
                                break

                    self.goal['sub_goal'].remove(sub_goal)
            if sub_goal['name'] == 'Rename_file':
                if agent_action['intent'] == 'Rename_file':
                    p_dir = FileTreeSimulator.last_dir_in_path(
                        agent_action['path'])
                    agent_action['parent_directory'] = p_dir
                    keys = ['old_name', 'new_name', 'parent_directory']
                    self.subgoal_reward = 2
                    for key in keys:
                        if agent_action[key] != sub_goal[key]:
                            self.subgoal_reward = -3

                    self.goal['sub_goal'].remove(sub_goal)
            if sub_goal['name'] == 'Open_file':
                if agent_action['intent'] == sub_goal['name']:
                    p_dir = FileTreeSimulator.last_dir_in_path(
                        agent_action['path'])
                    if p_dir != sub_goal['parent_directory'] or agent_action[
                            'file_name'] != sub_goal['file']:
                        self.subgoal_reward = -3
                    else:
                        self.subgoal_reward = 2
                    self.goal['sub_goal'].remove(sub_goal)
            if sub_goal['name'] == 'Move_file' or sub_goal[
                    'name'] == 'Copy_file':
                if agent_action['intent'] == sub_goal['name']:
                    keys = ['origin', 'dest']
                    agent_action['file'] = agent_action['file_name']
                    self.subgoal_reward = 2
                    for key in keys:
                        if not FileTreeSimulator.equal_paths(
                                sub_goal[key], agent_action[key]):
                            self.subgoal_reward = -3
                            break
                    if agent_action['file'] != sub_goal['file']:
                        self.subgoal_reward = -3
                    self.goal['sub_goal'].remove(sub_goal)
                    break