)iW ddlmZmZmZmZmZmZmZmZddl m Z m Z ddl m Z ddlmZmZmZmZmZddlmZddlmZddlZddlZddlZddlmZdd lmZdd lmZm Z m!Z!dd l"m#Z#ed d gZ$eejJZ&e&ejNZ(e(dZ)Gdde Z*Gdde Z+Gdde Z,Gdde Z-Gdde Z.e$j_ddde+fdZ0e$jcddZ2e$jcd d!e3fd"Z4e$jkd d!e3de.fd#Z6e$jod d!e3fd$Z8e$j_d%dde-fd&Z9e$jcd%d'Z:e$jcd(d!e3fd)Z;e$jkd(d!e3de-fd*Ze$jcd.d/e3fd0Z?Gd1d2e Z@Gd3d4e ZAe$j_d5dde@fd6ZBe$jcd5d7ZCe$jkd8d!e3deAfd9ZDe$jod8d!e3fd:ZEdd;lFmGZGGd<d=e ZHe$jcd>d?ZIe$jcd@d!e3fdAZJe$j_d>dBefdCZKe$jkd@d!e3dBefdDZLe$jdEd!e3dBefdFZNe$jod@d!e3fdGZOGdHdIe ZPe$j_dJddePfdKZQe$j_dLedMedfdBedNe3dOee3fdPZRe$jcdQdcdReSdSeSfdTZTe$jcdUdcdOe3dReSdSeSfdVZUe$jodWdOe3fdXZVe$j_dYedfdBedZeefd[ZWGd\d]e ZXe$jod^d_e3deXfd`ZYe$jcdad_e3fdbZZy)d) APIRouter HTTPExceptionDependsstatus UploadFileFileRequestForm) JSONResponse FileResponse) BaseModel)OptionalListDictAnyUnion) MongoClient)ObjectIdN)datetime)settings)verify_passwordcreate_access_tokenget_password_hash)get_assistant_servicez/adminAdmin)prefixtagsadminsc"eZdZUeed<eed<y) LoginRequestemailpasswordN__name__ __module__ __qualname__str__annotations__@/var/www/html/Whatsapp_Customer_Support_Chatbot/src/api/admin.pyr r s JMr*r c6eZdZUeed<eed<dZeeed<y)PermissionBasenamecodeN description)r$r%r&r'r(r0rr)r*r+r-r-s I I!%K#%r*r-cdeZdZUdZeeed<eed<dZeed<dZ eed<dZ eed<dZ eed<y) RolePermissionN module_id module_nameFviewaddeditdelete) r$r%r&r3rr'r(r5boolr6r7r8r)r*r+r2r2!s>#Ix}#D$CD$FDr*r2cJeZdZUeed<dZeeed<dZeed<e e ed<y)RoleBaseroleNr0F is_delete permission) r$r%r&r'r(r0rr=r9rr2r)r*r+r;r;)s* I!%K#%It^$$r*r;cJeZdZUdZeeed<dZeeed<dZeeed<y)PermissionUpdateNr.r/r0) r$r%r&r.rr'r(r/r0r)r*r+r@r@/s,D(3-D(3-!%K#%r*r@z /permissions) status_codedatacK|j}tdj|}t|j|j d|j d|j dd}ddd|d iSw) N permissionsr.r/r0idr.r/r0rCrAz Permission created successfully.)rmessagepermission_details)dictdb insert_oner' inserted_idget)rCdocresult response_datas r+create_permissionrR6s| ))+C   ) )# .F&$$%ww}- M 9"/  A8A:c Kg}tdjD]Q}|jt|d|j d|j d|j ddSddd |d iSw) NrE_idr.r/r0rFrCz#Permissions retrieved successfully.)rrHrE)rKfindappendr'rN)permsps r+get_all_permissionsr[Ks E   # # %  ah-EE&MEE&M55/     <   sA6A8z/permissions/{id}rGcKtdjdt|i}|s tddt |d|j d|j d|j dd }d d d |d iSw)NrErUPermission not foundrBdetailr.r/r0rFrCrVz"Permission retrieved successfully.rrHr>rKfind_onerrr'rN)rGperm perm_datas r+get_permission_by_idrf^s m  % %uhrl&; r3rEr/r4rUrolesr<r0r=Fr5r6r7r8r3r4r5r6r7r8rGr<r0r=r>rCrAzRole created successfully.rrHr<)rJrNrKrcr'rLrM)rCrOrZperm_docrPrQs r+ create_roler~s^ ))+C  uu[!-(1161];K2LMH!$Xe_!5+ [ # #C (F&$$%ww}-WW[%0WW\2.  !{B!78 uu]3fe,uuUE*fe,%%%0    M& 3!    s*EB$EA8E EcKg}tdjD]}t|d|jd|jdd|jdgDcgc]s}t|jdd|jd |jd d|jd d|jd d|jd dduc}d}|j |ddd|diScc}ww)NrxrUr<r=Fr>r3ryr4r5r6r7r8rzrGr<r=r>rCrVzRoles retrieved successfully.)rrHrx)rKrWr'rNrX)rxrrZ role_datas r+ get_all_rolesrs E [    ah-EE&M{E2|R0 "%QUU;%;!<#$55#7EE&%055.EE&%0eeHe4     Y# ( 6   sAC>!A8C9 %C>z /roles/{id}cKtdjdt|i}|s tddt |d|j d|j dd|j d gDcgc]s}t |j d d |j d |j d d|j dd|j dd|j ddduc}d}ddd|diScc}ww)NrxrUr]Role not foundr_r<r=Fr>r3ryr4r5r6r7r8rzrrCrVzRole retrieved successfully.r|rb)rGrrZrole_formatteds r+get_role_by_idrs 7 eXb\23A 4DEE!E(mf UU;.UU<,  !{B!78 uu]3fe,uuUE*fe,%%%0    N" 5"    sA5D7A8C< /DcK|j}|dD]E}|jdrtdjd|di}|s5t |d|d<Gtdj dt |id|itdjdt |i}t |d|jd |jd |jd d |jdgDcgc]s}t |jdd |jd|jdd |jdd |jdd |jdd duc}d}ddd|diScc}ww)Nr>r3rEr/r4rUrxrhr<r0r=Fryr5r6r7r8rzr{rCrVzRole updated successfully.r|)rJrNrKrcr'rjr)rGrCrOrZr}rrs r+ edit_rolers ))+C 6uu[!-(1161];K2LMH!$Xe_!5+ 6 wKE8B<063-@ 7 eXb\23A!E(mf uu]+UU;.UU<,  !{B!78 uu]3fe,uuUE*fe,%%%0    N& 3"    s*FF B.F;A8F 3FcKtdjdt|i}|jdk(r t ddddd dd iSw) NrxrUrr]rr_rCrVzRole deleted successfully.rprqrts r+ delete_roler)sZ [ # #UHRL$9 :F q 4DEE3  rvz/logincKtjdd|jid|jigi}|stddddddiSt |j |d std dd d ddiSd}|j d rtd jdt|d i}|rt|d|j d|j dd|j dgDcgc]s}t|j dd|j d|j dd|j dd|j dd|j ddduc}d}td|j dxs|j di}ddd|d t|d|j d|d|j d!|d"d#iScc}ww)$N$orr!usernamer]rCzaccount not found.rrHuserrBcontentr"zInvalid credentials.role_idrxrUr<r=Fr>r3ryr4r5r6r7r8rzrsub)rCrVzLogin successful.bearer full_name)rGr!rrr<)rrH access_token token_typer) admins_collectionrcr!r rr"rNrKrr'r)rCadmin role_inforole_docrZrs r+loginr8s  & & djj !  $ ( E !3    4==% *; <!5   I yyg;''y9I0J(KL (5/* V,%\\+u=&\\,; &){B)?%@'(uu]'; !fe 4 uuUE2 !fe 4"#%%%"8   I"'UEIIg4F4_%))T^J_,`aL*("%,'7+!*-"YY{3!     sC0G!2A8G *A7G!z/user/{user_id}user_idc fK tjdt|i}|s tdddddt |d|j d|d |j d |j d rt |j d ndd iS#t $rtd dwxYww)NrUr]User not foundr_rCrVz$User details retrieved successfully.r!rrr)rrHrGr!rrrrzInvalid User ID format)rrcrrr'rN Exception)rrs r+get_userrsN!**E8G3D+EFC8HI I A%,'7+!*-;/8= )8L3uyy34RV   N4LMMNsB1BBB1B..B1cJeZdZUeed<eed<eed<eed<dZeeed<y) AdminCreater!rr"rNr)r$r%r&r'r(rrr)r*r+rrs$ JMMN!GXc]!r*rcreZdZUdZeeed<dZeeed<dZeeed<dZ eeed<dZ eeed<y) AdminUpdateNr!rr"rr) r$r%r&r!rr'r(rr"rrr)r*r+rrsHE8C="Hhsm""Hhsm"#Ix}#!GXc]!r*rz/usersc >Ktjdd|jid|jigirt ddddddiS|j }t |j|d <d}|jd r@ td jd t|d i}|r|jd }nd|d <tj|}dddt|j|d|d|d|jd |ddiS#d|d <YXxYww)Nrr!rrrCz!Email or Username already exists.rrr"rrxrUr<rAz Admin user created successfully.rrGr!rrr role_name)rrcr!rr rJrr"rNrKrrLr'rM)rC admin_docrr<rPs r+ create_adminrsF!!5GTZZ+@:t}}B]*^"_`!B    I-dmmjA|d&|jt||tjd)t<j>jA|d*}tdj!|}td+|j"d,} tE}|r\td-|jFjI||.tdjKd|j"id/d0d,iitd1| j3t|j"|j|d4 tdjKdt|id5t | id6tjid7td8td9t | d:td;t |d:td<|td=d>d?t | d@||| |dAiS7#1swYxYw#tB$r:}td'||j3|jd(|d"Yd}~ d}~wwxYw#tB$rD}td2|d3}tdjKd|j"id/d0d3iiYd}~d}~wwxYww)BzUpload multiple files to folderz === FOLDER UPLOAD REQUEST ===z folder_name: z folder_id: Nz Form keys: filesfilezFiles received: rFile r: filenamez NO FILENAMEz ERROR: No files found in requestz3No files provided. Use 'files' or 'file' field namer_rrUrr]Folder not foundrTrz Processing z files...z --- Processing: z ---.pdfzREJECTED: Not a PDF fileOnly PDF files allowedrreasonz Saving to: wbzFile saved successfully, size: z byteszERROR saving file: Save failed: Indexing)rrr upload_datersizezDB record created: IndexedzIndexing file...rrhrzFile indexed successfullyzERROR indexing: Failed)file_idrrrr)$incrhz === UPLOAD COMPLETE ===z Uploaded: z fileszFailed: z Folder ID: z======================== rCrAz file(s) uploaded to folder)rrHrruploaded_files failed_files)&printformlistkeysgetlistlen enumeratehasattrrrrKrcrr'rrrLrMrrrrrlowerendswithrXopenshutil copyfileobjrospathgetsizerrmanagerindex_single_filerj)rrrrridxffinal_folder_idfinal_folder_namer new_folderrPdata_dirr r r file_pathbufferrfile_doc file_result status_msgrs r+upload_file_to_folderr*sV +- M+ '( K { #$ D KTYY[)* +, LL ! 9T\\&%9E 5SZa8 9: & ^FC E#a%J1G1::]"[\ ] ^  014ijjO#&'00%)9L1MN 'O &} 5 C8JK K&'00-1MN !&-0O &} 5  +&oo/&oo/ J *+66zBF!&"4"45O||H--.@H NN4$N/NL M#e*Y /05  "4==/67}}""$--f5 , .   T]]F^ _ ` t}},  I;'( i& 6&""499f5 6 3BGGOOI4N3OvV W !/2,#??, GGOOI.  k,,X6  #K$;$;#<=>  +-G(*11)IZ1[7 &&K334h 2313 ;223    c5 p%% )*!3~#67!8??#45  %' Js>*+6 23 HS&'v ./ K( )* &( n-..IJ(,,(    m v 6 6  's+ ,   T]] VWUXFY Z [  6  $QC( )!J wK " " //0(H-.   s;VSI/V. S+:!S4S+A=V A(T15C'VS( #S++ T.4/T)#V)T..V1 U>:9U93V9U>>Vz/folderspagelimitc K|dz |z}tdji}g}tdjjddj |j |D]>}|j t|d|d|d|d|jdd d @d d ||||||zdz |zd ddSw)z,Get all folders with file counts (Paginated)rrrrUrrrrrrrrrrVzFolders retrieved successfullytotalr+r,pages)folders paginationr) rKcount_documentsrWrskipr,rXr'rN)r+r,r6 total_foldersr3r s r+ list_foldersr8s 1H D)*::2>MG " # ( ( * / / b A F Ft L R RSX Y QuX]+L/L/%% a0    3&'%/!3=    sCCz/folders/{folder_id}/filesc tKtdjdt|i}|s tdd|dz |z}tdj dt|i}g}tdj dt|ij d d j|j|D]K}|jt|d|d |d |jd d |jdddMdd||d|d|d|jddd||||||zdz |zdddSw)z2Get all files in a specific folder with paginationrrUr]rr_rrrrr.rrrrr)r rrrrrVzFiles retrieved successfullyrrrrr/r0)rrr4r) rKrcrrr5rWrr6r,rXr'rN)rr+r,rr6 total_filesrr s r+get_folder_filesr;so" # , ,eXi5H-I JF 4FGG 1H DW+--{HY ,,x001I=K k"wK[(9*=>?%%uhy.A&BCk]*VW  R =k]"QCPQQRs/8D$'C="BD$= D!DD$D!!D$z /files/uploadrc\K|rt|dk(rZ|djsK|jd{}|jdxs|jd}|s t ddg}g}t j tj}|jdd |D]o}|jjjd s|j|jd d L||jz } t|d 5}tj|j |ddd|jt%j&dt(j*j-|d} t.dj1| } d} t3} | rF| j4j7|dt.dj9d| j:idddii|jt=| j:|j| drdddddiS7#1swYxYw#t"$r,} |j|jd| d Yd} ~ d} ~ wwxYw#t"$r5} d} t.dj9d| j:idddiiYd} ~ d} ~ wwxYww)Nrrrrrz6No files provided. Use 'files' field name in form-datar_Trrrrrrr)rrrrrrrUrhrr)rGrrrCrVzFiles uploaded successfully.rp)rrrrrrrrrrrrrXrrrrrrrrrrrKrLrrrrjrMr')rrrr r r$rr%r&rr'rPr)rs r+ upload_filerCs~ SZ1_U1X->->\\^# W%=f)=C8pq qNL||H--.H NN4$N/' }}""$--f5   T]]F^ _ ` t}},  i& 6&""499f5 6 #??, GGOOI.   G''1  `+-G11)1N7 &&v/A/A'BVhXaMbDcd f(()    G' T 5  g$& 6 6    T]] VWUXFY Z [  ( `!J wK " "E6+=+=#>(T\I]@^ _ _ `s~3J,H$CJ, H3!H'2H3:AJ,AI+'>J,'H0 ,H33 I(unlinkrrrjrr) r rCrFrsdeleted_files_detailsrIr'rrrrr%rs r+ delete_filesrN@sW -/ Idiik] #$#-T]]C"@ dmmHM F.F- F'{++UHW4E,FGH gYj9: +H [1I",,}5K,-G33H=#LL):):;c)nLxW #LL):):;hF !  " wK " "E8G+<#= >%&11HY/0lB/0 " ( (w<$/8S^d* *  Q M IhZ( )W.F` Im_$4S[M JK*=/A*2    F MM0 CF8D E E Fs=A'I*:H&$I%EH&:,I& I/ IIIIz/files/{file_id}/viewc K tdjdt|i}|s tdd|d}|j d}|r3t j tjt|z |z }n&t j tj|z }|jstdd|t|d | S#t$r-}t|tr|td t|d }~wwxYww) z%View/Fetch PDF file content directly.rrUr]z!File record not found in databaser_rrzPhysical file not found at zapplication/pdf)r media_typerrN)rKrcrrrNrrrrr'r>r rrJ)r r'rrr%rs r+view_pdfrQs<g;''0A(BCC8[\ \J'LL-   X%6%67#i.H8SI X%6%67(BI!C:UV_U`8ab b(  < a 'GCF;;res^^^833 $RR. (' 3 X)) * H  xL9&Y& Y%y% &y&^-..(N$  3!(  c)9!*"#  $ X3'*H*(*XH:MS:M%%8%%N} #  XDlDDL NCNN,")"")"X3'.[.(.`H@M*3*k**X} 3  49 J*Os"Z g g gDOg#ggg<$%g3gg&g>C ,9 C0%l%1%N #$Cy#DzI II}I%ITJSS: ())c)))*)V%&3'F_BFt*=w=tJ/?==~$ $()II+<I*IV #$