main.py

import sys
import time

from module import cs_noti, gnu_noti, agency_noti


def main():
    init_secret()

    interval = init()
    option = mode_select()

    if option == 'all':
        while True:
            gnuModule = gnu_noti.GnuNotification()
            gnuModule.run()

            # TODO: Bug fix후 주석해제
            # csModule = cs_noti.csNotification()
            # csModule.run()

            agencyModule = agency_noti.AgencyNotification()
            agencyModule.run()

            time.sleep(interval)

    elif option == 'cs':
        while True:
            csModule = cs_noti.CsNotification()
            csModule.run()
            time.sleep(interval)

    elif option == 'gnu':
        while True:
            gnuModule = gnu_noti.GnuNotification()
            gnuModule.run()
            time.sleep(interval)

    elif option == 'agency':
        while True:
            agencyModule = agency_noti.AgencyNotification()
            agencyModule.run()
            time.sleep(interval)


def mode_select():
    if len(sys.argv) == 1:
        print('ERROR: There is no Parameter. Please check README')
    # mode=[all,cs,gnu,agency]
    mode = [False, False, False, False]

    print(sys.argv)
    for a in sys.argv:
        if a == 'all':
            mode[0] = True
        elif a == 'cs':
            mode[1] = True
        elif a == 'gnu':
            mode[2] = True
        elif a == 'agency':
            mode[3] = True

    if mode[1] == True and mode[2] == True and mode[3] == True:
        return 'all'
    if mode[0] == True:
        return 'all'
    elif mode[1] == True:
        return 'cs'
    elif mode[2] == True:
        return 'gnu'
    elif mode[3] == True:
        return 'agency'
    # This case is occurred when user enter incorrect option
    print('ERROR:Please Enter Correct Option')
    exit()


def init_secret():
    import os

    try:
        _ = os.environ["GNU_CHANNEL"]
        _ = os.environ["CS_CHANNEL"]
    except KeyError:
        print("ERROR: Could not found TELEGRAM CHANNEL INFO")
        exit(1)

    try:
        _ = os.environ["NAVER_ID"]
        _ = os.environ["NAVER_SECRET"]
    except KeyError:
        try:
            path = os.path.dirname(os.path.dirname(os.path.abspath( __file__ )))
            file = open(path + "/source/naver", "r")
            file.close()
        except FileNotFoundError:
            print("ERROR: Could not found NAVER Secret")
            exit(1)

    try:
        _ = os.environ["TELEGRAM_API"]

    except KeyError:
        try:
            path = os.path.dirname(os.path.dirname(os.path.abspath( __file__ )))
            file = open(path + "/source/api_key")
            file.close()
        except FileNotFoundError:
            print("ERROR: Could not fount TELEGRAM API Secret")
            exit(1)


def init():
    import os

    try:
        param = os.environ["DEBUG"]
        if param == "True":
            print("DEBUG Mode Enabled")
            return 10
    except KeyError:
        print("PRODUCTION Mode Enabled")
        return 1800


if __name__ == '__main__':
    main()