changed cli a bit

Signed-off-by: hikarikumo <[email protected]>
hikarikumo · Dec 17, 2023 · 654ca34 · 654ca34
1 parent 51fa5d6
commit 654ca34
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 8 deletions.
diff --git a/brain_games/cli.py b/brain_games/cli.py
@@ -19,3 +19,13 @@ def welcome_user():
     user_name = prompt.string('May I know your name? ')
     print('Hello, ', user_name, '!', sep='')
     return user_name
+
+
+def main():
+    """Script to say hi to user."""
+    greet_first()
+    welcome_user()
+
+
+if __name__ == '__main__':
+    main()
diff --git a/brain_games/scripts/brain_games.py b/brain_games/scripts/brain_games.py
@@ -7,14 +7,7 @@
 current_script_dir = os.path.dirname(__file__)
 brain_games_path = os.path.join(current_script_dir, '..')
 sys.path.append(brain_games_path)
-from cli import greet_first
-from cli import welcome_user
-
-
-def main():
-    """Script to say hi to user."""
-    greet_first()
-    welcome_user()
+from cli import main
 
 
 if __name__ == '__main__':