Exit bug (#102)

* fix : EOFError handler added

* fix : get_input_standard function updated

* doc : CHANGELOG.md updated

* fix : minor bug in get_input_standard function fixed

* fix : import EXIT_MESSAGE

* fix : tests updated

* fix : tests updated

CHANGELOG.md

@@ -10,6 +10,7 @@ and this project adheres to [Semantic Versioning](http://semver.org/spec/v2.0.0.
 - `clear_screen` function
 ### Changed
 - Restart mode updated
+- Exit bug fixed
 ## [0.7] - 2024-08-27
 ### Added
 - `feature_request.yml` template

nafas/__main__.py

@@ -46,5 +46,5 @@ def main():
 if __name__ == "__main__":
     try:
         main()
-    except KeyboardInterrupt:
+    except (KeyboardInterrupt, EOFError):
         print("\n" + EXIT_MESSAGE)

nafas/functions.py

@@ -4,7 +4,7 @@
 import time
 from nafas.params import NAFAS_DESCRIPTION, NAFAS_NOTICE, STANDARD_MENU, STANDARD_MENU_ORDER, STEP_MAP
 from nafas.params import PROGRAMS, PROGRAM_DESCRIPTION, SOUND_MAP, STEP_TEMPLATE, CYCLE_TEMPLATE
-from nafas.params import SOUND_WARNING_MESSAGE
+from nafas.params import SOUND_WARNING_MESSAGE, EXIT_MESSAGE
 import nava
 import os
 from warnings import warn
@@ -221,6 +221,9 @@ def get_input_standard(input_func=input):
             try:
                 input_data[item] = int(input_func(""))
                 exit_flag = True
+            except KeyboardInterrupt:
+                print("\n" + EXIT_MESSAGE)
+                sys.exit()
             except Exception:
                 print("[Error] Bad Input!")
     return input_data

test/test.py

@@ -50,6 +50,8 @@
 True
 >>> input_data["level"] == 1
 True
+>>> def test_keyboard_interrupt(i):
+...     raise KeyboardInterrupt
 >>> input_data = get_input_standard(lambda x: "1")
 - Please choose a program :
 <BLANKLINE>
@@ -70,6 +72,10 @@
 1- Beginner
 2- Medium
 3- Advanced
+>>> get_input_standard(test_keyboard_interrupt)
+Traceback (most recent call last):
+...
+SystemExit
 >>> input_data["program"] == 1
 True
 >>> input_data["level"] == 1