From 4c384373df753b7ad784b5613d05c1b8b2eb56b3 Mon Sep 17 00:00:00 2001 From: Rosyid Haryadi Date: Fri, 26 Jul 2024 22:48:50 +0700 Subject: cleanup file after read --- csvReader.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) (limited to 'csvReader.py') diff --git a/csvReader.py b/csvReader.py index 7644bf5..29fc347 100644 --- a/csvReader.py +++ b/csvReader.py @@ -10,10 +10,11 @@ logger = setupLogger() def getCurrentMonth(): # Terlalu males setup locale MONTH = ['jan', 'feb', 'mar', 'apr', 'may', 'jun', 'jul', 'agu', 'sep', 'okt', 'nov', 'des'] - if datetime.now().day < 24: - return MONTH[datetime.now().month - 1] + now = datetime.now() + if now.day < 24: + return MONTH[now.month - 1] else: - return MONTH[datetime.now().month] # ikutnya bulan depannya + return MONTH[now.month] # ikutnya bulan depannya def readCSV(filename=os.getenv('DOWNLOAD_PATH') + os.getenv('FILENAME')): @@ -27,6 +28,7 @@ def readCSV(filename=os.getenv('DOWNLOAD_PATH') + os.getenv('FILENAME')): if row['name'] == 'Rosyid Haryadi': myPercentage = row[currentMonth] logger.info(f'GOT RESULT FOR {currentMonth} = {myPercentage}') + os.remove(filename) except FileNotFoundError: logger.error(f'File "{filename}" not found') finally: -- cgit v1.2.3-70-g09d2