iHeartTemplate.py

Go to the documentation of this file.

import re
 
class iHeartTemplate:
    # "gyrostream-pty-ltd_iheart_202203_Monthly-Sales_(1).txt"
    def date(self, filename):
        """Sets date column given the filename (it contains the date)
           For example gyrostream-pty-ltd_iheart_202203_Monthly-Sales_(1).txt
        Args:
            filename (str): current filename
        Returns: date_str (str)
        """
        date = re.findall(r'_\d{6}_', filename)[0]
        date = date.replace("_", "")
        date_str = date[:4]+"-"+date[4:]+"-01"
        # df["date_from_file"] = date_str
        return date_str