vscode-python/pythonFiles/tests/ipython/scripts.py at master · nam20485/vscode-python · GitHub

History

77 lines (68 loc) · 2.97 KB

Raw

# Licensed under the MIT License.

import re

import os

import json

import sys

import imp

def check_for_ipython():

if int(sys.version[0]) >= 3:

try:

from IPython import get_ipython

return not get_ipython() == None

except ImportError:

pass

return False

def execute_script(file, replace_dict = dict([])):

from IPython import get_ipython

regex = re.compile('|'.join(replace_dict.keys())) if len(replace_dict.keys()) > 0 else None

# Open the file. Read all lines into a string

contents = ''

with open(file, 'r') as fp:

for line in fp:

# Replace the key value pairs

contents += line if regex == None else regex.sub(lambda m: replace_dict[m.group()], line)

# Execute this script as a cell

result = get_ipython().run_cell(contents)

return result.success

def get_variables(capsys):

path = os.path.dirname(os.path.abspath(__file__))

file = os.path.abspath(os.path.join(path, '../../datascience/getJupyterVariableList.py'))

if execute_script(file):

read_out = capsys.readouterr()

return json.loads(read_out.out)

else:

raise Exception('Getting variables failed.')

def find_variable_json(varList, varName):

for sub in varList:

if sub['name'] == varName:

return sub

def get_variable_value(variables, name, capsys):

varJson = find_variable_json(variables, name)

path = os.path.dirname(os.path.abspath(__file__))

file = os.path.abspath(os.path.join(path, '../../datascience/getJupyterVariableValue.py'))

keys = dict([('_VSCode_JupyterTestValue', json.dumps(varJson))])

if execute_script(file, keys):

read_out = capsys.readouterr()

return json.loads(read_out.out)['value']

else:

raise Exception('Getting variable value failed.')

def get_data_frame_info(variables, name, capsys):

varJson = find_variable_json(variables, name)

path = os.path.dirname(os.path.abspath(__file__))

file = os.path.abspath(os.path.join(path, '../../datascience/getJupyterVariableDataFrameInfo.py'))

keys = dict([('_VSCode_JupyterTestValue', json.dumps(varJson))])

if execute_script(file, keys):

read_out = capsys.readouterr()

return json.loads(read_out.out)

else:

raise Exception('Get dataframe info failed.')

def get_data_frame_rows(varJson, start, end, capsys):

path = os.path.dirname(os.path.abspath(__file__))

file = os.path.abspath(os.path.join(path, '../../datascience/getJupyterVariableDataFrameRows.py'))

keys = dict([('_VSCode_JupyterTestValue', json.dumps(varJson)), ('_VSCode_JupyterStartRow', str(start)), ('_VSCode_JupyterEndRow', str(end))])

if execute_script(file, keys):

read_out = capsys.readouterr()

return json.loads(read_out.out)

else:

raise Exception('Getting dataframe rows failed.')

vscode-python/pythonFiles/tests/ipython/scripts.py at master · nam20485/vscode-python · GitHub

Expand file tree

Breadcrumbs

scripts.py

Latest commit

History

Breadcrumbs

scripts.py

File metadata and controls