summaryrefslogtreecommitdiff
path: root/csvReader.py
blob: 241cab38bbfa5ddc70dc57c1fb511e92a5cbbcbc (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
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()