Example #1
0
    def search(self, query, max_results=False, fields=None, group_by=None):
        """
        Search the current client for results
        """
        self._wait_for_load()
        frames = MultiResultList()
        for item in self:
            frame = item.search(
                self._language_parser.parse(
                    query,
                    self._get_item(
                        os.path.basename(item.filename)
                    ).keys()
                ),
                max_results=max_results,
                fields=fields,
                group_by=group_by
            )

            results = ResultList(
                name=os.path.basename(item.filename).split('.')[0].split('/')[-1]
            )
            results.dataframe = (frame, self._get_item(os.path.basename(item.filename)))
            if len(results) > 0:
                frames.append(results)

        return frames
Example #2
0
 def test_result_list_from_pandas_dict(self):
     dictionary = {0: {'line': 'hello'}, 1: {'line': 'world'}}
     result_list = ResultList()
     result_list.dataframe = pd.DataFrame.from_dict(dictionary)
     self.assertIsInstance(result_list.dataframe, pd.DataFrame)