NightlifeTemplate.py

Go to the documentation of this file.

import re
 
class NightlifeTemplate:
    def date(self, filename):
        """Sets date column given the filename (it contains the date)
           For example Nightlife_2022_Q4.csv
        Args:
            filename (str): current filename
        Returns: date_str (str)
        """
        # "Nightlife_2022_Q4.csv"
        quarter = {"Q1": "01", "Q2": "04", "Q3": "07", "Q4": "10"}
        year = re.findall(r'_20\d{2}', filename)[0]
        year = year.replace("_", "")
        q = re.findall(r'_Q\d{1}', filename)[0]
        q = q.replace("_", "")
        month = quarter[q]
        date_str = year+"-"+month+"-01"
        # df["date_from_file"] = date_str
        return date_str