aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rwxr-xr-xponysay.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/ponysay.py b/ponysay.py
index 90de92d..6e8b82c 100755
--- a/ponysay.py
+++ b/ponysay.py
@@ -2130,8 +2130,8 @@ linuxvt = ('TERM' in os.environ) and (os.environ['TERM'] == 'linux')
'''
Whether the script is executed as ponythink
'''
-isthink = (len(__file__) >= 5) and (__file__[-5:] == 'think')
-isthink = ((len(__file__) >= 8) and (__file__[-8:] == 'think.py')) or isthink
+isthink = (len(__file__) >= len('think')) and (__file__.endswith('think'))
+isthink = ((len(__file__) >= len('think.py')) and (__file__.endswith('think.py'))) or isthink
'''