mopidy-subidy/mopidy_subidy/uri.py

import re

SONG = 'song'
ARTIST = 'artist'
PLAYLIST = 'playlist'
ALBUM = 'album'
DIRECTORY = 'directory'
VDIR = 'vdir'
PREFIX = 'subidy'
SEARCH = 'search'

regex = re.compile(r'(\w+?):(\w+?)(?::|$)(.+?)?$')

def is_type_result_valid(result):
    return result is not None and result.group(1) == PREFIX

def is_id_result_valid(result, type):
    return is_type_result_valid(result) and result.group(1) == PREFIX and result.group(2) == type

def is_uri(uri):
    return regex.match(uri) is not None

def get_song_id(uri):
    result = regex.match(uri)
    if not is_id_result_valid(result, SONG):
        return None
    return result.group(3)

def get_artist_id(uri):
    result = regex.match(uri)
    if not is_id_result_valid(result, ARTIST):
        return None
    return result.group(3)

def get_playlist_id(uri):
    result = regex.match(uri)
    if not is_id_result_valid(result, PLAYLIST):
        return None
    return result.group(3)

def get_album_id(uri):
    result = regex.match(uri)
    if not is_id_result_valid(result, ALBUM):
        return None
    return result.group(3)

def get_directory_id(uri):
    result = regex.match(uri)
    if not is_id_result_valid(result, DIRECTORY):
        return None
    return result.group(3)

def get_vdir_id(uri):
    result = regex.match(uri)
    if not is_id_result_valid(result, VDIR):
        return None
    return result.group(3)

def get_type(uri):
    result = regex.match(uri)
    if not is_type_result_valid(result):
        return None
    return result.group(2)

def get_type_uri(type, id):
    return f'{PREFIX}:{type}:{id}'

def get_artist_uri(id):
    return get_type_uri(ARTIST, id)

def get_album_uri(id):
    return get_type_uri(ALBUM, id)

def get_song_uri(id):
    return get_type_uri(SONG, id)

def get_directory_uri(id):
    return get_type_uri(DIRECTORY, id)

def get_vdir_uri(id):
    return get_type_uri(VDIR, id)

def get_playlist_uri(id):
    return get_type_uri(PLAYLIST, id)

def get_search_uri(query):
    return get_type_uri(SEARCH, query)