PoLZy_Showcase/populate_gamification_tables.py

from polzybackend.models import GamificationBadgeLevel, GamificationBadgeType, GamificationBadgeDescription
from flask_sqlalchemy import SQLAlchemy
from polzybackend import create_app
from app import Config

app = create_app(Config)
db = SQLAlchemy(app)

print('Creating Gamification Badges...')
badge_levels = [
    ("Greenhorn", 0),
    ("Bronze", 10),
    ("Silver", 50),
    ("Gold", 100),
    ("Platinum", 500)
]
levels = []
next_level = None
for name, point in reversed(badge_levels):
    level = GamificationBadgeLevel(name=name, min_level=point, next_level=next_level, is_lowest=(name == 'Bronze'))
    levels.append(level)
    next_level = level

db.session.add_all(levels)

type_names = ["Antrag", "Polizze"]
#lobs = [LineOfBusiness.KFZ, LineOfBusiness.PnC, LineOfBusiness.KV, LineOfBusiness.Life, LineOfBusiness.UV]
lobs = ['Automobile']
types = []

for lob in lobs:
    for type_name in type_names:
        name = type_name + " " + lob
        types.append(GamificationBadgeType(name=name, title=name))
types.append(GamificationBadgeType(name="Login", title="Login"))
db.session.add_all(types)

db.session.add_all([
    GamificationBadgeDescription(
        level=level,
        type=type,
        description=f"Requirements to earn {level.name} Badge {type.title}",
    ) for level in levels for type in types]
)

db.session.commit()