slococo-uni
/
getendpoints
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add option to recreate the database

This commit is contained in:
Santiago Lo Coco 2024-12-09 15:52:10 +01:00
parent bacfc211e0
commit 7beee91d33
3 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/.env.dev
View File

@ -4,4 +4,5 @@ SERVICE_TYPE=_http._tcp.local.
SERVICE_IP=192.168.137.1 SERVICE_IP=192.168.137.1
SERVICE_PORT=5000 SERVICE_PORT=5000
DEFAULT_ENDPOINTS=false DEFAULT_ENDPOINTS=false
LOG_LEVEL=WARN LOG_LEVEL=WARN
RECREATE_DB=false

1
src/config.py
View File

@ -11,3 +11,4 @@ SERVICE_IP = os.getenv("SERVICE_IP")
SERVICE_PORT = os.getenv("SERVICE_PORT") SERVICE_PORT = os.getenv("SERVICE_PORT")
DEFAULT_ENDPOINTS = os.getenv("DEFAULT_ENDPOINTS") DEFAULT_ENDPOINTS = os.getenv("DEFAULT_ENDPOINTS")
LOG_LEVEL = os.getenv("LOG_LEVEL") LOG_LEVEL = os.getenv("LOG_LEVEL")
RECREATE_DB = os.getenv("RECREATE_DB")

4
src/db.py
View File

@ -2,7 +2,7 @@ import logging
from flask_sqlalchemy import SQLAlchemy from flask_sqlalchemy import SQLAlchemy
from config import DEFAULT_ENDPOINTS, SERVICE_IP from config import DEFAULT_ENDPOINTS, RECREATE_DB, SERVICE_IP
db = SQLAlchemy() db = SQLAlchemy()
@ -15,6 +15,8 @@ class Endpoint(db.Model):
def init_db(app): def init_db(app):
db.init_app(app) db.init_app(app)
with app.app_context(): with app.app_context():
if RECREATE_DB:
db.drop_all()
db.create_all() db.create_all()
if DEFAULT_ENDPOINTS: if DEFAULT_ENDPOINTS:
init_default_endpoints() init_default_endpoints()