From 7beee91d33e1c960b9bed8310185b21d21bda10e Mon Sep 17 00:00:00 2001 From: Santiago Lo Coco Date: Mon, 9 Dec 2024 15:52:10 +0100 Subject: [PATCH] Add option to recreate the database --- src/.env.dev | 3 ++- src/config.py | 1 + src/db.py | 4 +++- 3 files changed, 6 insertions(+), 2 deletions(-) diff --git a/src/.env.dev b/src/.env.dev index c1c6739..1c141a5 100644 --- a/src/.env.dev +++ b/src/.env.dev @@ -4,4 +4,5 @@ SERVICE_TYPE=_http._tcp.local. SERVICE_IP=192.168.137.1 SERVICE_PORT=5000 DEFAULT_ENDPOINTS=false -LOG_LEVEL=WARN \ No newline at end of file +LOG_LEVEL=WARN +RECREATE_DB=false \ No newline at end of file diff --git a/src/config.py b/src/config.py index 47cc2f2..60a0303 100644 --- a/src/config.py +++ b/src/config.py @@ -11,3 +11,4 @@ SERVICE_IP = os.getenv("SERVICE_IP") SERVICE_PORT = os.getenv("SERVICE_PORT") DEFAULT_ENDPOINTS = os.getenv("DEFAULT_ENDPOINTS") LOG_LEVEL = os.getenv("LOG_LEVEL") +RECREATE_DB = os.getenv("RECREATE_DB") diff --git a/src/db.py b/src/db.py index 8337d25..ba6c734 100644 --- a/src/db.py +++ b/src/db.py @@ -2,7 +2,7 @@ import logging from flask_sqlalchemy import SQLAlchemy -from config import DEFAULT_ENDPOINTS, SERVICE_IP +from config import DEFAULT_ENDPOINTS, RECREATE_DB, SERVICE_IP db = SQLAlchemy() @@ -15,6 +15,8 @@ class Endpoint(db.Model): def init_db(app): db.init_app(app) with app.app_context(): + if RECREATE_DB: + db.drop_all() db.create_all() if DEFAULT_ENDPOINTS: init_default_endpoints()