"""Common os related functions.""" import os from tempfile import gettempdir def create_temp_dir(feed_name: str, date: str) -> str: """Create temporary directory for particular flow. Args: date (str): Reporting date (YYYY-MM-DD). feed_name (str): Name of the feed. Return: temp_dir: path to temporary directory (i.e. '/var/tmp/deezer_daily_theorchard/2021-03-24') """ # additional storage for youtube_claim feed if feed_name.startswith('youtube_claim'): base_dir = '/mnt/efs' else: base_dir = gettempdir() temp_dir = os.path.join(base_dir, feed_name, date) os.makedirs(temp_dir, exist_ok=True) return temp_dir