o 5)i&@sddlmZddlmZmZddlmZddlmZm Z m Z m Z ddl Z ddl Z ddlZddlZddlmZedZGdd d Z dd ed e ed e fd dZdd ed e efddZdefddZdS))Cache) current_appgwraps)AnyOptionalCallableUnionN) get_logger cache_servicec @seZdZdZdZeddZededefddZedd ed e de fd d Z edd ed e de e de fddZed ede fddZede fddZedefddZdS) CacheServicez7Simple and robust caching service for Flask applicationNc Cs,zjtd|jd|jdd|jdd|jddd}|dd kr4|d |jd d i|dd krW||jd d|jdd|jdd|jddt||d|_td|dWd Sty}zt dt |t|ddid|_t dWYd }~d Sd }~ww)z!Initialize caching with Flask appz CACHE_TYPE CACHE_TYPEsimpleCACHE_DEFAULT_TIMEOUTi,CACHE_KEY_PREFIXqpgs_api)rrr filesystem CACHE_DIRNredisCACHE_REDIS_HOST localhostCACHE_REDIS_PORTiCACHE_REDIS_DBrCACHE_REDIS_PASSWORD)rrrr)configzCache initialized with type: zFailed to initialize cache: z&Falling back to simple in-memory cache) loggerdebugrgetupdatercacheinfo Exceptionerrorstrwarning)clsapp cache_configer*8/var/www/html/oms-prod-mcp/app/services/cache_service.pyinit_apps:         zCacheService.init_appprefixreturnc Osz*||r t|nid}tj|dtd}t|dd}|d|WSt yU}zt dt||dt t|t|WYd}~Sd}~ww)z-Generate consistent cache key from parameters)argskwargsT) sort_keysdefaultN :zError generating cache key: ) sorteditemsjsondumpsr$hashlibmd5encode hexdigestr"rr%hash)r&r-r/r0key_data key_stringkey_hashr)r*r*r+ get_cache_key;s*zCacheService.get_cache_keykeyr2c Csfz|jdur |WS|j|p|WSty2}ztd|dt||WYd}~Sd}~ww)z(Get value from cache with error handlingNzCache get error for key : )r rr"rr%r$)r&rBr2r)r*r*r+rPs zCacheService.getvaluetimeoutc Csfz|jdur WdS|jj|||dWSty2}ztd|dt|WYd}~dSd}~ww)z&Set value in cache with error handlingNF)rEzCache set error for key rC)r setr"rr%r$)r&rBrDrEr)r*r*r+rF[s zCacheService.setc Cs`z|jdur WdS|j|WSty/}ztd|dt|WYd}~dSd}~ww)zDelete value from cacheNFzCache delete error for key rC)r deleter"rr%r$)r&rBr)r*r*r+rGfs zCacheService.deletec CsXz|jdur WdS|jWSty+}ztdt|WYd}~dSd}~ww)zClear all cacheNFzCache clear error: )r clearr"rr%r$)r&r)r*r*r+rHqs  zCacheService.clearc Csz|jdur dddWStjdd}d|tt|jjdWStyB}ztd t|d t|d WYd}~Sd}~ww) zGet cache statisticsNdisablednone)statustyperunknownactive)rKrLbackendzError getting cache stats: r#)rKr#) r rrrr$rLr"rr%)r& cache_typer)r*r*r+ get_stats|s  zCacheService.get_statsN)__name__ __module__ __qualname____doc__r classmethodr,r$rArrrintboolrFrGrHdictrQr*r*r*r+r s" ' "   r cache_key_prefixrE conditioncsfdd}|S)z Decorator to cache function results Args: cache_key_prefix: Prefix for cache key timeout: Cache timeout in seconds (None = use default) condition: Function to determine if result should be cached cstfdd}|S)Nc stjdur |i|Sr|i|s|i|Stjg|Ri|}t|}|dur;td||Sz|i|}|durVt||td||WStys}ztdj dt |d}~ww)NzCache hit for key: zCache set for key: z Error executing cached function rC) r r rArrrrFr"r#rSr$)r/r0 cache_key cached_resultresultr))r[r\funcrEr*r+wrappers,  z0cached_query..decorator..wrapperrr`rar[r\rEr`r+ decorators"zcached_query..decoratorr*)r[rEr\rer*rcr+ cached_querys &rfcsfdd}|S)z: Decorator specifically for caching API responses cstfdd}|S)Nc stjdur |i|Sddlm}t|j|jr|ndd}tj |}t d|t |}t d||durLt d||Sz[|i|}d}t |drj|jd k}t d |jn*t|ttfryd }t d nt|trt|d kr|dd k}t d|d|rt||t d||WSty}zt dj dt|d}~ww)Nr)request)r/r7zGenerated cache key: zCache response: zAPI cache hit for: F status_codezFlask Response status: TzDirect data response receivedzTuple response status: zAPI response cached: zError in cached API function rC)r r flaskrgrZr/is_jsonget_jsonrArSrrrhasattrrh isinstancelisttuplelenrFr"r#r$) r/r0rgrequest_paramsr]cached_responseresponse should_cacher))r[r`rEr*r+rasH       z2cache_response..decorator..wrapperrrbr[rErdr+res6z!cache_response..decoratorr*)r[rErer*rxr+cache_responses:rypatternc Cslztjdur WdStd|tWSty5}ztd|dt|WYd}~dSd}~ww)z(Invalidate cache keys matching a patternNFzInvalidating cache pattern: z!Error invalidating cache pattern rC)r r rr!rHr"r#r$)rzr)r*r*r+invalidate_cache_patterns  r{)NNrR) flask_cachingrrlrr functoolsrtypingrrr r r9r7syspickleapp.utils.helpersr rr r$rXrfryr{r*r*r*r+s,    5B