SpomixAPI/firebasefunctions.py at main · EinarLop/SpomixAPI · 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
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
from firebaseconfig import *
import helperfunctions

def create_group(group_name, user_id):
    doc_ref = db.collection(u'groups')
    response = doc_ref.add({
        u'name': group_name,
        u'members': [user_id],
    })
    group_id = response[1].id
    return group_id


def join_group(group_id, user_id):
    doc_ref = db.collection(u'groups').document(group_id)
    doc_snapshot = doc_ref.get()
    if doc_snapshot.exists:
        doc_ref.update({
            u'members': firestore.ArrayUnion([user_id])
        })
        return True
    return False

def update_group_playlist(group_id, playlist):
    doc_ref = db.collection(u'groups').document(group_id)
    doc_snapshot = doc_ref.get()
    if doc_snapshot.exists:
        doc_ref.update({
            u'playlist': playlist
        })
        return True
    return False


def set_user(user_id: str,  refresh_token: str):
    doc_ref = db.collection(u'users').document(user_id)
    doc_ref.set({
        u'refresh_token': refresh_token,
        u'artists': [],
        u'tracks': [],
        u'genres': []
    })


def get_user_token(user_id):
    doc_ref = db.collection(u'users').document(user_id)
    doc_snapshot = doc_ref.get()
    if doc_snapshot.exists:
        doc_data = doc_snapshot.to_dict()
        print("Document data: {}".format(doc_data))
    else:
        print("Document {} does not exist!".format(user_id))


def update_user_artists(user_id, artists, genres):

    doc_ref = db.collection(u'users').document(user_id)
    doc_snapshot = doc_ref.get()
    if doc_snapshot.exists:
        doc_ref.update({
            u'artists': artists,
            u'genres': genres
        })
        return True
    return False

def update_user_tracks(user_id, tracks):
    doc_ref = db.collection(u'users').document(user_id)
    doc_snapshot = doc_ref.get()

    if doc_snapshot.exists:
        response = doc_ref.update({
            u'tracks': tracks,
        })
        print("update_user_tracks", response)

def get_group_seeds(group_id):
    group_tracks = []
    group_artists = []
    group_genres = []
    doc_ref = db.collection(u'groups').document(group_id)
    doc_snapshot = doc_ref.get()
    if doc_snapshot.exists:
        doc_data = doc_snapshot.to_dict()
        group_members = doc_data["members"]
        for member in group_members:
            user_ref = db.collection(u'users').document(member)
            user_snapshot = user_ref.get()
            if user_snapshot.exists:
                user_data = user_snapshot.to_dict()
                group_tracks += user_data["tracks"]
                group_artists += user_data["artists"]
                group_genres += user_data["genres"]
        return group_tracks, group_artists, group_genres

    else:
        print("Document {} does not exist!".format(group_id))
        return None


def get_group(group_id, access_token):
    doc_ref = db.collection(u'groups').document(group_id)

    doc_snapshot = doc_ref.get()
    if doc_snapshot.exists:
         doc_data = doc_snapshot.to_dict()
         members = doc_data.get("members")
         profiles = []
         if members:
             for member in members:
                current_member = helperfunctions.get_me_full(access_token, member)
                if current_member:
                    profiles.append(current_member)
         doc_data["profiles"] = profiles
         return doc_data
    return None

def get_my_groups(user_id, access_token ):
    doc_ref = db.collection(u'groups')
    docs = doc_ref.where('members', 'array_contains', user_id).stream()
    if docs:
        groups = []
        for doc in docs:
            current_doc = doc.to_dict()
            current_doc["id"] = doc.id
            current_members = current_doc.get("members")
            profiles = []
            for member in current_members:
                current_member = helperfunctions.get_me_full(access_token, member)
                if current_member:
                    profiles.append(current_member)
            current_doc["profiles"] = profiles
            groups.append(current_doc)
        return groups
    return None

def get_group_playlist(group_id):
    doc_ref = db.collection(u'groups').document(group_id)
    doc_snapshot = doc_ref.get()
    if doc_snapshot.exists:
         doc_data = doc_snapshot.to_dict()
         return doc_data.get("playlist")
    return None