g5 XddlmZddlmZddlmZddlmZddlmZm Z m Z ddl m Z m Z mZmZddlmZddlmZd ed efd Zd ed efd Zd ed efdZd ede d efdZd eded edefdZd ed efdZd ededed efdZd eded edefdZd eded efdZd ededed edef dZd eded efdZ y))Session)IntegrityError) HTTPException)datetime)StoreBranch StoreType) StoreCreate StoreResponse BranchCreateBranchResponse)get_current_user)Userdb current_userc|jtjtj|jk(j S)z. Check if a user already has a store. queryrfilteruser_idfirstrrs 7/var/www/html/hubwallet-dev/src/apps/stores/services.pyget_store_by_userr5 88E? ! !%--<3G3G"G H N N PPc|jtjtj|jk(j S)z. Check if a store exists by store_id. rrs rget_store_by_idrrr store_namec|jtjtj|k(j S)z" Get a store by its name. )rrrnamer)rrs rget_store_by_namer"s/ 88E? ! !%** ": ; A A CCrstorecL|jtjtj|jk(j }|st dddddt |j|j|jd}|j||j|j||jd}t|j|j|j|j |j" }|j||j|j||j|j|j|j$|j|j&j)|j*j)|j,|j|j|j |j"|j|j&j)|j*j)d gd S) NFzInvalid store type selected.statuscodemessage status_codedetail)r storetype_idr! branch_limitr)store_idr branch_nameaddress contact_info) branch_idr0r1r2r3r created_at updated_at)r0r!r.r/rr5r6branches)rr rr.rrrrr!addcommitrefreshr7rr0r1r2r3r/r5 isoformatr6r4)rr#r store_type new_store first_branch new_branchs r create_storer@"s)$++I,B,BeFXFX,XY__aJ #S=[\  $$'' ZZ I FF9IIKJJy>>!$L##$$ ,,$$!.. JFF:IIKJJz&&!..!..''**446**446(11&//)55%-- * 7 7'//(33==?(33==?    rnew_store_namereturnc t||}|stddddd|jtj tj |k(tj |j k7jrtddddd||_tj|_ |j|j||jtj tj |j k(j!}|Dcgc]R}|j"|j |j$|j&|j(|j*|jd T}}d d d |j |j |j,|j*|j|d gdS#t$r"|jtdddddwxYwcc}w)z Update the store's name and updated_at timestamp based on the current user. Ensures store name uniqueness before updating. FzStore not foundr&r*r%zStore name must be unique.z7Failed to update store name due to database constraint.)r4r0r1r2r3r5r6TzStore name updated successfully)r0r!rr5r6r7r'r(r)data)rrrrrr!r0rrutcnowr6r9r:rrollbackrallr4r1r2r3r5r)rrArr#r7branch branch_lists rupdate_store_namerM]s b, /E #S=NO   xxejjN:ENNenn<\]cce#S=YZ   EJ(E   5xx&&v%..'HIMMOH!  ))!--~~"// ++ ++  K 4"NN  ==#..#..'     /   #S=vw    s1!F=#AG+=+G(c Jt|tstddddd|jtj tj |j k(j}|stddddd |jtj tj|jk(j|j||jdd d dS#t$r"}td dd d t|dd }~wwxYw)z Delete a store and all its associated branches. The store ID is automatically retrieved based on the authenticated user. r%FzInvalid user objectr&r*rDzStore not found or unauthorizedTrEz6Store and all associated branches deleted successfullyz,An error occurred while deleting the store: N) isinstancerrrrrrrrr0deleter9 Exceptionstr)rrr#es r delete_storerUs) lD )uc]r4stt HHUO " "5==L4H4H#H I O O QE uc]~4A A X 5>> ABIIK % SU U XucaMNQRSNTMU_V5WX XXsA3C77 D"DD"r0r4c|jtjtj|k(tj|jk(j }|s t dd|jtjtj|k(tj|k(j }|s t dd|S)zq Retrieve a branch using store_id and branch_id. Ensures the branch belongs to the current user's store. rDz Store not found or unauthorized.r*Branch not found.) rrrr0rrrrr4)rr0r4rr#rKs rget_branch_by_idrXs HHUO " "5>>X#=u}}P\PdPd?d e k k mE 4VWW XXf  $ $V__%@&BRBRV_B_ ` f f hF 4GHH MrrKc |jtjtj|jk(j }|st ddddd|jt jt j|jk(t j|jk(j }|rt dddddt |j|j|j|j|jtjtj} |j||j|j|d d d |j$|j|j|j|j|j&|j(|jdgdS#t $r"|j#t d dd d dwxYw)zH Create a new branch for the authenticated user using store_id. rDFStore not found.r&r*r%,Branch name must be unique within the store.)r0r1r2r3rr5r6rOz,An error occurred while creating the branch.TrEzBranch created successfullyr4r0r1r2r3r5r6rrF)rrrrrrrr0r1r2r3rrHr8r9r:rRrIr4r5r6)rrKrr#existing_branchr?s r create_branchr^s HHUO " "5==L4H4H#H I O O QE #S=OP  hhv&--5>>)f000 eg#S=kl  &&(($$??$??$J   z  :0(11&//)55%-- * 7 7(33(33%--       #S=kl   s 2G+Hc n|jtjtj|k(tj|jk(j }|s t dd|jtjtj|k(j}|s t dd|Dcgc]e}t|j|j|j|j|j|j|j|jgc}Scc}w)zc Retrieve all branches of a given store_id if the store belongs to the authenticated user. rDz/Store not found or does not belong to the user.r*z!No branches found for this store.r\)rrrr0rrrrrJr r4r1r2r3r5r6)rr0rr#r7rKs rget_branches_by_store_idr`s HHUO " " (" ---  eg 4effxx&&v('BCGGIH 4WXX   &&__**NN,,((((NN    sA*D2c |jtjtj|jk(j }|st ddddd|jt jt j|jk(t j|k(j }|st ddddd|jt jt j|jk(t j|jk(j }|r*|j|jk7rt ddddd|j|_ |j|_ |j|_ tj|_ |j|j!|d d d |j|j|j|j|j|j&|j|jdgdS#t"$r"|j%t d dd d dwxYw)z` Update a branch using branch_id. Ensures the branch belongs to the authenticated user. rDFrZr&r*rWr%r[rOz,An error occurred while updating the branch.TrEzBranch updated successfullyr\rF)rrrrrrrr0r4r1r2r3rrHr6r9r:rRrIr5)rr4rKrr#branch_to_updater]s r update_branchrc.s( HHUO " "5==L4H4H#H I O O QE #S=OP  xx'..v%../PRXRbRbfoRopvvx #S=PQ  hhv&--foo.OQWQcQcgmgygyQyzAACO?448H8R8RR#S=kl  $*#5#5 %~~$*$7$7!"*//"3   #$0.77,55/;;+33 0 = =.99.99+33       #S=kl   s 5!H77+I"cR|jtjtj|jk(j }|st ddddd|jt jt j|jk(t j|k(j }|st ddddd |j||jd d d dS#t$r"|jt dddddwxYw) z` Delete a branch using branch_id. Ensures the branch belongs to the authenticated user. rDFrZr&r*rWrOz,An error occurred while deleting the branch.TrEzBranch deleted successfully.) rrrrrrrr0r4rQr9rRrI)rr4rr#branch_to_deletes r delete_branchrfjs HHUO " "5==L4H4H#H I O O QE uc]o4pqqxx'..v%../PRXRbRbfoRopvvx uc]p4qrrN "# 1 33 N uc^L5MN NNs !C;;+D&N)!sqlalchemy.ormrsqlalchemy.excrfastapirrsrc.apps.stores.modelsrrr src.apps.stores.schemasr r r r src.apps.auth.controllerrsrc.apps.users.modelsrrrrSr"r@dictrMrUrXr^r`rcrfrrrpsa")!;;\\5& Q'QQQQdQD'DsD 7W7[77v@'@3@d@t@FXWXDX@  C C t =g=|=4=D=~Ct@9g9#9|9SW9\`9x3g3#3T3r