spotframework/backup.py

from  spotframework.net.user import NetworkUser
from  spotframework.net.network import Network
import spotframework.io.csv as csvwrite

import sys
import datetime
import os
import logging

if __name__ == '__main__':

    logger = logging.getLogger('spotframework')

    log_format = '%(asctime)s %(levelname)s %(name)s:%(funcName)s - %(message)s'

    file_handler = logging.FileHandler(".spot/backup.log")
    formatter = logging.Formatter(log_format)
    file_handler.setFormatter(formatter)

    logger.addHandler(file_handler)

    stream_log_format = '%(levelname)s %(name)s:%(funcName)s - %(message)s'
    stream_formatter = logging.Formatter(stream_log_format)

    stream_handler = logging.StreamHandler()
    stream_handler.setFormatter(stream_formatter)

    logger.addHandler(stream_handler)

    # try:

    network = Network(NetworkUser(os.environ['SPOT_CLIENT'],
                                  os.environ['SPOT_SECRET'],
                                  os.environ['SPOT_REFRESH']))
    network.user.refresh_access_token()
    playlists = network.get_user_playlists()

    for playlist in playlists:
        playlist.tracks = network.get_playlist_tracks(playlist.uri)

    path = sys.argv[1]

    datepath = str(datetime.datetime.now()).split(' ')[0].replace('-', '/')

    totalpath = os.path.join(path, datepath)
    pathdir = os.path.dirname(totalpath)
    if not os.path.exists(totalpath):
        os.makedirs(totalpath)

    for play in playlists:
        csvwrite.export_playlist(play, totalpath)

    # except Exception as e:
    #     logger.exception(f'exception occured')