a AWe@s*ddlmZddlmZGdddZdS))Q)Userc@sxeZdZeddZeddZeddZeddZed d Zed d Z ed dZ eddZ eddZ dS) AuthServicecCs(tfi|}||d||S)z Create a new user. :param data: A dictionary containing user data. :return: The created user document. password)r set_passwordsave)datauserr /app/app/services/auth.py create_userszAuthService.create_usercCstj|d}|S)z Get a user by their ID. :param user_id: The ID of the user. :return: The user document or None if not found. ) public_idrobjectsfirst)user_idr r r r get_user_by_idszAuthService.get_user_by_idcCs"tt|dt|dB}|S)z Get a user by their email or username. :param email_or_username: The email or username of the user. :return: The user document or None if not found. email)username)rrrr)email_or_usernamer r r r get_user_by_email_or_usernamesz)AuthService.get_user_by_email_or_usernamecCstj|d}|S)z Get a user by their email. :param email: The email of the user. :return: The user document or None if not found. rr)rr r r r get_user_by_email&szAuthService.get_user_by_emailcCstt|dt|dB}|S)z Search for users containing a keyword in their name, email, or other attributes. :param keyword: The keyword to search for. :return: A list of matching user documents. )Zusername__icontains)Zemail__icontains)rrr)keywordZusersr r r search_users_by_keyword0sz#AuthService.search_users_by_keywordcCs t|}|r||r|SdS)a  Authenticate a user using their email and password. :param email: The email of the user. :param password: The password provided by the user. :return: The authenticated user document if successful, None if authentication fails. N)rrZcheck_password)rrr r r r authenticate:s zAuthService.authenticatecCs,t|}|r$|||dSdSdS)z Change the password of a user. :param user_id: The ID of the user to update the password for. :param new_password: The new password. :return: True if the password is updated, False if the user is not found. TFNrrrrrZ new_passwordr r r r change_passwordGs   zAuthService.change_passwordcCs,t|}|r$||_||SdSdS)z Promote a user by changing their role. :param user_id: The ID of the user to promote. :param new_role: The new role for the user. :return: The updated user document with the new role, or None if not found. N)rrZrolerto_dict)rZnew_roler r r r promote_userWs  zAuthService.promote_usercCs,t|}|r$|||dSdSdS)z Reset the password of a user. :param user_id: The ID of the user to reset the password for. :param new_password: The new password. :return: True if the password is reset, False if the user is not found. TFNrrr r r reset_passwordgs   zAuthService.reset_passwordN) __name__ __module__ __qualname__ staticmethodr rrrrrrr r!r r r r rs$        rN)mongoengine.queryset.visitorrZ app.modelsrrr r r r s