Added DB Silent Fallback Feature to SQLite

Author: app-generator

api/__init__.py

@@ -3,7 +3,7 @@
 Copyright (c) 2019 - present AppSeed.us
 """
 
-import json
+import os, json
 
 from flask import Flask
 from flask_cors import CORS
@@ -22,7 +22,18 @@
 # Setup database
 @app.before_first_request
 def initialize_database():
-    db.create_all()
+    try:
+        db.create_all()
+    except Exception as e:
+
+        print('> Error: DBMS Exception: ' + str(e) )
+
+        # fallback to SQLite
+        BASE_DIR = os.path.abspath(os.path.dirname(__file__))
+        app.config['SQLALCHEMY_DATABASE_URI'] = SQLALCHEMY_DATABASE_URI = 'sqlite:///' + os.path.join(BASE_DIR, 'db.sqlite3')
+
+        print('> Fallback to SQLite ')
+        db.create_all()
 
 """
    Custom responses

api/config.py

@@ -3,20 +3,60 @@
 Copyright (c) 2019 - present AppSeed.us
 """
 
-import os
+import os, random, string
 from datetime import timedelta
 
 BASE_DIR = os.path.dirname(os.path.realpath(__file__))
 
-
 class BaseConfig():
-
-    SQLALCHEMY_DATABASE_URI        = 'sqlite:///' + os.path.join(BASE_DIR, 'apidata.db')
-    SQLALCHEMY_TRACK_MODIFICATIONS = False
 
-    SECRET_KEY           = os.getenv('SECRET_KEY'    , 'S#perS3crEt_913' )
-    JWT_SECRET_KEY       = os.getenv('JWT_SECRET_KEY', 'S#perS3crEt_JWT' )
-    GITHUB_CLIENT_ID     = os.getenv('GITHUB_CLIENT_ID')
-    GITHUB_CLIENT_SECRET = os.getenv('GITHUB_SECRET_KEY')
+    SECRET_KEY = os.getenv('SECRET_KEY', None)
+    if not SECRET_KEY:
+        SECRET_KEY = ''.join(random.choice( string.ascii_lowercase  ) for i in range( 32 ))
+
+    JWT_SECRET_KEY = os.getenv('JWT_SECRET_KEY', None)
+    if not JWT_SECRET_KEY:
+        JWT_SECRET_KEY = ''.join(random.choice( string.ascii_lowercase  ) for i in range( 32 ))
+
+    GITHUB_CLIENT_ID     = os.getenv('GITHUB_CLIENT_ID' , None)
+    GITHUB_CLIENT_SECRET = os.getenv('GITHUB_SECRET_KEY', None)
 
     JWT_ACCESS_TOKEN_EXPIRES = timedelta(hours=1)
+
+    SQLALCHEMY_TRACK_MODIFICATIONS = False
+
+    DB_ENGINE   = os.getenv('DB_ENGINE'   , None)
+    DB_USERNAME = os.getenv('DB_USERNAME' , None)
+    DB_PASS     = os.getenv('DB_PASS'     , None)
+    DB_HOST     = os.getenv('DB_HOST'     , None)
+    DB_PORT     = os.getenv('DB_PORT'     , None)
+    DB_NAME     = os.getenv('DB_NAME'     , None)
+
+    USE_SQLITE  = True 
+
+    # try to set up a Relational DBMS
+    if DB_ENGINE and DB_NAME and DB_USERNAME:
+
+        try:
+            
+            # Relational DBMS: PSQL, MySql
+            SQLALCHEMY_DATABASE_URI = '{}://{}:{}@{}:{}/{}'.format(
+                DB_ENGINE,
+                DB_USERNAME,
+                DB_PASS,
+                DB_HOST,
+                DB_PORT,
+                DB_NAME
+            ) 
+
+            USE_SQLITE  = False
+
+        except Exception as e:
+
+            print('> Error: DBMS Exception: ' + str(e) )
+            print('> Fallback to SQLite ')    
+
+    if USE_SQLITE:
+
+        # This will create a file in <app> FOLDER
+        SQLALCHEMY_DATABASE_URI = 'sqlite:///' + os.path.join(BASE_DIR, 'db.sqlite3')