__init__.py 726 B

123456789101112131415161718192021222324252627
  1. from extensions.ext_database import db
  2. from models.model import EndUser
  3. def create_or_update_end_user_for_user_id(app_model, user_id):
  4. """
  5. Create or update session terminal based on user ID.
  6. """
  7. end_user = db.session.query(EndUser) \
  8. .filter(
  9. EndUser.tenant_id == app_model.tenant_id,
  10. EndUser.session_id == user_id,
  11. EndUser.type == 'service_api'
  12. ).first()
  13. if end_user is None:
  14. end_user = EndUser(
  15. tenant_id=app_model.tenant_id,
  16. app_id=app_model.id,
  17. type='service_api',
  18. is_anonymous=True,
  19. session_id=user_id
  20. )
  21. db.session.add(end_user)
  22. db.session.commit()
  23. return end_user