o ̿Si&@sddlZddlZddlZddlmZmZzddlZeeds)GdddZee_ Wn e y3Ynwddl m Z ddl mZmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlZee dgddZ eddZ!dZ"dZ#dZ$dZ%Gdddej&Z'ddZ(de)de)fdd Z*de)d!ede)fd"d#Z+de)fd$d%Z,d&e)d'e)de)fd(d)Z-de)fd*d+Z.dS),N)datetime timedelta __about__c@seZdZeeddZdS)_About __version__0N)__name__ __module__ __qualname__getattr_bcryptrr r 8/var/www/html/Trade-python/app/v1/services/saas/login.pyr sr) CryptContext) HTTPExceptionstatus)create_access_token)str_to_objectid)get_role_service) FeatureBase)verify_google_token)send_welcome_email)ObjectId) load_dotenvbcryptauto)schemes deprecatedi7)minutesusersaccounts subscriptionsprojectscseZdZfddZZS)CustomJSONEncodercs0t|tr |St|trt|St|S)N) isinstancer isoformatrstrsuperdefault)selfobj __class__r rr(.s   zCustomJSONEncoder.default)rr r r( __classcell__r r r+rr#-sr#cCs^t|trdd|DSt|trdd|DSt|tjr$|St|tr-t|S|S)NcSsi|] \}}|t|qSr serialize_data).0kvr r r 7sz"serialize_data..cSsg|]}t|qSr r.)r0itemr r r 9sz"serialize_data..)r$dictitemslistrr%rr&)r*r r rr/5s    r/ user_datareturnc sjttdd}|dd|dd|dd|d||d d|d r|t}|d t|d i}|r|dd|d<|d d|d<|dd|d<|d r}|t}|d t|d i}|r||d d|d <|d||d<n|t }| |d dd}||d <|d} | rt | |} | r| di|d<|d} t | } | |d<|S)z Enriches user data with common details: - Account information (subscription, active agents, etc.) - Role rights from roles service. ALLOWED_FREE_AGENTS5 account_typersubscription_idsubscription_statussubscription_agents active_agents account_id_idactive_subscription_idactive_agents_countsubscription_agents_countACTIVE)rCrroles permissions role_rightsfeatures role_features)intosgetenv setdefaultgetACCOUNT_COLLECTION_NAMEfind_onerSUBSCRIPTION_COLLECTION_NAMEPROJECT_COLLECTION_NAMEcount_documentsrr8find) r9dbr;accounts_collection account_infosubscription_collection subscriptionproject_collectionactive_projects_countrole_idrolefeatures_collection features_listr r renrich_user_dataAsL          rdtoken_expires_deltac sFt||IdH}td|d|ddi|d}t|}d||dS) z Common login processing: - Enrich user data with account/role details. - Generate a JWT token. - Serialize user data for the response. Nsubemailmobiler?)data expires_deltazLogin Successful)messageusertoken)rdrrRr/)r9rYrermr r rprocess_user_loginws rncsX|td|ji}|stdddt|j|ds#tdddt||t IdHS) z+ Handles the email/password login. rgizUser not found status_codedetailhashed_passwordizInvalid credentialsN) COLLECTION_NAMErTrgr pwd_contextverifypasswordrRrnaccess_token_expires) user_loginrY user_in_dbr r r login_users  rzpayload otp_storec s|d}|d}|r|durtddd||vr!tddd||}t|dkr6||=tdd d|dt|krDtdd d||=|t}|d |i}|sn|ttd d d d dd}||j} | |d<n| d|diddtii|}t ||t IdHS)a Verifies the OTP for a phone login. This function handles: - Validating the OTP (using an in-memory store for demo purposes) - Looking up (or creating) the user by phone - Processing the login (enriching user data and generating JWT token) phoneotpNiz"Phone number and OTP are required.roz#OTP not sent for this phone number.expiresz OTP expired.z Invalid OTP.rhTr?)rh created_date last_login is_active is_verifiedmobile_verifiedrCr`rD$setr) rRrrutcnowrNrsrT insert_one inserted_id update_onernrw) r{rYr|r} otp_submittedstoredusers_collectionryr9rr r rverify_phone_otp_servicesD         rc s6t|j}|sttjdd|t}|d|di}|sx|d|d|dd|dt t t ddd d }|d }|d|di}|rZ|d d|d <|d d|d<| |j } | |d<| t |d|dt||tIdHSt |dd} |d|did| i|| t||tIdHS)z' Handles Google Connect login. zInvalid Google token.rorgnamepicturer? refresh_tokenTF) rgrpicurlgoogle_connectrr date_of_birthrrremailsrCra r`rDN)rrr)rrmrrHTTP_401_UNAUTHORIZEDrsrTrRrrrradd_taskrrnrwrupdate) google_tokenbackground_tasksrY user_inforryr9emails_collection invitationr update_datar r rgoogle_connect_users>     r)/jsonrOrandomrrrr hasattrrr Exceptionpasslib.contextrfastapirrapp.v1.dependencies.authrapp.v1.libraries.objectrapp.v1.services.saas.rolesrapp.v1.models.crudmodelrapp.v1.libraries.googleconnectr app.v1.libraries.email_templatesrbsonrdotenvrrtrwrsrSrUrV JSONEncoderr#r/r6rdrnrzrrr r r rsL             6 :