parent
8fcc67022d
commit
ddb8d16d13
5 changed files with 6758 additions and 85787 deletions
File diff suppressed because it is too large
Load Diff
@ -1,2 +0,0 @@ |
|||||||
venv |
|
||||||
mobileclient.cred |
|
File diff suppressed because it is too large
Load Diff
@ -1,217 +0,0 @@ |
|||||||
#!/usr/bin/env python3 |
|
||||||
|
|
||||||
from gmusicapi import Mobileclient |
|
||||||
import argparse |
|
||||||
import sys |
|
||||||
import requests |
|
||||||
import json |
|
||||||
import urllib.parse |
|
||||||
|
|
||||||
creds_path = sys.path[0] + '/mobileclient.cred' |
|
||||||
|
|
||||||
|
|
||||||
def authenticate(api): |
|
||||||
creds = api.perform_oauth(storage_filepath=creds_path, open_browser=False) |
|
||||||
|
|
||||||
def uploadLibrary(mudbase_api, mudbase_user, mudbase_password, tracks): |
|
||||||
# First, attempt to login and start a session. |
|
||||||
s = requests.Session() |
|
||||||
response = s.post(mudbase_api + '/login?username=' |
|
||||||
+ urllib.parse.quote(mudbase_user) |
|
||||||
+ '&password=' |
|
||||||
+ urllib.parse.quote(mudbase_password)) |
|
||||||
|
|
||||||
if response.status_code != 200: |
|
||||||
print("Unable to log in to MuDBase API.") |
|
||||||
|
|
||||||
# Helpers |
|
||||||
def getArtistStoreIds(track): |
|
||||||
if 'artistId' in track: |
|
||||||
return [track['artistId'][0]] |
|
||||||
return [] |
|
||||||
|
|
||||||
def getTrackStoreIds(track): |
|
||||||
if 'storeId' in track: |
|
||||||
return [track['storeId']] |
|
||||||
return [] |
|
||||||
|
|
||||||
# Create GPM import tag |
|
||||||
gpmTagIdResponse = s.post(mudbase_api + '/tag', json={ |
|
||||||
'name': 'GPM Import', |
|
||||||
'mbApi_typename': 'tag', |
|
||||||
'parentId': None, |
|
||||||
}).json() |
|
||||||
gpmTagId = gpmTagIdResponse['id'] |
|
||||||
print(f"Created tag \"GPM Import\", response: {gpmTagIdResponse}") |
|
||||||
|
|
||||||
# Create the root genre tag |
|
||||||
genreRootResponse = s.post(mudbase_api + '/tag', json={ |
|
||||||
'name': 'Genre', |
|
||||||
'mbApi_typename': 'tag', |
|
||||||
'parentId': None, |
|
||||||
}).json() |
|
||||||
genreRootTagId = genreRootResponse['id'] |
|
||||||
print(f"Created tag \"Genre\", response: {genreRootResponse}") |
|
||||||
|
|
||||||
# For keeping track what we have already created. |
|
||||||
storedArtists = dict() |
|
||||||
storedAlbums = dict() |
|
||||||
storedGenreTags = dict() |
|
||||||
|
|
||||||
for track in tracks: |
|
||||||
# TODO: check if these items already exist |
|
||||||
|
|
||||||
# Determine artist properties. |
|
||||||
artist = { |
|
||||||
'mbApi_typename': 'artist', |
|
||||||
'name': track['artist'], |
|
||||||
'storeLinks': ['https://play.google.com/music/m' + id for id in getArtistStoreIds(track)], |
|
||||||
'tagIds': [gpmTagId] |
|
||||||
} if 'artist' in track else None |
|
||||||
|
|
||||||
# Upload artist if not already done |
|
||||||
artistId = None |
|
||||||
if artist: |
|
||||||
for key, value in storedArtists.items(): |
|
||||||
if value == artist: |
|
||||||
artistId = key |
|
||||||
break |
|
||||||
if not artistId: |
|
||||||
response = s.post(mudbase_api + '/artist', json=artist).json() |
|
||||||
artistId = response['id'] |
|
||||||
print( |
|
||||||
f"Created artist \"{artist['name']}\", response: {response}") |
|
||||||
storedArtists[artistId] = artist |
|
||||||
|
|
||||||
# Determine album properties. |
|
||||||
album = { |
|
||||||
'mbApi_typename': 'album', |
|
||||||
'name': track['album'], |
|
||||||
'tagIds': [gpmTagId], |
|
||||||
'artistIds': [artistId], |
|
||||||
} if 'album' in track else None |
|
||||||
|
|
||||||
# Determine genre properties. |
|
||||||
genre = { |
|
||||||
'mbApi_typename': 'tag', |
|
||||||
'name': track['genre'], |
|
||||||
'parentId': genreRootTagId |
|
||||||
} if 'genre' in track else None |
|
||||||
|
|
||||||
# Upload album if not already done |
|
||||||
albumId = None |
|
||||||
if album: |
|
||||||
for key, value in storedAlbums.items(): |
|
||||||
if value == album: |
|
||||||
albumId = key |
|
||||||
break |
|
||||||
if not albumId: |
|
||||||
response = s.post(mudbase_api + '/album', json=album).json() |
|
||||||
albumId = response['id'] |
|
||||||
print( |
|
||||||
f"Created album \"{album['name']}\", response: {response}") |
|
||||||
storedAlbums[albumId] = album |
|
||||||
|
|
||||||
# Upload genre if not already done |
|
||||||
genreTagId = None |
|
||||||
if genre: |
|
||||||
for key, value in storedGenreTags.items(): |
|
||||||
if value == genre: |
|
||||||
genreTagId = key |
|
||||||
break |
|
||||||
if not genreTagId: |
|
||||||
response = s.post(mudbase_api + '/tag', json=genre).json() |
|
||||||
genreTagId = response['id'] |
|
||||||
print( |
|
||||||
f"Created genre tag \"Genre / {genre['name']}\", response: {response}") |
|
||||||
storedGenreTags[genreTagId] = genre |
|
||||||
|
|
||||||
# Upload the track itself |
|
||||||
tagIds = [gpmTagId] |
|
||||||
if genreTagId: |
|
||||||
tagIds.append(genreTagId) |
|
||||||
_track = { |
|
||||||
'mbApi_typename': 'track', |
|
||||||
'name': track['title'], |
|
||||||
'artistIds': [artistId] if artistId != None else [], |
|
||||||
'albumId': albumId if albumId else None, |
|
||||||
'tagIds': tagIds, |
|
||||||
'storeLinks': ['https://play.google.com/music/m/' + id for id in getTrackStoreIds(track)], |
|
||||||
} |
|
||||||
response = s.post(mudbase_api + '/track', json=_track).json() |
|
||||||
print( |
|
||||||
f"Created track \"{track['title']}\" with artist ID {artistId}, album ID {albumId}, response: {response}") |
|
||||||
|
|
||||||
|
|
||||||
def getData(api): |
|
||||||
return { |
|
||||||
"tracks": api.get_all_tracks(), |
|
||||||
"playlists": api.get_all_user_playlist_contents() |
|
||||||
} |
|
||||||
|
|
||||||
|
|
||||||
def getTracks(data): |
|
||||||
# Get tracks from library |
|
||||||
tracks = [] # data['tracks'] |
|
||||||
|
|
||||||
# Append tracks from playlists |
|
||||||
for playlist in data['playlists']: |
|
||||||
for track in playlist['tracks']: |
|
||||||
if 'track' in track: |
|
||||||
tracks.append(track['track']) |
|
||||||
|
|
||||||
# Uniquify by using a dict. After all, same track may appear in |
|
||||||
# multiple playlists. |
|
||||||
def sI(track): return track['artist'] + '-' + \ |
|
||||||
track['title'] if 'artist' in track and 'title' in track else 'z' |
|
||||||
return list(dict((sI(track), track) for track in tracks).values()) |
|
||||||
|
|
||||||
|
|
||||||
api = Mobileclient() |
|
||||||
|
|
||||||
parser = argparse.ArgumentParser( |
|
||||||
description="Import Google Music library into MudBase.") |
|
||||||
parser.add_argument( |
|
||||||
'--authenticate', help="Generate credentials for authentication", action="store_true") |
|
||||||
parser.add_argument('--store-to', help="Store GPM library to JSON for later upload", |
|
||||||
action='store', dest='store_to') |
|
||||||
parser.add_argument('--load-from', help="Load GPM library from JSON for upload", |
|
||||||
action='store', dest='load_from') |
|
||||||
parser.add_argument('--mudbase_api', help="Address for the Mudbase back-end API to upload to", |
|
||||||
action='store', dest='mudbase_api') |
|
||||||
parser.add_argument('--mudbase_user', help="Username for the Mudbase API", |
|
||||||
action='store', dest="mudbase_user") |
|
||||||
parser.add_argument('--mudbase_password', help="Password for the Mudbase API", |
|
||||||
action='store', dest="mudbase_password") |
|
||||||
|
|
||||||
args = parser.parse_args() |
|
||||||
|
|
||||||
if args.authenticate: |
|
||||||
authenticate(api) |
|
||||||
|
|
||||||
data = None |
|
||||||
|
|
||||||
# Determine whether we need to log in to GPM and get tracks |
|
||||||
if args.store_to or (not args.load_from and args.mudbase_api): |
|
||||||
api.oauth_login(Mobileclient.FROM_MAC_ADDRESS, |
|
||||||
oauth_credentials=creds_path) |
|
||||||
data = getData(api) |
|
||||||
|
|
||||||
# Determine whether to save to a file |
|
||||||
if args.store_to: |
|
||||||
with open(args.store_to, 'w') as outfile: |
|
||||||
json.dump(data, outfile, sort_keys=True, indent=2) |
|
||||||
|
|
||||||
# Determine whether to load from a file |
|
||||||
if args.load_from: |
|
||||||
with open(args.load_from, 'r') as f: |
|
||||||
data = json.load(f) |
|
||||||
|
|
||||||
tracks = getTracks(data) |
|
||||||
print(f"Found {len(tracks)} tracks.") |
|
||||||
|
|
||||||
if args.mudbase_api: |
|
||||||
api.oauth_login(Mobileclient.FROM_MAC_ADDRESS, |
|
||||||
oauth_credentials=creds_path) |
|
||||||
uploadLibrary(args.mudbase_api, args.mudbase_user, |
|
||||||
args.mudbase_password, tracks) |
|
@ -1,3 +0,0 @@ |
|||||||
gmusicapi |
|
||||||
argparse |
|
||||||
requests |
|
Loading…
Reference in new issue