FazBrowse GitHub Viewer
|
Trending
|
URL:
|
Home
Tools:
[Download Repo ZIP]
[View Raw Code]
[Original HTTPS Page]
vscode-python/pythonFiles/tests/ipython/scripts.py at master · deepstqte/vscode-python · GitHub
deepstqte
/
vscode-python
Public
forked from
microsoft/vscode-python
Notifications
You must be signed in to change notification settings
Fork
0
Star
0
Code
Pull requests
0
Actions
Projects
Security and quality
0
Insights
Additional navigation options
Code
Pull requests
Actions
Projects
Security and quality
Insights
Expand file tree
Breadcrumbs
vscode-python
/
pythonFiles
/
tests
/
ipython
/
scripts.py
Copy path
More file actions
More file actions
Latest commit
History
History
History
120 lines (100 loc) · 3.71 KB
Breadcrumbs
vscode-python
/
pythonFiles
/
tests
/
ipython
/
scripts.py
Copy path
File metadata and controls
120 lines (100 loc) · 3.71 KB
Raw
Copy raw file
Download raw file
Open symbols panel
Edit and raw actions
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
# Copyright (c) Microsoft Corporation. All rights reserved.
# 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
execute_code
(
code
):
# Execute this script as a cell
result
=
get_ipython
().
run_cell
(
code
)
return
result
def
get_variables
(
capsys
):
path
=
os
.
path
.
dirname
(
os
.
path
.
abspath
(
__file__
))
file
=
os
.
path
.
abspath
(
os
.
path
.
join
(
path
,
"./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
,
"./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__
))
syspath
=
os
.
path
.
abspath
(
os
.
path
.
join
(
path
,
"../../vscode_datascience_helpers/dataframes"
)
)
syscode
=
'import sys
\n
sys.path.append("{0}")'
.
format
(
syspath
.
replace
(
"
\\
"
,
"
\\
\\
"
))
importcode
=
"import vscodeGetDataFrameInfo
\n
print(vscodeGetDataFrameInfo._VSCODE_getDataFrameInfo({0}))"
.
format
(
name
)
result
=
execute_code
(
syscode
)
if
not
result
.
success
:
result
.
raise_error
()
result
=
execute_code
(
importcode
)
if
result
.
success
:
read_out
=
capsys
.
readouterr
()
info
=
json
.
loads
(
read_out
.
out
[
0
:
-
1
])
varJson
.
update
(
info
)
return
varJson
else
:
result
.
raise_error
()
def
get_data_frame_rows
(
varJson
,
start
,
end
,
capsys
):
path
=
os
.
path
.
dirname
(
os
.
path
.
abspath
(
__file__
))
syspath
=
os
.
path
.
abspath
(
os
.
path
.
join
(
path
,
"../../vscode_datascience_helpers/dataframes"
)
)
syscode
=
'import sys
\n
sys.path.append("{0}")'
.
format
(
syspath
.
replace
(
"
\\
"
,
"
\\
\\
"
))
importcode
=
"import vscodeGetDataFrameRows
\n
print(vscodeGetDataFrameRows._VSCODE_getDataFrameRows({0}, {1}, {2}))"
.
format
(
varJson
[
"name"
],
start
,
end
)
result
=
execute_code
(
syscode
)
if
not
result
.
success
:
result
.
raise_error
()
result
=
execute_code
(
importcode
)
if
result
.
success
:
read_out
=
capsys
.
readouterr
()
return
json
.
loads
(
read_out
.
out
[
0
:
-
1
])
else
:
result
.
raise_error
()
Back
|
FazBrowse Home
|
New Git URL