mirror of
https://github.com/mealie-recipes/mealie.git
synced 2025-11-01 02:33:22 -04:00
1b0de02b71
* add json editor to theme editor * add toolbars tools to recipe sections * fix recipe yield * add updated_date to recipe schema * update time cards * fix mobile buttons * fix asset URL * fix PG errors CRUD * remove -d from docker-pro * fix theme tests * remvoe old typing * abstract count function Co-authored-by: hay-kot <hay-kot@pm.me>
83 lines
2.2 KiB
Python
83 lines
2.2 KiB
Python
from mealie.core import root_logger
|
|
from mealie.core.config import settings
|
|
from mealie.core.security import get_password_hash
|
|
from mealie.db.database import db
|
|
from mealie.db.db_setup import create_session
|
|
from mealie.schema.settings import SiteSettings
|
|
from mealie.schema.theme import SiteTheme
|
|
from mealie.services.events import create_general_event
|
|
from sqlalchemy.orm import Session
|
|
|
|
logger = root_logger.get_logger("init_db")
|
|
|
|
|
|
def init_db(db: Session = None) -> None:
|
|
if not db:
|
|
db = create_session()
|
|
|
|
default_group_init(db)
|
|
default_settings_init(db)
|
|
default_theme_init(db)
|
|
default_user_init(db)
|
|
|
|
db.close()
|
|
|
|
|
|
def default_theme_init(session: Session):
|
|
default_themes = [
|
|
SiteTheme().dict(),
|
|
{
|
|
"name": "Dark",
|
|
"colors": {
|
|
"primary": "#424242",
|
|
"accent": "#455A64",
|
|
"secondary": "#00796B",
|
|
"success": "#43A047",
|
|
"info": "#1976D2",
|
|
"warning": "#FF6F00",
|
|
"error": "#EF5350",
|
|
},
|
|
},
|
|
]
|
|
for theme in default_themes:
|
|
db.themes.create(session, theme)
|
|
|
|
|
|
def default_settings_init(session: Session):
|
|
document = db.settings.create(session, SiteSettings().dict())
|
|
logger.info(f"Created Site Settings: \n {document}")
|
|
|
|
|
|
def default_group_init(session: Session):
|
|
default_group = {"name": settings.DEFAULT_GROUP}
|
|
logger.info("Generating Default Group")
|
|
db.groups.create(session, default_group)
|
|
|
|
|
|
def default_user_init(session: Session):
|
|
default_user = {
|
|
"full_name": "Change Me",
|
|
"email": settings.DEFAULT_EMAIL,
|
|
"password": get_password_hash(settings.DEFAULT_PASSWORD),
|
|
"group": settings.DEFAULT_GROUP,
|
|
"admin": True,
|
|
}
|
|
|
|
logger.info("Generating Default User")
|
|
db.users.create(session, default_user)
|
|
|
|
|
|
def main():
|
|
session = create_session()
|
|
init_user = db.users.get(session, "1", "id")
|
|
if init_user:
|
|
print("Database Exists")
|
|
else:
|
|
print("Database Doesn't Exists, Initializing...")
|
|
init_db()
|
|
create_general_event("Initialize Database", "Initialize database with default values", session)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|