made InfluxDB optional

src/app.py

@@ -40,12 +40,14 @@ api = Api(app)
 health = HealthCheck()
 db.init_app(app)
 ma.init_app(app)
-influx_db.init_app(app)
+if Config.ENABLE_INFLUXDB:
+    influx_db.init_app(app)
 
 
 @app.before_first_request
 def init_db():
-    influx_db.database.create(Config.INFLUXDB_DATABASE)
+    if Config.ENABLE_INFLUXDB:
+        influx_db.database.create(Config.INFLUXDB_DATABASE)
     db.create_all()
 
 

src/config.py

@@ -40,6 +40,7 @@ class Config:
 
     STORAGE_HOSTNAME = os.getenv("INPUT_STORAGE_HOSTNAME", "localhost:8042")
 
+    ENABLE_INFLUXDB = os.environ.get("INPUT_ENABLE_INFLUX", "true").lower() in ["true", "yes", "1"]
     INFLUXDB_HOST = os.getenv("INFLUX_HOST", "input-influx")
     INFLUXDB_PORT = os.getenv("INFLUX_PORT", "8086")
     INFLUXDB_USER = os.getenv("INFLUX_USERNAME", "input-service")

src/resources.py

@@ -118,20 +118,21 @@ class SampleResource(Resource):
             current_app.logger.exception(e)
             return abort(569, "AMPQ Publish error")
 
-        influx_db.write_points(
-            [
-                {
-                    'time': datetime.now(tz=tzlocal.get_localzone()),
-                    'measurement': 'cloudinput',
-                    'tags': {
-                        'device': desc['device_id']
-                    },
-                    'fields': {
-                        'bruh': 1.0
+        if current_app.config['ENABLE_INFLUXDB']:
+            influx_db.write_points(
+                [
+                    {
+                        'time': datetime.now(tz=tzlocal.get_localzone()),
+                        'measurement': 'cloudinput',
+                        'tags': {
+                            'device': desc['device_id']
+                        },
+                        'fields': {
+                            'bruh': 1.0
+                        }
                     }
-                }
-            ]
-        )
+                ]
+            )
 
         db.session.commit()
         return {"tag": generated_tag}, 200