diff options
Diffstat (limited to 'csvReader.py')
-rw-r--r-- | csvReader.py | 35 |
1 files changed, 25 insertions, 10 deletions
diff --git a/csvReader.py b/csvReader.py index a3c03b8..7644bf5 100644 --- a/csvReader.py +++ b/csvReader.py @@ -1,22 +1,37 @@ -import sys -from datetime import datetime import csv -from util import setupLogger +import os +from datetime import datetime -# Terlalu males setup locale -MONTH = ['jan', 'feb', 'mar', 'apr', 'may', 'jun', 'jul', 'agu', 'sep', 'okt', 'nov', 'des'] -currentMonth = MONTH[datetime.now().month - 1] +from util import setupLogger logger = setupLogger() -def readCSV(filename): +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] + else: + return MONTH[datetime.now().month] # ikutnya bulan depannya + + +def readCSV(filename=os.getenv('DOWNLOAD_PATH') + os.getenv('FILENAME')): + myPercentage = None + currentMonth = getCurrentMonth() try: - with open(filename) as csvfile: - pass + with open(filename, mode='r', newline='') as csvfile: + reader = csv.DictReader(csvfile, delimiter=',') + for row in reader: + # noinspection PyTypeChecker + if row['name'] == 'Rosyid Haryadi': + myPercentage = row[currentMonth] + logger.info(f'GOT RESULT FOR {currentMonth} = {myPercentage}') except FileNotFoundError: logger.error(f'File "{filename}" not found') + finally: + return myPercentage if __name__ == '__main__': - readCSV('ngetest') + readCSV() |