Handle non-existent log file or empty logs in auto playlist

Previously, errors weren't handled.

playx/main.py

@@ -275,9 +275,17 @@ def main():
 
  # first check for auto playlist
  elif args.auto:
- # ap = CountBasedAutoPlaylist('~/.playx/logs/log.cat')
- ap = MarkovBasedAutoPlaylist("~/.playx/logs/log.cat")
- song = ap.generate()
+ try:
+ # ap = CountBasedAutoPlaylist('~/.playx/logs/log.cat')
+ log_path = "~/.playx/logs/log.cat"
+ ap = MarkovBasedAutoPlaylist(log_path)
+ song = ap.generate()
+ except (AssertionError):
+ logger.info("Unable to generate playlist...")
+ logger.info(
+ f"Either log at [{log_path}] is not in existence or is a pure void... :/"
+ )
+ exit(0)
  else:
  song = ""
 

playx/playlist/autoplaylist.py

@@ -23,6 +23,7 @@
 class AbstractAutoPlaylist(ABC):
  def __init__(self, log_path):
  self.log_path = pathlib.Path(log_path).expanduser()
+ assert self.log_path.exists()
  self.data = []
 
  @abstractmethod
@@ -44,6 +45,8 @@ def get_timeseries_data(self, log_path):
  with open(log_path) as f:
  for line in f:
  line = line.strip().lower()
+ if not line:
+ continue
  if "playing" not in line:
  continue
  matches = re.findall(r"\[.*?\]", line)
@@ -71,6 +74,7 @@ def __init__(self, log_path):
 
  def generate(self):
  data = self.get_timeseries_data(self.log_path)
+ assert data
  ts, songs = zip(*data)
  counter = Counter(songs)
  songs_frequent, c = zip(*counter.most_common())
@@ -101,6 +105,7 @@ def __init__(self, log_path):
 
  def generate(self):
  data = self.get_timeseries_data(self.log_path)
+ assert data
  ts, songs = zip(*data)
 
  # build trie and probabilities