mirror of
https://github.com/langgenius/dify.git
synced 2024-11-16 19:59:50 +08:00
28 lines
726 B
Python
28 lines
726 B
Python
|
from extensions.ext_database import db
|
||
|
|
from models.model import EndUser
|
||
|
|
|
||
|
|
|
||
|
|
def create_or_update_end_user_for_user_id(app_model, user_id):
|
||
|
|
"""
|
||
|
|
Create or update session terminal based on user ID.
|
||
|
|
"""
|
||
|
|
end_user = db.session.query(EndUser) \
|
||
|
|
.filter(
|
||
|
|
EndUser.tenant_id == app_model.tenant_id,
|
||
|
|
EndUser.session_id == user_id,
|
||
|
|
EndUser.type == 'service_api'
|
||
|
|
).first()
|
||
|
|
|
||
|
|
if end_user is None:
|
||
|
|
end_user = EndUser(
|
||
|
|
tenant_id=app_model.tenant_id,
|
||
|
|
app_id=app_model.id,
|
||
|
|
type='service_api',
|
||
|
|
is_anonymous=True,
|
||
|
|
session_id=user_id
|
||
|
|
)
|
||
|
|
db.session.add(end_user)
|
||
|
|
db.session.commit()
|
||
|
|
|
||
|
|
return end_user
|