db.py

import datetime

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"]


def get_gap(song: pandas.Series):
    return song["GAP"]


def get_video_gap(song: pandas.Series):
    return song["VideoGAP"]


def get_start(song: pandas.Series):
    return song["Start"]


def get_end(song: pandas.Series):
    return song["End"]


def get_language(song: pandas.Series):
    return song["Language"]

def get_date(song: pandas.Series):
    return datetime.date.fromisoformat(song["Release Date"])


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