from services.DatabaseService import DatabaseService
from services.BadgeGeneratorService import BadgeGeneratorService


def main():
    db = DatabaseService("config.toml")
    db.init_db()

    user_id = input("Einzelticket ID:")
    if user_id == "":
        badges = db.get_user_badges()
    else:
        badges = db.get_user_badges(int(user_id.strip()))

    badge_generator = BadgeGeneratorService("config.toml")

    for user_badge in badges:
        badge_generator.generate_badge(
            user_name=user_badge["user_name"],
            seat_id=user_badge["seat_id"],
            picture=user_badge["picture"]
        )


if __name__ == "__main__":
    main()