-
Notifications
You must be signed in to change notification settings - Fork 5
/
startup.py
177 lines (153 loc) · 6.34 KB
/
startup.py
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
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
#
# Copyright 2015 Telefonica Investigacion y Desarrollo, S.A.U
#
# This file is part of IoT orchestrator
#
# IoT orchestrator is free software: you can redistribute it and/or
# modify it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# IoT orchestrator is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Affero
# General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with IoT orchestrator. If not, see http://www.gnu.org/licenses/.
#
# For those usages not covered by this license please contact with
# iot_support at tid dot es
#
# Author: IoT team
#
import logging
import os
import json
from django.conf import settings
from orchestrator.core.keystone import IdMKeystoneOperations as IdMOperations
from orchestrator.core.keypass import AccCKeypassOperations as AccCOperations
from orchestrator.core.mongo import MongoDBOperations as MongoDBOperations
from orchestrator.core.orion import CBOrionOperations as CBOrionOperations
from orchestrator.core.perseo import PerseoOperations as PerseoOperations
from orchestrator.core.openldap import OpenLdapOperations as OpenLdapOperations
from orchestrator.core.mailer import MailerOperations as MailerOperations
from orchestrator.common.util import ContextFilterCorrelatorId
from orchestrator.common.util import ContextFilterTransactionId
from orchestrator.common.util import ContextFilterService
from orchestrator.common.util import ContextFilterSubService
from orchestrator.common.util import ContextFilterFrom
logger = logging.getLogger('orchestrator_api')
logger.addFilter(ContextFilterCorrelatorId("n/a"))
logger.addFilter(ContextFilterTransactionId("n/a"))
logger.addFilter(ContextFilterService("None"))
logger.addFilter(ContextFilterSubService(""))
logger.addFilter(ContextFilterFrom("n/a"))
logger1 = logging.getLogger('orchestrator_core')
logger1.addFilter(ContextFilterCorrelatorId("n/a"))
logger1.addFilter(ContextFilterTransactionId("n/a"))
logger1.addFilter(ContextFilterService("None"))
logger1.addFilter(ContextFilterSubService(""))
logger1.addFilter(ContextFilterFrom("n/a"))
logger2 = logging.getLogger('django')
logger2.addFilter(ContextFilterCorrelatorId("n/a"))
logger2.addFilter(ContextFilterTransactionId("n/a"))
logger2.addFilter(ContextFilterService("None"))
logger2.addFilter(ContextFilterSubService(""))
logger2.addFilter(ContextFilterFrom("n/a"))
logger3 = logging.getLogger('django.request')
logger3.addFilter(ContextFilterCorrelatorId("n/a"))
logger3.addFilter(ContextFilterTransactionId("n/a"))
logger3.addFilter(ContextFilterService("None"))
logger3.addFilter(ContextFilterSubService(""))
logger3.addFilter(ContextFilterFrom("n/a"))
def read_banner():
banner_dir = os.path.dirname(__file__)
handle = open(banner_dir + '/' + 'banner.txt', 'r+')
var = handle.read()
return var
def check_endpoints():
KEYSTONE_PROTOCOL = settings.KEYSTONE['protocol']
KEYSTONE_HOST = settings.KEYSTONE['host']
KEYSTONE_PORT = settings.KEYSTONE['port']
KEYPASS_PROTOCOL = settings.KEYPASS['protocol']
KEYPASS_HOST = settings.KEYPASS['host']
KEYPASS_PORT = settings.KEYPASS['port']
idm = IdMOperations(KEYSTONE_PROTOCOL, KEYSTONE_HOST, KEYSTONE_PORT)
try:
idm.checkIdM()
logger.info("Keystone endpoint OK")
except Exception as ex:
logger.error("keystone endpoint not found: %s" % ex)
return "ERROR keystone endpoint not found: %s" % ex
ac = AccCOperations(KEYPASS_PROTOCOL, KEYPASS_HOST, KEYPASS_PORT)
try:
ac.checkAccC()
logger.info("Keypass endpoint OK")
except Exception as ex:
logger.error("keypass endpoint not found: %s" % ex)
return "ERROR keypass endpoint not found: %s" % ex
# MongoDB: optional
MONGODB_URI = settings.MONGODB['URI']
mongo = MongoDBOperations(MONGODB_URI)
try:
mongo.checkMongo()
logger.info("MongoDB endpoint OK")
except Exception as ex:
logger.warn("MongoDB endpoint not found: %s" % ex)
# ContextBroker: optional
ORION_HOST = settings.ORION['host']
ORION_PORT = settings.ORION['port']
ORION_PROTOCOL = settings.ORION['protocol']
orion = CBOrionOperations(ORION_PROTOCOL, ORION_HOST, ORION_PORT)
try:
orion.checkCB()
logger.info("Orion endpoint OK")
except Exception as ex:
logger.warn("Orion endpoint not found: %s" % ex)
# Perseo: optional
PERSEO_HOST = settings.PERSEO['host']
PERSEO_PORT = settings.PERSEO['port']
PERSEO_PROTOCOL = settings.PERSEO['protocol']
orion = PerseoOperations(PERSEO_PROTOCOL, PERSEO_HOST, PERSEO_PORT)
try:
orion.checkPERSEO()
logger.info("PERSEO endpoint OK")
except Exception as ex:
logger.warn("PERSEO endpoint not found: %s" % ex)
# OpenLDAP: optional
LDAP_HOST = settings.LDAP['host']
LDAP_PORT = settings.LDAP['port']
LDAP_BASEDN = settings.LDAP['basedn']
openldap = OpenLdapOperations(LDAP_HOST, LDAP_PORT, LDAP_BASEDN)
try:
openldap.checkLdap()
logger.info("LDAP endpoint OK")
except Exception as ex:
logger.warn("LDAP endpoint not found: %s" % ex)
# Mailer: optional
MAILER_HOST = settings.MAILER['host']
MAILER_PORT = settings.MAILER['port']
mailer = MailerOperations(MAILER_HOST, MAILER_PORT)
try:
mailer.checkMailer()
logger.info("MAILER endpoint OK")
except Exception as ex:
logger.warn("MAILER endpoint not found: %s" % ex)
return "OK"
def show_conf():
conf = {}
from django.conf import settings
custom_settings_entries = ['KEYSTONE', 'KEYPASS',
'ORION',
'PEP', 'PEP_PERSEO', 'SCIM_API_VERSION',
'CYGNUS', 'STH', 'PERSEO',
'LDAP', 'MAILER', 'MONGODB'
]
for name in custom_settings_entries:
conf[name] = getattr(settings, name)
return conf
def run():
logger.info("Starting Service %s " % read_banner())
logger.info("Checking endpoints %s " % check_endpoints())
logger.info("Using custom conf: %s " % json.dumps(show_conf(), indent=3))