QobuzTemplate.py

Go to the documentation of this file.

import re
import awswrangler as wr
from datetime import datetime
 
class QobuzTemplate:
    def type(self, asset_type, filename):
        """Maps the sale type with the name specified by the client according their requirements
        Args:
            asset_type (str): sale type shown in the original file
            filename (str):  current filename, it says if file contains streaming or downloads
        Returns: (sale type in a single one letter)
        """
        if "download" in filename:
            if asset_type=="track":
                return 'T'
            elif asset_type=="album":
                return 'A'
        return 'S'
 
    def assetType(self, df, filename):
        """Applies type functions to current dataframe
        Args:
            df (pandas dataframe): dataframe where changes will applied
        Returns: df (pandas dataframe)
        """
        # Add a new column to insert type
        df["asset_type"] = df.apply(lambda Row: self.type(Row['album or track'], filename), axis=1)
        return df
 
    def date(self, filename):
        """Sets date column given the filename (it contains the date)
           For example qobuz-eu-gyrost-202205-streaming.tsv
        Args:
            filename (str): current filename
        Returns: date_str (str)
        """
        # "qobuz-eu-gyrost-202205-streaming.tsv"
        date = re.findall(r'-20\d{4}', filename)[0]
        date = date.replace("-", "")
        date_str = date[:4]+"-"+date[4:6]
        # df["date_from_file"] = date_str
        return date_str