SpotifyRandomizer/main.py at master · DanielHaitink/SpotifyRandomizer · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
import randomizer as r
import json
import os
import sys
import re


def found_playlist_by_name(rand, playlist):
    try:
        rand.set_playlist_by_name(playlist)
    except r.NotFound:
        return False
    return True


def found_playlist_by_id(rand, playlist):
    try:
        rand.set_playlist_by_id(str(playlist))
    except r.NotFound:
        return False
    return True


def is_playlist_found(rand, playlist):
    return found_playlist_by_name(rand, playlist) or found_playlist_by_id(rand, playlist)


def authenticate_spotify(username):
    try:
        auth = r.SpotifyAuth(username)
        auth.wait_for_auth()
    except r.FailedAuth:
        print("Authentication failed")
        sys.exit()
    auth.stop_server()
    return auth


def print_user_playlists(user_playlists):
    print("User playlists:")
    for i, item in enumerate(user_playlists):
        print("{}: {}".format(i + 1, item))


def get_playlists_by_input(randomizer):
    user_playlists = [x['name'] for x in randomizer.get_all_playlists()]
    print_user_playlists(user_playlists)

    while True:
        picks = input("Enter the # of the playlist(s), separated by comma, or 'cancel' to cancel: ")

        if picks.lower() == "cancel":
            print("Goodbye!")
            return

        playlists = []

        for pick in picks.split(","):
            pick = pick.strip()

            if not re.match("^[0-9]+$", pick):
                print('Invalid number "{}"!'.format(pick))
                continue
            else:
                pick = int(pick)

                if pick < 1 or pick > len(user_playlists):
                    print("The number must be between {}-{}.".format(1, len(user_playlists)))
                    continue

                playlists.append(user_playlists[pick - 1])
        break

    return playlists


def lambda_handler(event, context):

    if os.environ["USER"]:
        username = os.environ["USER"]
    else:
        if len(sys.argv) > 1:
            username = sys.argv[1]
        else:
            username = input("Please type in your username: ")

    auth = authenticate_spotify(username)
    randomizer = r.SpotifyRandomizer(username, auth.get_spotify())

    if os.environ["PLAYLISTS"]:
        playlists = os.environ["PLAYLISTS"].split(',')
    else:
        if len(sys.argv) > 2:
            playlists = sys.argv[2: len(sys.argv)]
        else:
            playlists = get_playlists_by_input(randomizer)

    for playlist in playlists:
        if not is_playlist_found(randomizer, playlist):
            print("Playlist %s was not found" % playlist)
            continue
        randomizer.set_playlist_by_name(playlist)
        randomizer.randomize_playlist()

    return {
        'statusCode': 200,
        'body': json.dumps('Hello from Lambda!')
    }


if __name__ == "__main__":
    lambda_handler(None, None)