| 77 | |
| 78 | |
| 79 | def get_data_frame_info(variables, name, capsys): |
| 80 | varJson = find_variable_json(variables, name) |
| 81 | path = os.path.dirname(os.path.abspath(__file__)) |
| 82 | syspath = os.path.abspath( |
| 83 | os.path.join(path, "../../vscode_datascience_helpers/dataframes") |
| 84 | ) |
| 85 | syscode = 'import sys\nsys.path.append("{0}")'.format(syspath.replace("\\", "\\\\")) |
| 86 | importcode = "import vscodeGetDataFrameInfo\nprint(vscodeGetDataFrameInfo._VSCODE_getDataFrameInfo({0}))".format( |
| 87 | name |
| 88 | ) |
| 89 | result = execute_code(syscode) |
| 90 | if not result.success: |
| 91 | result.raise_error() |
| 92 | result = execute_code(importcode) |
| 93 | if result.success: |
| 94 | read_out = capsys.readouterr() |
| 95 | info = json.loads(read_out.out[0:-1]) |
| 96 | varJson.update(info) |
| 97 | return varJson |
| 98 | else: |
| 99 | result.raise_error() |
| 100 | |
| 101 | |
| 102 | def get_data_frame_rows(varJson, start, end, capsys): |