mirror of
https://github.com/clearlinux/clr-cloud-init-svc.git
synced 2026-08-19 13:26:34 +00:00
George T Kramer
89dfb8f9cf
Separating the application from its configuration to enable the creation of install scripts with parameters users can modify
93 lines
2.4 KiB
Python
93 lines
2.4 KiB
Python
#
|
|
# Copyright 2016 Intel Corporation
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
#
|
|
|
|
from flask import Flask, abort, request, jsonify
|
|
from settings import APP_STATIC, SUBDIR
|
|
import os
|
|
import re
|
|
|
|
app = Flask(__name__)
|
|
|
|
|
|
def get_config_data(mac_addr):
|
|
data = dict()
|
|
ddata = dict()
|
|
pattern = '^%s' % mac_addr
|
|
expr = re.compile(pattern)
|
|
dpattern = '^default'
|
|
dexpr = re.compile(dpattern)
|
|
found_default = False
|
|
|
|
with open(os.path.join(APP_STATIC, 'config.txt')) as f:
|
|
for line in f:
|
|
if expr.match(line):
|
|
items = line.rstrip().split(',')
|
|
data['mac_addr'] = items[0]
|
|
data['role'] = items[1]
|
|
return jsonify(data)
|
|
if dexpr.match(line):
|
|
items = line.rstrip().split(',')
|
|
ddata['mac_addr'] = items[0]
|
|
ddata['role'] = items[1]
|
|
found_default = True
|
|
|
|
if found_default:
|
|
return jsonify(ddata)
|
|
|
|
return None
|
|
|
|
|
|
@app.route(SUBDIR + '/')
|
|
def index():
|
|
return 'Clear Cloud Init Service... is alive\n'
|
|
|
|
|
|
@app.route(SUBDIR + '/get_config/<mac_addr>')
|
|
@app.route(SUBDIR + '/get_config/')
|
|
def get_config(mac_addr=None):
|
|
reply = None
|
|
if mac_addr is None:
|
|
mac_addr = request.args.get('MAC')
|
|
if mac_addr:
|
|
reply = get_config_data(mac_addr)
|
|
if not reply:
|
|
return 'No config info for %s' % mac_addr
|
|
else:
|
|
return reply
|
|
else:
|
|
abort(404)
|
|
|
|
|
|
@app.route(SUBDIR + '/get_role/<role>')
|
|
@app.route(SUBDIR + '/get_role/')
|
|
def get_role(role=None):
|
|
reply = None
|
|
if role is None:
|
|
role = request.args.get('role')
|
|
if role:
|
|
role_file = os.path.join(APP_STATIC, "roles", role)
|
|
if os.path.isfile(role_file):
|
|
with open(role_file) as f:
|
|
return f.read()
|
|
else:
|
|
abort(404)
|
|
else:
|
|
abort(404)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
app.run()
|