mirror of
https://github.com/anatolykopyl/youtube-cdl.git
synced 2026-03-26 12:55:11 +00:00
🎨 Cleaner getch
This commit is contained in:
33
main.py
33
main.py
@@ -11,8 +11,24 @@ from print_logo import print_logo
|
||||
|
||||
if platform.system() == 'Windows':
|
||||
import msvcrt
|
||||
def uni_getch():
|
||||
char = str(msvcrt.getch())
|
||||
if char == "b'y'":
|
||||
return True
|
||||
elif char == "b'n'":
|
||||
return False
|
||||
else:
|
||||
return None
|
||||
else:
|
||||
from getch import getch
|
||||
def uni_getch():
|
||||
char = getch()
|
||||
if char == "y":
|
||||
return True
|
||||
elif char == "n":
|
||||
return False
|
||||
else:
|
||||
return None
|
||||
|
||||
parser = argparse.ArgumentParser()
|
||||
parser.add_argument('-i', '--input', action='store', type=str,
|
||||
@@ -50,22 +66,11 @@ else:
|
||||
curr_channel += 1
|
||||
c.print(f'[dim][{curr_channel}/{len(all_channels)}]:[/dim] {ch["title"]} [cyan]\[y/n]')
|
||||
while True:
|
||||
if platform.system() == 'Windows':
|
||||
key = str(msvcrt.getch())
|
||||
if key == "b'y'":
|
||||
key = uni_getch()
|
||||
if key == True:
|
||||
ch['download'] = True
|
||||
break
|
||||
elif key == "b'n'":
|
||||
ch['download'] = False
|
||||
break
|
||||
else:
|
||||
c.print('Press "y" or "n"', style='yellow')
|
||||
else:
|
||||
key = getch()
|
||||
if key == "y":
|
||||
ch['download'] = True
|
||||
break
|
||||
elif key == "n":
|
||||
elif key == False:
|
||||
ch['download'] = False
|
||||
break
|
||||
else:
|
||||
|
||||
Reference in New Issue
Block a user