State51Template.py

Go to the documentation of this file.

import re
 
class State51Template: 
    def serviceFinder(self, service):
        # print(service)
        if "Red" in service:
            return "YouTube Music"
        elif "UMA" in service:
            return "UMA"
        elif "Deezer" in service:
            return "Deezer"
        elif "Google Play" in service:
            return "Google Play"
        elif "- YouTube" in service or "Youtube Ads Revenue" in service:
            return "YouTube Content ID"
        elif "YouTube - Subscription" in service:
            return "YouTube Music"
        elif "iTunes" in service:
            return "iTunes"
        elif "Amazon" in service:
            return "Amazon"
        elif "Spotify" in service:
            return "Spotify"
        elif "Tencent" in service:
            return "Tencent"
        elif "Tidal" in service:
            return "Tidal"
        elif "Alibaba" in service:
            return "Alibaba"
        elif "Netease" in service:
            return "Netease"
        elif "Pandora" in service:
            return "Pandora"
        elif "Yandex" in service:
            return "Yandex"
        elif "Facebook" in service:
            return "Facebook"
        elif "Audio tier" in service:
            return "Youtube Audio Tier"
        elif "Tik Tok" in service:
            return "TikTok"
        elif "Slacker" in service:
            return "Slacker"
        elif "Soundcloud" in service:
            return "Soundcloud"
        return service
    
    def type(self, service):
        """Maps the sale type using the service, according to their requirements
        Args:
            service (str): service shown in the original file
        Returns: (sale type in a single one letter)
        """
        if "Amazon" in service or "Google Play" in service:
            return "T"
        return "S"
    
    def assetType(self, df):
        """Applies type functions to current dataframe
        Args:
            df (pandas dataframe): dataframe where changes will applied
        Returns: df (pandas dataframe)
        """
        df['getservice'] = df.apply(lambda Row: self.serviceFinder(Row['music service']), axis=1)
        df["gettypestate51"] = df.apply(lambda Row: self.type(Row['getservice']), axis=1)
        print("type")
        return df
 
    def date(self, filename):
        """Sets date column given the filename (it contains the date)
           For example 12822_detail_2021_12_14_982.csv
        Args:
            filename (str): current filename
        Returns: date_str (str)
        """
        # "12822_detail_2021_12_14_982.csv"
        year = re.findall(r'_\d{4}_', filename)[0]
        month = re.findall(r'_\d{2}_', filename)[0]
        year = year.replace("_", "")
        month = month.replace("_", "")
        date_str = year+month
        return date_str