o
g)gB
@ sz d dl Z d dlZd dlmZmZmZmZmZmZ d dl m
Z
m Z m Z d dl
mZ d dlmZ d dlmZ d dlmZmZmZmZ d dlmZ d dlm Z d dl m Z d dl T d d l m!Z! e Z"e Z#e! e $d
Z%e $d Z&e'e $d d
Z(e'e $dd
Z)e $dZ*dZ+e"j,dej-dedeej.fde'defddZ/e"j,dej-deej.fdej0defddZ1e"j,d ej-dee#eej.fdej0d e defd d Z2e"3d ee#eej.fd e defd!d"Z4e"5d#ee#eej.fd$e'd e defd%d&Z6e",d'ee#eej.fd(ejd e defd)d(Z7e",d*ededee#eej.fd+e'de'd e defd,d-Z8dS ). N) APIRouterDepends
HTTPExceptionstatusRequestForm)OAuth2PasswordRequestForm
HTTPBearer HTTPAuthorizationCredentials) JSONResponse)Session)UserFolder ChatHistory SharedFolder) timedelta)*) load_dotenvDEFAULT_ALLOCATED_STORAGEDEFAULT_FOLDER_LIMITDB_PATH/ STORAGE_PATHDEFAULT_FOLDERz0^[a-zA-Z0-9_.+-]+@[a-zA-Z0-9-]+\.[a-zA-Z0-9-.]+$z /retrieve)response_model.emaildbc
s zR| s t dW S | } tt| s ttjtjdddW S |t
t j | k }|s3t dW S d}d|j
d|j d }t| ||I d H ttjd
d ddW S tyj } z td d
t| dd }~ww )NEmail is required.Email: Invalid value.r message status_codecontentzUnregistered Email Address.zAnswerous Access TokenDear z2,
PYour access token for Answerous API is zG
With Regards,
Answerous Team
e: support@contactous.com z9The token has been sent to your registered email address. zServer Error : r" detail)
errorResponsestriprematch EMAIL_REGEXr r HTTP_400_BAD_REQUESTqueryr
filterr firstnametoken
send_email HTTP_200_OK Exceptionr str)r r usersubject mail_contente r< +/var/www/html/answerous_api/routers/auth.py
retrieveToken s@ r> z /registerr8 c sn z | j r | j nd}| jr| j nd}| jr | j nd}| jr)| j nd}| jr3| j nd}| jr=| j nd}t|ddrJtdW S t
t |s\t t
jt
jddd W S |ttj|k }|rptd
W S t } || } t| }
td d t } t||| jr|nd| jr|nd||
| | ttd
} || | | | tj t t!t | st"t t!t | tj t t#t | st"t t#t | t$t%}
t&t%d| j'd|
d}|| | | | |dkrd}d| j d| j( d}t)|||I d H t*dt+j,-| . W S t/y6 } z t0ddt | dd }~ww )N 0000Yes 2 Name: Invalid valueEmail: Invalid valuer r! zLEmail already registered. Retrieve the token or register a new email address TOKEN_PREFIX-
r2 r org_idorg_nameplan
password_hashr3 user_codeallocated_storage folder_limitr No)r2 parent_iduser_idsharedsanitized_namez Your Answerous API Access Tokenr$ a ,
Thank you for signing up with Answerous. It is a RAG-as-a-service designed with the objective of giving you a simple set of flexible APIs with high quality output. The APIs can be tested within minutes and the documentation will help you to integrate them in your environment easily.
All users of Answerous have a unique access API token.
Please take note of your token. It is: z}
Please do not hesitate to contact us if you need any help in using Answerous.
Warm Regards,
Answerous TeamzPUser successfully created with this email address. Please take note of the tokenr& Failed to save file: r' )1r2 r* r rI rJ rK
welcome_emailvalidate_string_lengthr) r+ r, r- r r r. r/ r
r0 r1 generate_random_text
hash_passwordosgetenvuuiduuid4r r addcommitrefreshpathexistsr7 DB_FOLDERmakedirs
UPLOAD_FOLDERreplace_special_charactersr r rR r3 r4 successReponseschemas UserResponsefrom_ormdictr6 r )r8 r r2 r rI rJ rK rV db_user random_texthashed_passwordr3 new_userrT
new_folderr9 r: r; r< r< r=
register_userB s
rq z/add credentialsc s |j }t||}z| jr| j nd}| jr | j nd}| jr'| j nd}| jr1| j nd}| jr;| j nd} |sDt dW S |
t t j|k
}
t|
j t|j |
sbt dW S |
j|jkrmt dW S |j} t|ddr{t dW S tt|sttjtjd d
d W S |
t t j|k
} | rt d W S t }
t |
}td
dt }t ||| jr|nd| jr| nd||||j t t d
}| | |! |"| t#dt$j%&|' W S t(y } z t)ddt*| dd }~ww )Nr? rA zOrganization ID is requiredz Organization ID does not existz7Organization ID does not match with the organization IDrB rC rD rE r r! zEmail already registered.rF rG r@ rH z2User successfully created with this email address.r& rU r' )+rr authget_current_userr2 r* r rI rK rV r) r/ r
r0 r1 printrJ rW r+ r, r- r r r. rX rY rZ r[ r\ r] rM r r r^ r_ r` rg rh OrgUserResponserj rk r6 r r7 )r8 rr r r3 org_datar2 r rI rK rV user_orgrJ rl rm rn ro r; r< r< r= add_user s
ry z/listc
s | j }t||}z/|jr5|ttj|jktj|jk }dd |D }t ddd |D W S t
dW S t yP } z t ddt
| dd }~ww ) Nc S g | ]}t j|qS r< )rh rv rj ).0usersr< r< r=
z list_users..z Users retrieved successfullyc S g | ]}| qS r< rk )r{ r8 r< r< r= r} z)This account is not organization account.r& Something went wrong: r' )rr rs rt rI r/ r
r0 rR allrg r) r6 r r7 )rr r r3 user_datar| r; r< r< r=
list_users s$ * r z/remove/{user_email}
user_emailc
s |j }t||}z?| stdW S | }|ttj|k }|s*tdW S |j
|j
kr5tdW S | | | t
tjtjdddW S ty` } z tddt| d d }~ww )
Nr User does not exist.z*User does not belong to your organization.zUser removed successfully.r r! r& r r' )rr rs rt r) r* r/ r
r0 r r1 rI deleter_ r r r5 r6 r r7 )r rr r r3 r r r8 r; r< r< r= remove_user s<
r z/share share_folderc
s |j }t||}z| jstdW S |ttj| jktj|jk }|s/tdW S | j
s7tdW S | j
}| j }| j
rI| j
nd}tt|s]ttjtjdddW S |dkr|ttj| jktj
|k } | r|tdW S t|| j|r|nd d
}
||
| ||
td tj|
W S |d kr|ttj| jktj
|k } | r| | | ttj tj d
ddW S ttj tj d
ddW S tdW S t y } z t ddt!| dd } ~ ww )NzFolder is required.zFolder ID does not exist.r Giver r r! z4The folder is already shared with the email address.r? )r folder_id organizationz0The folder is now shared with the email address.Removez9The folder is no longer shareable with the email address.z;Invalid access value. The value must be 'Give' or 'Remove'.r& r r' )"rr rs rt r r) r/ r r0 rR r1 r r* r accessr+ r, r- r r r. r r^ r_ r` rg rh SharedFolderResponserj rk r r5 r6 r r7 )
r rr r r3 r8 folder_existr r access_typeis_existrS
shared_folderr; r< r< r= r + s *
(
(
z/historydoc_idc
s |j }t||}|r| nd}zV| stdW S |s!tdW S |ttj|ktj |j k
}|s.zNo any document found.z History retrieved successfullyc S r r< r )r{ chatr< r< r= r} r r& r r' )rr rs rt r* r) r/ r
r0 r rI r1 r rR r r rg r6 r r7 )
r r rr r r3 r8 chat_user chat_history chat_datar; r< r< r= r y s@ (( r )9rZ r\ fastapir r r r r r fastapi.securityr r r
fastapi.responsesr sqlalchemy.ormr db_config.databasedatabaser db_config.modelsr
r r r db_config.schemasrh db_config.authrs datetimer dependencies.helperdotenvr routersecurityr[ r r r7 rc re r r- postri get_dbr>
UserCreaterq ry getr r r r r r< r< r< r= s`
$' P*d$( *M