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
|
#!/usr/bin/python3
# Copyright (C) 2018 Lars Wirzenius
#
# This program 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.
#
# This program 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 this program. If not, see <http://www.gnu.org/licenses/>.
import logging
import logging.handlers
import os
import sys
import apifw
import slog
import yaml
import ick2
transactions = slog.Counter()
def counter():
return 'HTTP transaction {}'.format(transactions.increment())
def dict_logger(log, stack_info=None):
ick2.log.log(exc_info=stack_info, **log)
default_config = {
'token-public-key': None,
'token-audience': None,
'token-issuer': None,
'log': [],
'from_addr': None,
'smtp_server': None,
'smtp_port': None,
'smtp_user': None,
'smtp_password': None,
'controller_url': None,
}
def load_config(filename, defconf):
conf = yaml.safe_load(open(filename, 'r'))
actual_config = dict(defconf)
actual_config.update(conf)
return actual_config
def check_config(config, musthave):
for key in config:
if key not in musthave:
raise Exception('Config %s is not known' % key)
for key in musthave:
if config.get(key) is None:
raise Exception('Config %s must not be None' % key)
def main():
logger = logging.getLogger()
logger.setLevel(logging.DEBUG)
handler = logging.StreamHandler(sys.stderr)
logger.addHandler(handler)
logging.info('Starting notification service')
try:
name = 'NOTIFICATION_SERVICE_CONFIG'
config_filename = os.environ.get(name)
logging.info('config filename %r', config_filename)
if not config_filename:
msg = 'No %s defined in environment' % name
logging.error(msg)
raise Exception(msg)
logging.info('reading config from %r', config_filename)
config = load_config(config_filename, default_config)
logging.info('config is %r', config)
ick2.setup_logging(config)
check_config(config, default_config)
ick2.log.log(
'info', msg_text='Notification service starts', config=config)
api = ick2.NotificationAPI(config)
ick2.log.log('info', msg_text='created NotificationAPI')
application = apifw.create_bottle_application(
api, counter, dict_logger, config)
ick2.log.log('info', msg_text='called apifw.create_bottle_application')
return application
except SystemExit:
raise
except BaseException as e:
logging.error(str(e))
ick2.log.log(
'error', msg_text='Uncaught exception', exception=str(e),
exc_info=True)
sys.exit(1)
app = main()
# If we are running this program directly with Python, and not via
# gunicorn, we can use the Bottle built-in debug server, which can
# make some things easier to debug.
if __name__ == '__main__':
print('running in debug mode')
app.run(host='127.0.0.1', port=12767)
|