;&i) ddlmZmZmZmZddlZddlmZddlm Z ddl m Z m Z m Z mZmZmZddlmZddlmZddlmZdd lmZdd lmZdd lmZej:eZd e d ededefdZ d e dee!deefdZ"dd e dee!deefdZ# dd e de!de!deeeeffdZ$d e dee!ee%e!ee!fffdZ&d e dee!ee%e!ee!fffdZ'y))DictListUnionOptionalN)status)Session)selectand_or_funcdescasc) APIException) SiteMaster) UserSchema)SiteMasterSchema)SiteMasterCreateRequestSchema)DEFAULT_SITE_MASTERdbpayload created_byreturnc K t|j|j|j|j}|j ||j |j|tj|S#t$rZ}|jtjd|ttit j"dt%|d}~wwxYww)z= Create a new site master entry using SQLAlchemy 2.0 )keyvaluelabelgroupz"Error creating site master entry: z$Failed to create site master entry: moduleerror status_codemessageN)rrrrraddcommitrefreshrmodel_validate Exceptionrollbackloggerr r__name__rHTTP_400_BAD_REQUESTstr)rrr site_masteres \/var/www/html/hwPaymentPortal-be-dev/hw-payment-portal-api/src/apps/site_masters/services.pycreate_master_entryr0s    ------   {  ;..{;;    9!=>33:3q6(C    s*C(A=BC( C% AC  C%%C(keysc ttjttjj |}|j |jj}|Dcgc]}tj|c}Scc}w#t$r$}tjd|gcYd}~Sd}~wwxYw)z> Get site master entries by keys using SQLAlchemy 2.0 z+Error getting site master entries by keys: N)r rwherer rin_executescalarsallrr&r'r)r )rr1stmtresultitemr.s r/ get_by_keysr;2s j!'' ""4(   D!))+//1BHI$ //5III  B1#FG s0A5B7BBB C$C=CCsearchcn tt}ttdr.|jtjj d}|rtj jd|dtjjd|dtjjd|dtjjd|dg}|jt|}|jttjttj }|j|j!j#}|Dcgc]}t%j&|c}Scc}w#t($r$}t*j-d|gcYd}~Sd}~wwxYw)z: Get all site master entries using SQLAlchemy 2.0 deleted_atN%z'Error getting all site master entries: )r rhasattrr3r>is_rilikerrrr order_byr display_orderr5r6r7rr&r'r)r )rr<r8search_conditionsr9r:r.s r/get_allrFEsZj! :| ,::j3377=>D $$q]3  &&6(!}5  &&6(!}5  &&6(!}5 !  ::c#456D}}S!9!9:C qcBC s0E!F#F?FF F4F/)F4/F4cKg}t|dkDrT|jd}t|dkDr!ttitj dt ||}d}d|vrn t|| }i}|D]=}|j|vrg||j<||jj|?d|vrr||d<|Sw) z6 List site master entries with SQLAlchemy 2.0 r, z0A maximum of 10 entries can be requested at oncer)rr1N industries)r<) lensplitrr*rr+r;rFrappend)rr1r<itemsentriesrJresponseentrys r/list_master_entriesrRds %'E 4y1}**S/ w<" "77J  r0 4  6*H* 99H $"$HUYY ""5)* t !+ OsCCc  tjdttjt j }|j|jxsd}tjd|g}g}g}g}tD]} tt jtt j|jdk(t j|jdk(}|j|j} | st di|} |j!| |j!|jdd|jdtj#d|jdd|jdnm|j!|jdd|jdtj#d|jdd|jd|rD|j+||j-tjd t/|dntjd|j|jxsd} ddt/|t/|t/|t/t|| |r|d dng|r|d dng|d }tjdt/|dt/|dt/|d|S#t$$rR} d |jdd d t'| } |j!| tj)| Yd } ~ d } ~ wwxYw#t$$rL}|j1dt'|} tj)| d| ddddddgg| gd cYd }~Sd }~wwxYw)z Populate SiteMaster table with default data from seeder using SQLAlchemy 2.0 Args: db: Database session Returns: Dictionary containing operation results with counts and messages z0Starting to populate default site master data...rz(Current SiteMaster records in database: rrz -> z Will insert: zSkipped (exists): zError processing item unknownz: NzSuccessfully inserted z new SiteMaster recordsz2No new records to insert - all data already existsTz%Site master data population completedrI successr"records_insertedrecords_skippedrecords_with_errorstotal_processeddatabase_total_beforedatabase_total_afterinserted_items skipped_items error_itemszPopulation summary: z inserted, z skipped, z errorsz%Failed to populate site master data: F)r)infor r countridr5scalarrr3r rgetrscalar_one_or_nonerMdebugr'r,r add_allr$rKr()r total_stmt current_total to_insertinserted_recordsskipped_records error_recordsr: exists_stmtexisting_recordr- item_error error_msg final_totalr9r.s r/!populate_default_site_master_datarvsW  FGDJJz}}56  :.557<1  >}oNO  ( (D ($Z066"$((5/9"((DHHW,== #%**["9"L"L"N&","4t"4K$$[1$++txx.?tDHHWDUCV,WXLL=%0AdhhwFWEX!YZ#**dhhuo->d488GCTBU+VWLL#5dhhuo5Fd488T[K\J]!^_' (6  JJy ! IIK KK0Y0@@WX Y KKL Mjj,335: > #$4 5"?3#&}#5"#67%2$/7G.s3R4C_Ra0(    *3/?+@*ASQ`MaLbblmpq~mmAAHI J A (4TXXeY5O4PPRSVWaSbRcd $$Y/ Y'' (D   ;CF8D  Y ! #$ %&$% %;   sLBM0FLDM0 M-AM("M0(M--M00 O9AO:OOc  ttjtj}|j |j xsd}tt}||k\rdd|d|dd|dd||gd|d|ggd Stjd|d|d t|S#t$r<}d t|}tj|d |ddd dddgg|gd cYd }~Sd }~wwxYw)z Populate site master data only if the table is empty or has fewer records than the default dataset Args: db: Database session Returns: Dictionary containing operation results rTz/Site master table already has sufficient data (/z records)z Table has z records, expected rVz. Proceeding with population...z"Failed to check site master data: FraN)r r rdrrer5rfrKrr)rcrvr'r,r )rrkrlexpected_totalr.rts r/seed_site_master_data_if_emptyrzs"' DJJz}}56  :.557<1 01 N *L]O[\]k\lluv$%#0'(#$)6(5"$$.}o=PQ_P`"a!b!    j/B>BRRqrs044  8QA  Y ! #$ %&$% %;   s$A>B(&B(( C-11C("C-(C-)N)(typingrrrrlogging starlettersqlalchemy.ormr sqlalchemyr r r r r rsrc.core.exceptionsr(src.apps.site_masters.models.site_masterr"src.apps.users.schemas.user_commonr0src.apps.site_masters.schemas.site_master_commonr2src.apps.site_masters.schemas.site_master_requestsr)src.apps.site_masters.seeders.site_masterr getLoggerr*r)r0r,r;rFrRintrvrzrbr/rsJ.."99,?9M\I   8 $  7 EO  DG496F1G&#$?O:P@+/((($'( 4 !4 '((Va 'a d3c3PTUXPYFY@Z;Z6[a H1 w1 4U3TRUYCV=W8W3X1 r