python_falcon_api_server/jdbe.py at master · birendra7654/python_falcon_api_server · GitHub

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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131

"""This file is mainly used to parse  .ini config file and making database connection."""
# !/usr/bin/env python
#

import logging
from ConfigParser import ConfigParser
import utils
from dbstore.jdb_mysql import jdb_mysql
from dbstore.jdb_mongodb import jdb_mongodb
from dbstore.jdb_redis import jdb_redis
from dbstore.jdb_elasticsearch import jdb_elasticsearch
import sys
import traceback

__author__ = 'Sanctum Networks (P) Ltd.'


app_log = logging.getLogger("falcon")


# JDB engines


class jdb(jdb_redis, jdb_mysql, jdb_mongodb, jdb_elasticsearch):
    """JDB engine class."""

    def __init__(self):
        """Init method of this class and calling base class init method."""
        super(jdb, self).__init__()

    def setup(self, u, v, w, z):
        """Set up various db source connection."""
        try:
            jdb_redis.setup(self, u)
            jdb_mysql.setup(self, v)
            jdb_mongodb.setup(self, w)
            jdb_elasticsearch.setup(self, z)
        except AttributeError, e:
            app_log.info("jdb:setup: NOIMPL {}".format(e))
            traceback.print_exc(file=sys.stdout)
            sys.stderr.close()


class jdbe:
    def __init__(self, conf):
        app_log.debug("jdbe:init:start Parse {}".format(conf))

        jdbe_ini = ConfigParser()
        jdbe_ini.read(conf)

        self.mongo_ip = jdbe_ini.get('JDBE', 'mongo_ip')
        self.mongo_port = jdbe_ini.get('JDBE', 'mongo_port')
        self.j_mongodb_name = jdbe_ini.get('JDBE', 'jupiter_mongodb_name')

        self.redis_ip = jdbe_ini.get('redis', 'redis_ip')
        self.redis_port = jdbe_ini.get('redis', 'redis_port')

        self.mip = jdbe_ini.get('mysql', 'mysql_ip')
        self.mport = jdbe_ini.get('mysql', 'mysql_port')
        self.muser = jdbe_ini.get('mysql', 'user')
        self.mpassword = jdbe_ini.get('mysql', 'password')

        self.es_ip = jdbe_ini.get('ELASTICSEARCH', 'es_ip')
        self.es_port = jdbe_ini.get('ELASTICSEARCH', 'es_port')

        # Assemble parameters
        self.jdb_redis_params = {
            "server": self.redis_ip,
            "port": self.redis_port
        }
        self.jdb_mongodb_params = {
            "server": self.mongo_ip,
            "port": self.mongo_port,
            "dbname": self.j_mongodb_name
        }
        self.jdb_mysql_params = {
            "server": self.mip,
            "port": self.mport,
            "user": self.muser,
            "password": self.mpassword
        }

        self.jdb_elasticsearch_params = {
            "server": self.es_ip,
            "port": self.es_port,
        }

        app_log.debug("jdbe:init:complete")

    def _check_params(self):
        app_log.debug("jdbe:check:start")
        utils.dump_object(self)
        # raise TypeError, "Parameter error"
        app_log.debug("jdbe:check:complete")

    def start(self):
        app_log.debug("jdbe:start:start")

        #
        # Checkinng var(self) has proper valid json
        self._check_params()

        #
        utils.setup_sighandlers()

        #
        self.jdb = jdb()

        self.jdb.setup(
                self.jdb_redis_params,
                self.jdb_mysql_params,
                self.jdb_mongodb_params,
                self.jdb_elasticsearch_params
                )

        # raise TypeError, "bogus type error for testing"
        #
        app_log.debug("jdbe:start:complete")

    def get_mongodb_connection_auth(self):
        return self.jdb.get_mongodb_client(self.j_mongodb_name)

    def get_redis_connection(self):
        return self.jdb.get_redis_client()

    def get_mysql_connection(self):
        return self.jdb.get_mysql_client()

    def get_elasticsearch_connection(self):
        return self.jdb.get_elasticsearch_client()