config.py

import configparser
import logging
import time
import dt_util
import sys

from pymongo import MongoClient

class ProgramConfig(object):
    def __init__(self, logfileName, environment=None, loglevel = logging.INFO, show_log = False):

        #Logging
        today = time.strftime("%Y-%m-%d")
        log_filename = "Logs/%s-%s.log" %(logfileName, today)
        logging.basicConfig(filename=log_filename, level=loglevel)
        logFormatter = logging.Formatter("%(asctime)s [%(threadName)-12.12s] [%(levelname)-5.5s]  %(message)s")
        self.loglevel = loglevel
        self.logger = logging.getLogger(__name__)

        if show_log:
            consoleHandler = logging.StreamHandler()
            consoleHandler.setFormatter(logFormatter)
            self.logger.addHandler(consoleHandler)

        self.config = configparser.ConfigParser()
        self.config.read('config.ini')

        #FIRST OF ALL, GET ENVIRONMENT
        if environment is None:
            self.environment = self.config['DZJINTONIK'].get('environment', 'test')
        else:
            self.environment = environment

        #MongoDB
        self.mongodb_server = self.config['MONGODB'].get('server', 'localhost')
        self.mongodb_port = self.config['MONGODB'].getint('port', 27017)
        self.mongodb_client = MongoClient(self.mongodb_server, self.mongodb_port)
        self.mongodb_database = None

        #SQL
        self.sql_database = self.config['SQL'].get('database')

        #LDAP
        self.ldap_server_name = self.config['LDAP'].get('server_name')
        self.ldap_domain_name = self.config['LDAP'].get('domain_name ')
        self.ldap_user_name = self.config['LDAP'].get('user_name')
        self.ldap_password = self.config['LDAP'].get('password')
        self.ldap_dsn = self.config['LDAP'].get('ldap_dsn')

        #DzjinTonik - https://dzjintonik.eu/
        self.chrome_path = self.config['DZJINTONIK'].get('chrome_path', 'C:/Program Files (x86)/Google/Chrome/Application/chrome.exe')
        self.default_dzjintonik_user_email = self.config['DZJINTONIK'].get('default_dzjintonik_user_email')
        self.internal_company_id = self.config['DZJINTONIK'].get('internal_company_id')
        self.program_id = self.config['DZJINTONIK'].get('program_id')
        self.master_program_id  = self.config['DZJINTONIK'].get('master_program_id')
        self.domain = ''
        self.api_domain = ''
        self.api_authorization = ''

        self.store_contact_pictures_in = self.config['DZJINTONIK'].get('store_contact_pictures_in', '')

        if self.continue_in_environment() == False:
            sys.exit()

        self.set_env_specific_values()
        self.headers = {'Accept': 'application/json','Content-Type': 'application/x-www-form-urlencoded'}
        self.cookies = {'InternalCompanyId': '1', 'ASP.NET_SessionId': self.asp_session, '__RequestVerificationToken': self.request_token, 'ProgramId':''}

        # Now in models
        #self.contact_get = self.domain + '/Contact/Get'
        #self.contact_update = self.domain + '/Contact/Update'
        #self.person_get = self.domain + '/Person/Get'
        #self.person_update = self.domain + '/Person/Update'
        #self.set_recup_hours = self.domain + '/PercentageReportPlanBalance/Create'

    def set_env_specific_values(self):
        the_env = self.environment.upper()
        if the_env == 'TEST':
            dt_domain = self.config['DZJINTONIK'].get('test_domain')
            my_api_domain = self.config['DZJINTONIK'].get('api_test_domain')
            my_api_authorization = self.config['DZJINTONIK'].get('api_test_authorization')
            my_mongodb = self.mongodb_client.dbDzjinTonikTEST
            my_asp_session = self.config['DZJINTONIK'].get('asp_session_test')
            my_request_token =  self.config['DZJINTONIK'].get('request_token_test')
            my_sql_server = self.config['SQL'].get('server_test')

        elif the_env == 'UAT':
            dt_domain = self.config['DZJINTONIK'].get('uat_domain')
            my_api_domain = self.config['DZJINTONIK'].get('api_uat_domain')
            my_api_authorization = self.config['DZJINTONIK'].get('api_uat_authorization')
            my_mongodb = self.mongodb_client.dbDzjinTonikUAT
            my_asp_session = self.config['DZJINTONIK'].get('asp_session_uat')
            my_request_token =  self.config['DZJINTONIK'].get('request_token_uat')
            my_sql_server = self.config['SQL'].get('server_uat')
        elif the_env == 'PROD':
            dt_domain = self.config['DZJINTONIK'].get('prod_domain')
            my_api_domain = self.config['DZJINTONIK'].get('api_prod_domain')
            my_api_authorization = self.config['DZJINTONIK'].get('api_prod_authorization')
            my_mongodb = self.mongodb_client.dbDzjinTonikPROD
            my_asp_session = self.config['DZJINTONIK'].get('asp_session_prod', None)
            my_request_token =  self.config['DZJINTONIK'].get('request_token_prod', 'test')
            my_sql_server = self.config['SQL'].get('server_prod')
        else:
            print("No valid environment")
            sys.exit()



        print("THE API DOMAIN WILL BE: %s" %my_api_domain)
        self.api_domain = my_api_domain

        print("The API AUTH: %s" %my_api_authorization)
        self.api_authorization = my_api_authorization

        print("THE DOMAIN WILL BE: %s" %dt_domain)
        self.domain = dt_domain

        self.mongodb_database = my_mongodb
        self.asp_session = my_asp_session
        self.request_token = my_request_token
        self.sql_server = my_sql_server

        self.api_headers = {
                'Accept': "application/json",
                'Authorization': self.api_authorization
                }

    def continue_in_environment(self):

        if self.environment is None:
            print("Please set an environment...")
            system.exit()

        self.environment = self.environment.upper()

        if 'PROD' in self.environment.upper():
            print("YOU ARE WORKING IN PRODUCTION ENVIRONMENT!")
            answer = dt_util.ask_yes_no_question("Are you sure you'd like to continue in PRODUCTION?")
            if not answer:
                print("Stopping actions!")
                sys.exit()
        else:
            print("YOU ARE WORKING IN %s, WHICH IS NICE!" %self.environment)