Simplify ps1/ps2 handling

sebastinas · sebastinas · commit 0acf2688fafb · 2021-05-16T15:59:41.000+02:00
diff --git a/bpython/repl.py b/bpython/repl.py
@@ -459,18 +459,12 @@ def __init__(self, interp, config):
             )
 
     @property
-    def ps1(self):
-        try:
-            return sys.ps1
-        except AttributeError:
-            return ">>> "
+    def ps1(self) -> str:
+        return getattr(sys, "ps1", ">>> ")
 
     @property
-    def ps2(self):
-        try:
-            return sys.ps2
-        except AttributeError:
-            return "... "
+    def ps2(self) -> str:
+        return getattr(sys, "ps2", "... ")
 
     def startup(self):
         """
