27 lines
842 B
Python
27 lines
842 B
Python
|
from .base import BasePrompt
|
||
|
|
|
||
|
|
from rich.table import Table, Column
|
||
|
|
from rich import print
|
||
|
|
|
||
|
|
from sqlalchemy import func
|
||
|
|
from groove import db
|
||
|
|
from groove.playlist import Playlist
|
||
|
|
|
||
|
|
|
||
|
|
class browse(BasePrompt):
|
||
|
|
"""Browse the playlists."""
|
||
|
|
|
||
|
|
def process(self, cmd, *parts):
|
||
|
|
count = self.parent.manager.session.query(func.count(db.playlist.c.id)).scalar()
|
||
|
|
print(f"Displaying {count} playlists:")
|
||
|
|
query = self.parent.manager.session.query(db.playlist)
|
||
|
|
table = Table(
|
||
|
|
*[Column(k.name.title()) for k in db.playlist.columns]
|
||
|
|
)
|
||
|
|
for row in db.windowed_query(query, db.playlist.c.id, 1000):
|
||
|
|
columns = tuple(Playlist.from_row(row, self.manager.session).record)[0:-1]
|
||
|
|
table.add_row(*[str(col) for col in columns])
|
||
|
|
print()
|
||
|
|
print(table)
|
||
|
|
print()
|