mudhorn/oh-my-posh
1
0
Fork 0
mirror of https://github.com/JanDeDobbeleer/oh-my-posh.git synced 2025-02-02 05:41:10 -08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: show python segment when Jupyter Notebook files are present

Browse source
This commit is contained in:
Aksel Kvitberg 2020-10-08 11:27:45 +02:00 committed by Jan De Dobbeleer
parent 59ed9c239f
commit 0b0fd4cb1f
3 changed files with 46 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
docs/docs/segment-python.md
View file

@ -6,7 +6,7 @@ sidebar_label: Python
## What ## What
Display the currently active python version and virtualenv when a folder contains `.py` files. Display the currently active python version and virtualenv when a folder contains `.py` files or `.ipynb` files.
Supports conda, virtualenv and pyenv. Supports conda, virtualenv and pyenv.
## Sample Configuration ## Sample Configuration

2
segment_python.go
View file

@ -30,7 +30,7 @@ func (p *python) init(props *properties, env environmentInfo) {
} }
func (p *python) enabled() bool { func (p *python) enabled() bool {
if !p.env.hasFiles("*.py") { if !p.env.hasFiles("*.py") && !p.env.hasFiles("*.ipynb") {
return false return false
} }
pythonVersions := []string{ pythonVersions := []string{

48
segment_python_test.go
View file

@ -15,7 +15,8 @@ type pythonArgs struct {
pathSeparator string pathSeparator string
pythonVersion string pythonVersion string
python3Version string python3Version string
hasFiles bool hasPyFiles bool
hasNotebookFiles bool
} }
func newPythonArgs() *pythonArgs { func newPythonArgs() *pythonArgs {
@ -27,13 +28,15 @@ func newPythonArgs() *pythonArgs {
pathSeparator: "/", pathSeparator: "/",
pythonVersion: "", pythonVersion: "",
python3Version: "", python3Version: "",
hasFiles: true, hasPyFiles: true,
hasNotebookFiles: true,
} }
} }
func bootStrapPythonTest(args *pythonArgs) *python { func bootStrapPythonTest(args *pythonArgs) *python {
env := new(MockedEnvironment) env := new(MockedEnvironment)
env.On("hasFiles", "*.py").Return(args.hasFiles) env.On("hasFiles", "*.py").Return(args.hasPyFiles)
env.On("hasFiles", "*.ipynb").Return(args.hasNotebookFiles)
env.On("runCommand", "python", []string{"--version"}).Return(args.pythonVersion) env.On("runCommand", "python", []string{"--version"}).Return(args.pythonVersion)
env.On("runCommand", "python3", []string{"--version"}).Return(args.python3Version) env.On("runCommand", "python3", []string{"--version"}).Return(args.python3Version)
env.On("getenv", "VIRTUAL_ENV").Return(args.virtualEnvName) env.On("getenv", "VIRTUAL_ENV").Return(args.virtualEnvName)
@ -49,11 +52,48 @@ func bootStrapPythonTest(args *pythonArgs) *python {
func TestPythonWriterDisabledNoPythonFiles(t *testing.T) { func TestPythonWriterDisabledNoPythonFiles(t *testing.T) {
args := newPythonArgs() args := newPythonArgs()
args.hasFiles = false args.hasPyFiles = false
args.hasNotebookFiles = false
args.python3Version = "3.4.5"
python := bootStrapPythonTest(args) python := bootStrapPythonTest(args)
assert.False(t, python.enabled(), "there are no Python files in the current folder") assert.False(t, python.enabled(), "there are no Python files in the current folder")
} }
func TestPythonWriterDisabledHasPythonFiles(t *testing.T) {
args := newPythonArgs()
args.hasPyFiles = true
args.hasNotebookFiles = false
args.python3Version = "3.4.5"
python := bootStrapPythonTest(args)
assert.True(t, python.enabled(), "there should be a Python file in the current folder")
}
func TestPythonWriterDisabledHasJupyterNotebookFiles(t *testing.T) {
args := newPythonArgs()
args.hasPyFiles = false
args.hasNotebookFiles = true
args.python3Version = "3.4.5"
python := bootStrapPythonTest(args)
assert.True(t, python.enabled(), "there should be a Jupyter Notebook file in the current folder")
}
func TestPythonWriterDisabledHasPyAndJupyterNotebookFiles(t *testing.T) {
args := newPythonArgs()
args.hasPyFiles = true
args.hasNotebookFiles = true
args.python3Version = "3.4.5"
python := bootStrapPythonTest(args)
assert.True(t, python.enabled(), "there should be a Jupyter Notebook file in the current folder")
}
func TestPythonWriterDisabledHasPyAndJupyterNotebookFilesButNoVersion(t *testing.T) {
args := newPythonArgs()
args.hasPyFiles = true
args.hasNotebookFiles = true
python := bootStrapPythonTest(args)
assert.False(t, python.enabled(), "there should be a Jupyter Notebook file in the current folder")
}
func TestPythonWriterDisabledNoPythonInstalled(t *testing.T) { func TestPythonWriterDisabledNoPythonInstalled(t *testing.T) {
args := newPythonArgs() args := newPythonArgs()
python := bootStrapPythonTest(args) python := bootStrapPythonTest(args)