Refactor to not use curses.wrapper
This commit is contained in:
parent
202280f760
commit
082cdc3222
|
@ -2,4 +2,4 @@
|
||||||
comp
|
comp
|
||||||
====
|
====
|
||||||
|
|
||||||
console/curses online media player
|
Curses Online Media Player
|
||||||
|
|
160
comp.py
160
comp.py
|
@ -22,8 +22,7 @@ def mpv_wrapper(media, video=True):
|
||||||
del player
|
del player
|
||||||
|
|
||||||
|
|
||||||
def interface(stdscr):
|
def reattr(stdscr, y):
|
||||||
def reattr(y):
|
|
||||||
track = DATA[start + y - 1]
|
track = DATA[start + y - 1]
|
||||||
invert = 8 if track['highlight'] else 0
|
invert = 8 if track['highlight'] else 0
|
||||||
if track['error']:
|
if track['error']:
|
||||||
|
@ -37,7 +36,8 @@ def interface(stdscr):
|
||||||
else:
|
else:
|
||||||
stdscr.chgat(y, 0, curses.color_pair(0) | curses.A_NORMAL)
|
stdscr.chgat(y, 0, curses.color_pair(0) | curses.A_NORMAL)
|
||||||
|
|
||||||
def reprint():
|
|
||||||
|
def reprint(stdscr):
|
||||||
stdscr.clear()
|
stdscr.clear()
|
||||||
stdscr.addstr(0, curses.COLS-12, 'URL')
|
stdscr.addstr(0, curses.COLS-12, 'URL')
|
||||||
stdscr.addstr(0, 0, 'Title')
|
stdscr.addstr(0, 0, 'Title')
|
||||||
|
@ -45,7 +45,7 @@ def interface(stdscr):
|
||||||
for i, d in enumerate(DATA[start : start+curses.LINES-3]):
|
for i, d in enumerate(DATA[start : start+curses.LINES-3]):
|
||||||
stdscr.addstr(i + 1, 0, d['url'].rjust(curses.COLS - 1))
|
stdscr.addstr(i + 1, 0, d['url'].rjust(curses.COLS - 1))
|
||||||
stdscr.addstr(i + 1, 0, d['title'][:curses.COLS-12])
|
stdscr.addstr(i + 1, 0, d['title'][:curses.COLS-12])
|
||||||
reattr(i + 1)
|
reattr(stdscr, i + 1)
|
||||||
stdscr.addstr(
|
stdscr.addstr(
|
||||||
curses.LINES - 2,
|
curses.LINES - 2,
|
||||||
curses.COLS - 16,
|
curses.COLS - 16,
|
||||||
|
@ -54,24 +54,25 @@ def interface(stdscr):
|
||||||
stdscr.chgat(curses.LINES - 2, 0, curses.color_pair(8))
|
stdscr.chgat(curses.LINES - 2, 0, curses.color_pair(8))
|
||||||
stdscr.refresh()
|
stdscr.refresh()
|
||||||
|
|
||||||
def move(y, delta):
|
|
||||||
nonlocal start
|
def move(stdscr, y, delta):
|
||||||
reattr(y)
|
global start
|
||||||
|
reattr(stdscr, y)
|
||||||
if start + y + delta < 1:
|
if start + y + delta < 1:
|
||||||
start = 0
|
start = 0
|
||||||
reprint()
|
reprint(stdscr)
|
||||||
stdscr.move(1, 0)
|
stdscr.move(stdscr, 1, 0)
|
||||||
DATA[0]['highlight'] = True
|
DATA[0]['highlight'] = True
|
||||||
reattr(1)
|
reattr(stdscr, 1)
|
||||||
DATA[0]['highlight'] = False
|
DATA[0]['highlight'] = False
|
||||||
return 1
|
return 1
|
||||||
elif start + y + delta > len(DATA):
|
elif start + y + delta > len(DATA):
|
||||||
start = len(DATA) - curses.LINES + 3
|
start = len(DATA) - curses.LINES + 3
|
||||||
reprint()
|
reprint(stdscr)
|
||||||
y = curses.LINES - 3
|
y = curses.LINES - 3
|
||||||
stdscr.move(y, 0)
|
stdscr.move(stdscr, y, 0)
|
||||||
DATA[-1]['highlight'] = True
|
DATA[-1]['highlight'] = True
|
||||||
reattr(y)
|
reattr(stdscr, y)
|
||||||
DATA[-1]['highlight'] = False
|
DATA[-1]['highlight'] = False
|
||||||
return y
|
return y
|
||||||
|
|
||||||
|
@ -79,79 +80,19 @@ def interface(stdscr):
|
||||||
y = y + delta
|
y = y + delta
|
||||||
elif y + delta < 1:
|
elif y + delta < 1:
|
||||||
start += y + delta - 1
|
start += y + delta - 1
|
||||||
reprint()
|
reprint(stdscr)
|
||||||
y = 1
|
y = 1
|
||||||
else:
|
else:
|
||||||
start += y + delta - curses.LINES + 3
|
start += y + delta - curses.LINES + 3
|
||||||
reprint()
|
reprint(stdscr)
|
||||||
y = curses.LINES - 3
|
y = curses.LINES - 3
|
||||||
stdscr.move(y, 0)
|
stdscr.move(stdscr, y, 0)
|
||||||
DATA[start + y - 1]['highlight'] = True
|
DATA[start + y - 1]['highlight'] = True
|
||||||
reattr(y)
|
reattr(stdscr, y)
|
||||||
DATA[start + y - 1]['highlight'] = False
|
DATA[start + y - 1]['highlight'] = False
|
||||||
stdscr.refresh()
|
stdscr.refresh()
|
||||||
return y
|
return y
|
||||||
|
|
||||||
# Init color pairs
|
|
||||||
curses.use_default_colors()
|
|
||||||
curses.init_pair(1, 1, -1)
|
|
||||||
curses.init_pair(2, 2, -1)
|
|
||||||
curses.init_pair(3, 3, -1)
|
|
||||||
curses.init_pair(4, 4, -1)
|
|
||||||
curses.init_pair(5, 5, -1)
|
|
||||||
curses.init_pair(6, 6, -1)
|
|
||||||
curses.init_pair(7, 7, -1)
|
|
||||||
curses.init_pair(8, -1, 7)
|
|
||||||
curses.init_pair(9, -1, 1)
|
|
||||||
curses.init_pair(10, -1, 2)
|
|
||||||
curses.init_pair(11, -1, 3)
|
|
||||||
curses.init_pair(12, -1, 4)
|
|
||||||
curses.init_pair(13, -1, 5)
|
|
||||||
curses.init_pair(14, -1, 6)
|
|
||||||
curses.curs_set(False)
|
|
||||||
|
|
||||||
# Print initial content
|
|
||||||
start = 0
|
|
||||||
reprint()
|
|
||||||
y = 1
|
|
||||||
DATA[0]['highlight'] = True
|
|
||||||
stdscr.move(1, 0)
|
|
||||||
reattr(1)
|
|
||||||
DATA[0]['highlight'] = False
|
|
||||||
|
|
||||||
while True:
|
|
||||||
c = stdscr.getch()
|
|
||||||
if c == curses.KEY_RESIZE:
|
|
||||||
curses.update_lines_cols()
|
|
||||||
reprint()
|
|
||||||
y = 1
|
|
||||||
reattr(y)
|
|
||||||
elif c in (ord('j'), curses.KEY_DOWN):
|
|
||||||
y = move(y, 1)
|
|
||||||
elif c in (ord('k'), curses.KEY_UP):
|
|
||||||
y = move(y, -1)
|
|
||||||
elif c == curses.KEY_PPAGE:
|
|
||||||
y = move(y, -curses.LINES)
|
|
||||||
elif c == curses.KEY_NPAGE:
|
|
||||||
y = move(y, curses.LINES)
|
|
||||||
elif c == curses.KEY_HOME:
|
|
||||||
y = move(y, -len(DATA))
|
|
||||||
elif c == curses.KEY_END:
|
|
||||||
y = move(y, len(DATA))
|
|
||||||
elif c == ord(' '):
|
|
||||||
DATA[start + y - 1]['selected'] = not DATA[start + y - 1]['selected']
|
|
||||||
y = move(y, 1)
|
|
||||||
elif c == ord('x'): # temporally behavior
|
|
||||||
mpv_wrapper('https://youtu.be/' + DATA[start + y - 1]['url'], video)
|
|
||||||
elif c == ord('v'):
|
|
||||||
player.quit_watch_later()
|
|
||||||
elif c in (ord('q'), 27): # 27 is Escape key
|
|
||||||
stdscr.addstr(curses.LINES - 1, 0, 'Quit comp? [y/N]')
|
|
||||||
if stdscr.getch() == ord('y'):
|
|
||||||
break
|
|
||||||
reprint()
|
|
||||||
stdscr.refresh()
|
|
||||||
|
|
||||||
|
|
||||||
parser = ArgumentParser(description="console/curses online media player")
|
parser = ArgumentParser(description="console/curses online media player")
|
||||||
parser.add_argument('-j', '--json-playlist', required=False,
|
parser.add_argument('-j', '--json-playlist', required=False,
|
||||||
|
@ -172,4 +113,67 @@ for i in DATA:
|
||||||
i['playing'] = False
|
i['playing'] = False
|
||||||
i['selected'] = False
|
i['selected'] = False
|
||||||
i['highlight'] = False
|
i['highlight'] = False
|
||||||
curses.wrapper(interface)
|
|
||||||
|
|
||||||
|
stdscr = curses.initscr()
|
||||||
|
curses.noecho()
|
||||||
|
curses.cbreak()
|
||||||
|
stdscr.keypad(True)
|
||||||
|
curses.curs_set(False)
|
||||||
|
curses.start_color()
|
||||||
|
curses.use_default_colors()
|
||||||
|
curses.init_pair(1, 1, -1)
|
||||||
|
curses.init_pair(2, 2, -1)
|
||||||
|
curses.init_pair(3, 3, -1)
|
||||||
|
curses.init_pair(4, 4, -1)
|
||||||
|
curses.init_pair(5, 5, -1)
|
||||||
|
curses.init_pair(6, 6, -1)
|
||||||
|
curses.init_pair(7, 7, -1)
|
||||||
|
curses.init_pair(8, -1, 7)
|
||||||
|
curses.init_pair(9, -1, 1)
|
||||||
|
curses.init_pair(10, -1, 2)
|
||||||
|
curses.init_pair(11, -1, 3)
|
||||||
|
curses.init_pair(12, -1, 4)
|
||||||
|
curses.init_pair(13, -1, 5)
|
||||||
|
curses.init_pair(14, -1, 6)
|
||||||
|
|
||||||
|
# Print initial content
|
||||||
|
start = 0
|
||||||
|
reprint(stdscr)
|
||||||
|
y = 1
|
||||||
|
DATA[0]['highlight'] = True
|
||||||
|
stdscr.move(stdscr, 1, 0)
|
||||||
|
reattr(stdscr, 1)
|
||||||
|
DATA[0]['highlight'] = False
|
||||||
|
|
||||||
|
c = stdscr.getch()
|
||||||
|
while c != 113: # letter q
|
||||||
|
if c == curses.KEY_RESIZE:
|
||||||
|
curses.update_lines_cols()
|
||||||
|
reprint(stdscr)
|
||||||
|
y = 1
|
||||||
|
reattr(stdscr, y)
|
||||||
|
elif c in (ord('j'), curses.KEY_DOWN):
|
||||||
|
y = move(stdscr, y, 1)
|
||||||
|
elif c in (ord('k'), curses.KEY_UP):
|
||||||
|
y = move(stdscr, y, -1)
|
||||||
|
elif c == curses.KEY_PPAGE:
|
||||||
|
y = move(stdscr, y, -curses.LINES)
|
||||||
|
elif c == curses.KEY_NPAGE:
|
||||||
|
y = move(stdscr, y, curses.LINES)
|
||||||
|
elif c == curses.KEY_HOME:
|
||||||
|
y = move(stdscr, y, -len(DATA))
|
||||||
|
elif c == curses.KEY_END:
|
||||||
|
y = move(stdscr, y, len(DATA))
|
||||||
|
elif c == ord(' '):
|
||||||
|
DATA[start + y - 1]['selected'] = not DATA[start + y - 1]['selected']
|
||||||
|
y = move(stdscr, y, 1)
|
||||||
|
elif c == ord('x'): # temporally behavior
|
||||||
|
mpv_wrapper('https://youtu.be/' + DATA[start + y - 1]['url'], video)
|
||||||
|
stdscr.refresh()
|
||||||
|
c = stdscr.getch()
|
||||||
|
|
||||||
|
curses.nocbreak()
|
||||||
|
stdscr.keypad(False)
|
||||||
|
curses.echo()
|
||||||
|
curses.endwin()
|
||||||
|
|
4
setup.py
4
setup.py
|
@ -8,12 +8,12 @@ with open('README.rst') as f:
|
||||||
setup(name = 'comp',
|
setup(name = 'comp',
|
||||||
version = '0.1.0a1',
|
version = '0.1.0a1',
|
||||||
url = 'https://github.com/McSinyx/comp',
|
url = 'https://github.com/McSinyx/comp',
|
||||||
description = ('console/curses online media player'),
|
description = ('Curses Online Media Player'),
|
||||||
long_description=long_description,
|
long_description=long_description,
|
||||||
author = 'McSinyx',
|
author = 'McSinyx',
|
||||||
author_email = 'vn.mcsinyx@gmail.com',
|
author_email = 'vn.mcsinyx@gmail.com',
|
||||||
py_modules = ['mpv'],
|
py_modules = ['mpv'],
|
||||||
scripts=['comp'],
|
scripts=['comp.py'],
|
||||||
classifiers = ['Development Status :: 1 - Planning',
|
classifiers = ['Development Status :: 1 - Planning',
|
||||||
'Environment :: Console :: Curses',
|
'Environment :: Console :: Curses',
|
||||||
'Intended Audience :: End Users/Desktop',
|
'Intended Audience :: End Users/Desktop',
|
||||||
|
|
Loading…
Reference in a new issue