Skip to content
Snippets Groups Projects
db.py 647 B
import pandas as pd
import pandas.core.series


def load(file: str):
    with open(file, 'rb') as f:
        return pd.read_excel(f)

def get_usdb_url(song: pandas.Series):
    return song["TXT Link"]

def get_cover_image_url(song: pandas.Series):
    return song["Cover Link"]

def get_yt_video_url(song: pandas.Series):
    return song["Video Link"]

def get_artist_name(song: pandas.Series):
    return song["Artist Name"]

def get_song_name(song: pandas.Series):
    return song["Track Name"]

def get_track_id(song: pandas.Series):
    return song["Spotify URI"]

class USDB:
    def __init__(self, file: str):
        self.data = load(file)