import csv import os from datetime import datetime from util import setupLogger logger = setupLogger() def getCurrentMonth(): # Terlalu males setup locale MONTH = ['jan', 'feb', 'mar', 'apr', 'may', 'jun', 'jul', 'agu', 'sep', 'okt', 'nov', 'des'] now = datetime.now() if now.day < 24: return MONTH[now.month - 1] else: return MONTH[now.month] # ikutnya bulan depannya def readCSV(filename=os.getenv('DOWNLOAD_PATH') + os.getenv('FILENAME')): myScore = None currentMonth = getCurrentMonth() try: with open(filename, mode='r', newline='') as csvfile: reader = csv.DictReader(csvfile, delimiter=',') for row in reader: # noinspection PyTypeChecker if row['name'] == os.getenv('NAME'): myScore = row[currentMonth] logger.info(f'GOT RESULT FOR {currentMonth} = {myScore}') break os.remove(filename) except FileNotFoundError: logger.error(f'File "{filename}" not found') finally: return myScore if __name__ == '__main__': readCSV()