From 6f7247382b6645c94011d4a3433da25e1e6bb4b3 Mon Sep 17 00:00:00 2001
From: Martinez <overwatch@covalent.cz>
Date: Thu, 11 Jan 2018 19:22:46 +0100
Subject: [PATCH] fix python color prompt in bpython

---
 lib/python-startup.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/lib/python-startup.py b/lib/python-startup.py
index 08afad7..a2d5cd9 100755
--- a/lib/python-startup.py
+++ b/lib/python-startup.py
@@ -9,5 +9,7 @@ try:
 except:
 	pass
 
-sys.ps1 = '\001\x1b[1;32m\002>>> \001\x1b[0m\002'
-sys.ps2 = '\001\x1b[0;32m\002... \001\x1b[0m\002'
+# bpython defines __name__ == "__console__" and gets broken by ANSI colors, so...
+if __name__ == "__main__":
+	sys.ps1 = '\001\x1b[1;32m\002>>> \001\x1b[0m\002'
+	sys.ps2 = '\001\x1b[0;32m\002... \001\x1b[0m\002'