tĀi UdZddlZddlmZddlmZmZmZddlZ ddlm Z ddl m Z ddlmZej eZdaee ed<d e fd Zed ee j,j.ddffd Ze dd ed ed ee j,j4j6ddffdZ ddedeeded eee eeffdZ!dedeed e"fdZ#d efdZ$y)z/ MariaDB connection pool for content pipeline. N)contextmanager)Any GeneratorOptional)Error)MySQLConnectionPool) get_settings_poolreturnct8t}tddddd|jatj dtS)zGet or create connection pool.content_pipelineT) pool_name pool_sizepool_reset_sessionz Database connection pool created)r r r db_configloggerinfo)settingss 5/var/www/html/content-pipeline/database/connection.pyget_poolrsL }># (#      67 Lc#Kt}|j} ||jy#|jwxYww)zGet a connection from the pool.N)rget_connectionclose)pool connections rrr#s@ :D$$&J sA3AAA dictionarybufferedc#\Kt5}|j||} ||j |j dddy#t$r.}|j t j d|d}~wwxYw#|jwxYw#1swYyxYww)z2Get a cursor with automatic connection management.)rr zDatabase error: N)rcursorcommit MySQLErrorrollbackrerrorr)rr rr"es r get_cursorr(.s   Z""j8"L L     LLN        ! LL+A3/ 0   LLN  sJ B,B AB  B, B)BBB  BB  B)%B,queryparamsfetchct5}|j|||r|jcdddS dddy#1swYyxYw)z-Execute a query and optionally fetch results.N)r(executefetchall)r)r*r+r"s r execute_queryr/@sJ uf% ??$ s$AAA  params_listc~t5}|j|||jcdddS#1swYyxYw)z-Execute a query with multiple parameter sets.N)r( executemanyrowcount)r)r0r"s r execute_manyr4Ks5 5+.s3<c td}|duxrt|dkDS#t$r"}tj d|Yd}~yd}~wwxYw)zTest database connectivity.zSELECT 1 as testNrzConnection test failed: F)r/lenr$rr&)resultr's rtest_connectionr8RsQ12T!5c&kAo5  /s34s! A AA )TT)NT)%__doc__logging contextlibrtypingrrrmysql.connectormysqlrr$mysql.connector.poolingrconfig.settingsr getLogger__name__rr __annotations__r connectorMySQLConnectionrboolr" MySQLCursorr(strtuplelistdictr/intr4r8rrrrMs:%++/7(   8 $'+x#$+ %  %//"A"A4"MN.2'+u%%114=>$?C  7; d4S>"#$u+#r