demixed-dl/cli/cli.py

import sys

from deezer.deezer import Deezer

def main():
  deezer = Deezer()

  def grab_track(song_id):
    track = deezer.get_track(int(song_id))
    artists_string = ', '.join(map(lambda a: a.name, track.artists))
    print("grabbing {} by {}".format(track.title, artists_string))
    track.download(print_progress = True)

  def login(email, password):
    user = deezer.login(email, password)
    print("logged in as %s" % user["name"])

  print('''
        Welcome to...
         _|_._ _ o  _ _|
        (_(/| | ||>(/(_|
        ''')
  if len(sys.argv) > 1:
    login(sys.argv[1], sys.argv[2])
  while True:
    command = input("-> ").split(" ")

    if command[0] == 'login':
      login(command[1], command[2])
    elif command[0] == 'grab':
      if command[1] == 'album':
        album = deezer.get_album(int(command[2]))
        print("grabbing album %s" % album.title)
        for track in album.songs:
          grab_track(track.song_id)
      elif command[1] == 'track':
        grab_track(command[2])
      else:
        grab_track(command[1])
    elif command[0] == 'exit':
      break
    else:
      print("unknown command... :/")