From d1be7ac1d204e0b4fd139558c0dfa90363ce5f27 Mon Sep 17 00:00:00 2001
From: Jan De Dobbeleer <jan.de.dobbeleer@gmail.com>
Date: Fri, 15 Jan 2021 19:41:09 +0100
Subject: [PATCH] feat: add known Python extensions/folder

---
 src/segment_python.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/segment_python.go b/src/segment_python.go
index 0c189a72..210c10db 100644
--- a/src/segment_python.go
+++ b/src/segment_python.go
@@ -29,7 +29,7 @@ func (p *python) init(props *properties, env environmentInfo) {
 		props:        props,
 		commands:     []string{"python", "python3"},
 		versionParam: "--version",
-		extensions:   []string{"*.py", "*.ipynb"},
+		extensions:   []string{"*.py", "*.ipynb", "pyproject.toml", "venv.bak", "venv", ".venv"},
 		versionRegex: `Python (?P<version>[0-9]+.[0-9]+.[0-9]+)`,
 		loadContext:  p.loadContext,
 		inContext:    p.inContext,