Browse Source

[O] Clear screen with title

Azalea (on HyDEV-Daisy) 3 years ago
parent
commit
299adb8e4d
2 changed files with 7 additions and 4 deletions
  1. 4 1
      hyfetch/color_util.py
  2. 3 3
      hyfetch/main.py

+ 4 - 1
hyfetch/color_util.py

@@ -49,12 +49,15 @@ def printc(msg: str):
     print(color(msg + '&r'))
 
 
-def clear_screen():
+def clear_screen(title: str = ''):
     """
     Clear screen using ANSI escape codes
     """
     print('\033[2J\033[H', end='')
 
+    if title:
+        printc(title)
+
 
 def redistribute_rgb(r: int, g: int, b: int) -> tuple[int, int, int]:
     """

+ 3 - 3
hyfetch/main.py

@@ -103,8 +103,8 @@ def create_config() -> Config:
 
     :return: Config object (automatically stored)
     """
-    clear_screen()
-    printc('\nWelcome to &b&lhy&f&lfetch&r! Let\'s set up some colors first.\n')
+    title = '\nWelcome to &b&lhy&f&lfetch&r! Let\'s set up some colors first.\n'
+    clear_screen(title)
 
     # Select color system
     try:
@@ -129,7 +129,7 @@ def create_config() -> Config:
         color_system = literal_input('Which &acolor &bsystem &rdo you want to use?',
                                      ['8bit', 'rgb'], 'rgb')
 
-    clear_screen()
+    clear_screen(title)
 
     # Print preset
     print('Available presets:\n')