initial drop of example

Signed-off-by: Martin Buchleitner <[email protected]>
infralovers · Sep 25, 2024 · 0233870 · 0233870
commit 0233870
Show file tree

Hide file tree

Showing 47 changed files with 1,952 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1,2 @@
+__pycache__/
+.DS_Store
diff --git a/Dockerfile b/Dockerfile
@@ -0,0 +1,15 @@
+FROM python:3
+
+WORKDIR /usr/src/app
+ADD requirements.txt /tmp/
+RUN pip install --no-cache-dir -r /tmp/requirements.txt
+
+ADD templates ./templates/
+ADD static ./static/
+ADD app.py db_client.py db_client_transform.py ./
+ADD config/config.ini ./config/
+
+
+VOLUME ["/usr/src/app/config"]
+
+CMD python ./app.py
diff --git a/app.py b/app.py
@@ -0,0 +1,168 @@
+from flask import Flask, request, render_template, abort
+
+from datetime import datetime
+from os import getenv
+import configparser
+import json
+import logging
+import logging.config
+
+import db_client
+import db_client_transform
+
+dbc = None
+vclient = None
+
+log_level = {
+  'CRITICAL' : 50,
+  'ERROR'	   : 40,
+  'WARN'  	 : 30,
+  'INFO'	   : 20,
+  'DEBUG'	   : 10
+}
+
+logger = logging.getLogger('app')
+
+app = Flask(__name__)
+app.config['TEMPLATES_AUTO_RELOAD'] = True
+
+def read_config():
+  conf = configparser.ConfigParser()
+  with open('config/config.ini') as f:
+    conf.read_file(f)
+  return conf
+
+def read_vault_token():
+  token = getenv('VAULT_TOKEN')
+  return token.strip()
+
+@app.route("/health", methods=['GET'])
+def health():
+    global dbc
+    if not dbc.is_initialized:
+      return "Unhealthy - no database", 500
+
+    return "Healthy", 200
+
+@app.route('/customers', methods=['GET'])
+def get_customers():
+    global dbc
+    customers = dbc.get_customer_records()
+    logger.debug('Customers: {}'.format(customers))
+    return json.dumps(customers)
+
+@app.route('/customer', methods=['GET'])
+def get_customer():
+    global dbc
+    cust_no = request.args.get('cust_no')
+    if not cust_no:
+      return '<html><body>Error: cust_no is a required argument for the customer endpoint.</body></html>', 500
+    record = dbc.get_customer_record(cust_no)
+    #logger.debug('Request: {}'.format(request))
+    return json.dumps(record)
+
+@app.route('/customers', methods=['POST'])
+def create_customer():
+    global dbc
+    logging.debug("Form Data: {}".format(dict(request.form)))
+    customer = {k:v for (k,v) in dict(request.form).items()}
+    for k,v in customer.items():
+      if type(v) is list:
+        customer[k] = v[0]
+    logging.debug('Customer: {}'.format(customer))
+    if 'create_date' not in customer.keys():
+      customer['create_date'] = datetime.now().isoformat()
+    new_record = dbc.insert_customer_record(customer)
+    logging.debug('New Record: {}'.format(new_record))
+    return json.dumps(new_record)
+
+@app.route('/customers', methods=['PUT'])
+def update_customer():
+    global dbc
+    logging.debug('Form Data: {}'.format(dict(request.form)))
+    customer = {k:v for (k,v) in dict(request.form).items()}
+    logging.debug('Customer: {}'.format(customer))
+    new_record = dbc.update_customer_record(customer)
+    logging.debug('New Record: {}'.format(new_record))
+    return json.dumps(new_record)
+
+@app.route('/', methods=['GET'])
+def index():
+    return render_template('index.html')
+
+@app.route('/records', methods=['GET'])
+def records():
+    records = json.loads(get_customers())
+    return render_template('records.html', results = records)
+
+@app.route('/dbview', methods=['GET'])
+def dbview():
+    global dbc
+    records = dbc.get_customer_records(raw = True)
+    return render_template('dbview.html', results = records)
+
+@app.route('/add', methods=['GET'])
+def add():
+    return render_template('add.html')
+
+@app.route('/add', methods=['POST'])
+def add_submit():
+    records = create_customer()
+    return render_template('records.html', results = json.loads(records), record_added = True)
+
+@app.route('/update', methods=['GET'])
+def update():
+    return render_template('update.html')
+
+@app.route('/update', methods=['POST'])
+def update_submit():
+    records = update_customer()
+    return render_template('records.html', results = json.loads(records), record_updated = True)
+
+if __name__ == '__main__':
+  logger.warn('In Main...')
+  conf = read_config()
+
+
+  logging.basicConfig(
+    level=log_level[conf['DEFAULT']['LogLevel']],
+    format='%(asctime)s - %(levelname)8s - %(name)9s - %(funcName)15s - %(message)s'
+  )
+
+  try:
+    dbc = db_client.DbClient()
+
+    if conf.has_section('VAULT'):
+      if conf['VAULT']['Enabled'].lower() == 'true':
+        logger.info('Vault is enabled...')
+        # if conf['VAULT']['Transform'].lower() == 'true':
+        #   logger.info('Using Transform database client...')
+        #   try:
+        #     dbc = db_client_transform.DbClient()
+        #   except Exception as e:
+        #     logging.error("There was an error starting the server: {}".format(e))
+        vault_token = ""
+        if conf['VAULT']['InjectToken'].lower() == 'true':
+          logger.info('Using Injected vault token')
+          vault_token = read_vault_token()
+        else:
+          vault_token = conf['VAULT']['Token']  
+
+        # transform_path=conf['VAULT']['TransformPath'], ssn_role=conf['VAULT']['SSNRole'], transform_masking_path=conf['VAULT']['TransformMaskingPath'], ccn_role=conf['VAULT']['CCNRole']
+        dbc.init_vault(addr=conf['VAULT']['Address'], token=vault_token, namespace=conf['VAULT']['Namespace'], path=conf['VAULT']['KeyPath'], key_name=conf['VAULT']['KeyName'])
+
+      if dbc.is_initialized is False: # we didn't use dynamic credentials
+        logger.info('Using DB credentials from config.ini...')
+        dbc.init_db(
+          uri=conf['DATABASE']['Address'],
+          prt=conf['DATABASE']['Port'],
+          uname=conf['DATABASE']['User'],
+          pw=conf['DATABASE']['Password'],
+          db=conf['DATABASE']['Database']
+        )
+    appPort = conf["DEFAULT"]["port"]
+    logger.info('Starting Flask server on {} listening on port {}'.format('0.0.0.0', appPort))
+    app.run(host='0.0.0.0', port=appPort)
+
+  except Exception as e:
+    logging.error("There was an error starting the server: {}".format(e))
diff --git a/config/config.ini b/config/config.ini
@@ -0,0 +1,11 @@
+    [DEFAULT]
+    LogLevel = DEBUG
+    Port = 8080
+    [DATABASE]
+    Address = mysql
+    Port = 3306
+    Database = my_app
+    User = root
+    Password = root
+    [VAULT]
+    Enabled = False