fix: msg title error

Rishang · Jun 19, 2022 · 25b5b19 · 25b5b19
1 parent 0ea01a6
commit 25b5b19
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/getlooks/core.py b/getlooks/core.py
@@ -15,7 +15,7 @@
 from bs4 import BeautifulSoup
 
 # local
-from getlooks.utils import show_table, message, logger
+from getlooks.utils import show_table, rprint, logger
 from getlooks.looks_path import DESK_THEME_PATH, STATE_PATH
 
 
@@ -131,9 +131,9 @@ def env_prompt(self):
             os.makedirs(self.state_file.parent)
 
         logger.error("Can't figureout desktop environment")
-        message.title(
-            "Which one is your desktop environment ?\n"
-            "\nChoose you desktop environment id, \nExample: 1 for gnome\n"
+        rprint(
+            "[yellow]Which one is your desktop environment ?\n"
+            "\nChoose you desktop environment id,[reset] \nExample: 1 for gnome\n"
         )
         show_table([self.valid_env], title="Select desktop environment")
 

diff --git a/getlooks/looks.py b/getlooks/looks.py
@@ -16,12 +16,13 @@
     show_table,
     logger,
     console,
+    message
 )
+
 from getlooks.core import (
     ProductInfo,
     DeskEnv,
     scrapGnomeLooks,
-    message,
     product_state_cache,
     path_for,
 )