From c16646951a296baebd55bc1ff2e622d85bc89d3e Mon Sep 17 00:00:00 2001 From: Rosyid Haryadi Date: Fri, 26 Jul 2024 22:39:11 +0700 Subject: upd csv reader --- README | 2 +- csvReader.py | 35 +++++++++++++++++++++++++---------- main.py | 4 +++- 3 files changed, 29 insertions(+), 12 deletions(-) diff --git a/README b/README index c31b9b0..f8ba134 100644 --- a/README +++ b/README @@ -1 +1 @@ -Pengingat ngisi leantime \ No newline at end of file +### Over-engineered alarm reminder 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() diff --git a/main.py b/main.py index 5a7e9f7..7f249ae 100644 --- a/main.py +++ b/main.py @@ -1,3 +1,5 @@ from downloader import downloadCsv +from csvReader import readCSV -downloadCsv() \ No newline at end of file +downloadCsv() +thisMonthPercentage = readCSV() \ No newline at end of file -- cgit v1.2.3-70-g09d2