cleanup, more tests

This commit is contained in:
evilchili
2022-12-02 00:21:19 -08:00
parent a0de0eef48
commit ed16ebdd0e
7 changed files with 103 additions and 57 deletions
+1 -1
View File
@@ -4,4 +4,4 @@ from .help import help
from .browse import browse
from .stats import stats
from .playlist import _playlist
from .create import create
from .load import load
+5 -7
View File
@@ -1,4 +1,3 @@
from rich import print
from slugify import slugify
from groove.db.manager import database_manager
@@ -36,7 +35,7 @@ class CommandPrompt(BasePrompt):
def default_completer(self, document, complete_event):
def _formatter(row):
self._playlist = Playlist.from_row(row, self.manager)
self._playlist = Playlist.from_row(row, self.manager.session)
return self.playlist.record.name
return self.manager.fuzzy_table_completer(
db.playlist,
@@ -48,18 +47,17 @@ class CommandPrompt(BasePrompt):
name = cmd + ' ' + ' '.join(parts)
if cmd in self.commands:
self.commands[cmd].start(name)
elif not parts:
print(f"Command not understood: {cmd}")
else:
slug = slugify(name)
self._playlist = Playlist(
slug=slug,
name=name,
session=self.manager.session,
create_if_not_exists=False
create_ok=True
)
self.commands['_playlist'].start()
self._playlist = None
res = self.commands['_playlist'].start()
if res is False:
return res
return True
@@ -5,12 +5,12 @@ from slugify import slugify
from groove.playlist import Playlist
class create(BasePrompt):
class load(BasePrompt):
"""Create a new playlist."""
@property
def usage(self):
return "create PLAYLIST_NAME"
return "load PLAYLIST_NAME"
def process(self, cmd, *parts):
name = ' '.join(parts)
@@ -22,6 +22,7 @@ class create(BasePrompt):
slug=slug,
name=name,
session=self.manager.session,
create_if_not_exists=True
create_ok=True
)
print(self.parent.playlist.summary)
return self.parent.commands['_playlist'].start()