33 lines
632 B
Python
33 lines
632 B
Python
|
from google.cloud import firestore
|
||
|
|
db = firestore.Client()
|
||
|
|
|
||
|
|
|
||
|
|
def get_user_query_stream(user):
|
||
|
|
|
||
|
|
users = db.collection(u'spotify_users').where(u'username', u'==', user).stream()
|
||
|
|
users = [i for i in users]
|
||
|
|
|
||
|
|
return users
|
||
|
|
|
||
|
|
|
||
|
|
def get_user_doc_ref(user):
|
||
|
|
|
||
|
|
users = get_user_query_stream(user)
|
||
|
|
|
||
|
|
if len(users) == 1:
|
||
|
|
|
||
|
|
return db.collection(u'spotify_users').document(u'{}'.format(users[0].id))
|
||
|
|
|
||
|
|
else:
|
||
|
|
print(len(users))
|
||
|
|
raise ValueError
|
||
|
|
|
||
|
|
|
||
|
|
def get_user_playlists_collection(user_id):
|
||
|
|
|
||
|
|
playlists = db.document(u'spotify_users/{}'.format(user_id)).collection(u'playlists')
|
||
|
|
|
||
|
|
return playlists
|
||
|
|
|
||
|
|
|