AҐi5FdZddlZddlZddlmZddlmZmZmZm Z m Z m Z m Z m Z mZmZmZmZGddeZGdd eZGd d eZGd d ZGddZej.ZdadZdZdZdZddZdZddZ dZ!dZ"dZ#ddZ$dZ%dZ&dZ'y)aL Stash - Global Shared State for Dirty Workers Provides simple key-value tables stored in the arbiter process. All workers can read and write to the same tables. Usage:: from gunicorn.dirty import stash # Basic operations - table is auto-created on first access stash.put("sessions", "user:1", {"name": "Alice", "role": "admin"}) user = stash.get("sessions", "user:1") stash.delete("sessions", "user:1") # Dict-like interface sessions = stash.table("sessions") sessions["user:1"] = {"name": "Alice"} user = sessions["user:1"] del sessions["user:1"] # Query operations keys = stash.keys("sessions") keys = stash.keys("sessions", pattern="user:*") # Table management stash.ensure("cache") # Explicit creation (idempotent) stash.clear("sessions") # Delete all entries stash.delete_table("sessions") # Delete the table itself tables = stash.tables() # List all tables Declarative usage in DirtyApp:: class MyApp(DirtyApp): stashes = ["sessions", "cache"] # Auto-created on arbiter start def __call__(self, action, *args, **kwargs): # Tables are ready to use stash.put("sessions", "key", "value") Note: Tables are stored in the arbiter process and are ephemeral. If the arbiter restarts, all data is lost. N) DirtyError) DirtyProtocol STASH_OP_PUT STASH_OP_GETSTASH_OP_DELETE STASH_OP_KEYSSTASH_OP_CLEAR STASH_OP_INFOSTASH_OP_ENSURESTASH_OP_DELETE_TABLESTASH_OP_TABLESSTASH_OP_EXISTSmake_stash_messageceZdZdZy) StashErrorz$Base exception for stash operations.N)__name__ __module__ __qualname____doc__M/var/www/descvideos/venv/lib/python3.12/site-packages/gunicorn/dirty/stash.pyrrEs.rrc"eZdZdZfdZxZS)StashTableNotFoundErrorz#Raised when a table does not exist.c8||_t| d|y)NzStash table not found: ) table_namesuper__init__)selfr __class__s rrz StashTableNotFoundError.__init__Ls$ 2:,?@rrrrrr __classcell__r!s@rrrIs-AArrc"eZdZdZfdZxZS)StashKeyNotFoundErrorz,Raised when a key does not exist in a table.cL||_||_t| d|d|y)NzKey not found in z: )rkeyrr)r rr(r!s rrzStashKeyNotFoundError.__init__Ts,$ ,ZL3%@Arr"r$s@rr&r&Qs6BBrr&ceZdZdZddZdZdZdZddZdZ dd Z d Z dd Z d Z d ZdZddZdZdZdZdZdZdZy) StashClientzl Client for stash operations. Communicates with the arbiter which stores all tables in memory. c`||_||_d|_tj|_y)z Initialize the stash client. Args: socket_path: Path to the dirty arbiter's Unix socket timeout: Default timeout for operations in seconds N) socket_pathtimeout_sock threadingLock_lock)r r,r-s rrzStashClient.__init__as('  ^^% rc<ttjS)zGenerate a unique request ID.)struuiduuid4r s r_get_request_idzStashClient._get_request_idns4::<  rcxddl}|jy |j|j|j|_|jj |j |jj |jy#|jtf$r}d|_td||d}~wwxYw)z Establish connection to arbiter.rNzFailed to connect to arbiter: ) socketr.AF_UNIX SOCK_STREAM settimeoutr-connectr,errorOSErrorr)r r9es r_connectzStashClient._connectrs :: !  Jv~~v7I7IJDJ JJ ! !$,, / JJ  t// 0 g& JDJ=aSAB I JsA5B B9B44B9c|j# |jjd|_yy#t$rYwxYw)zClose the connection.N)r.close Exceptionr6s r_closezStashClient._closesC :: !    "DJ "  s 1 ==Nc 2|j5|j|j|j}t ||||||} t j |j|t j|j}|jd} | t jk(r|jdcdddS| t jk(rb|jdi} | jdd} | jdd } | d k(r t|| d k(r t||t| td | #t$r5} |jt!| trtd | | d} ~ wwxYw#1swYyxYw)a6 Execute a stash operation. Args: op: Operation code (STASH_OP_*) table: Table name key: Optional key value: Optional value pattern: Optional pattern for keys operation Returns: Result from the operation N)r(valuepatterntyperesultr> error_typermessagez Unknown errorrr&zUnexpected response type: zStash operation failed: )r1r.rAr7rr write_message read_messagegetMSG_TYPE_RESPONSEMSG_TYPE_ERRORrr&rrDrE isinstance)r optabler(rGrH request_idrLresponsemsg_type error_inforK error_msgr@s r_executezStashClient._executes~ZZ" Hzz! --/J(BugG  H++DJJ@(55djjA#<</}>>>#<<1!" H" H"!=!==!)gr!:J!+ l!KJ *y/ JI!%>>5e<<!%<<3E3??$Y//$'A(%LMM H a, #;A3!?@aG  H=" H" Hs1>F  A3E  BE  F 0FF  F  Fc6|jt|||y)z Store a value in a table. The table is automatically created if it doesn't exist. Args: table: Table name key: Key to store under value: Value to store (must be serializable) )r(rGN)rZr)r rTr(rGs rputzStashClient.puts lEs% @rcV |jt||S#t$r|cYSwxYw)z Retrieve a value from a table. Args: table: Table name key: Key to retrieve default: Default value if key not found Returns: The stored value, or default if not found r()rZrr&)r rTr(defaults rrOzStashClient.gets0 ==u#=> >$ N s  ((c2|jt||S)z Delete a key from a table. Args: table: Table name key: Key to delete Returns: True if key was deleted, False if it didn't exist r^)rZrr rTr(s rdeletezStashClient.delete}}_e}==rc2|jt||S)z Get all keys in a table, optionally filtered by pattern. Args: table: Table name pattern: Optional glob pattern (e.g., "user:*") Returns: List of keys rH)rZr )r rTrHs rkeyszStashClient.keyss}}]E7}CCrc0|jt|y)z] Delete all entries in a table. Args: table: Table name N)rZr r rTs rclearzStashClient.clears ne,rc.|jt|S)z Get information about a table. Args: table: Table name Returns: Dict with table info (size, etc.) )rZr rhs rinfozStashClient.infos}}]E22rc0|jt|y)z Ensure a table exists (create if not exists). This is idempotent - calling it multiple times is safe. Args: table: Table name N)rZr rhs rensurezStashClient.ensure s ou-rc2|jt||S)z Check if a table or key exists. Args: table: Table name key: Optional key to check within the table Returns: True if exists, False otherwise r^)rZrras rexistszStashClient.existsrcrc0|jt|y)zV Delete an entire table. Args: table: Table name N)rZr rhs r delete_tablezStashClient.delete_table$s +U3rc.|jtdS)zT List all tables. Returns: List of table names )rZrr6s rtableszStashClient.tables-s}}_b11rct||S)z Get a dict-like interface to a table. Args: name: Table name Returns: StashTable instance ) StashTable)r names rrTzStashClient.table6s$%%rcf|j5|jdddy#1swYyxYw)zClose the client connection.N)r1rEr6s rrCzStashClient.closeBs' ZZ  KKM   s'0c|SNrr6s r __enter__zStashClient.__enter__Gs rc$|jyrz)rC)r exc_typeexc_valexc_tbs r__exit__zStashClient.__exit__Js  r)g>@)NNNrz)rrrrrr7rArErZr\rOrbrfrirkrmrorqrtrTrCr{rrrrr*r*Zsk &! J0Hl A" > D- 3 . >42 & rr*cleZdZdZdZedZdZdZdZ dZ dZ d Z dd Z dd Zd ZdZdZy )rva2 Dict-like interface to a stash table. Example:: sessions = stash.table("sessions") sessions["user:1"] = {"name": "Alice"} user = sessions["user:1"] del sessions["user:1"] # Iteration for key in sessions: print(key, sessions[key]) c ||_||_yrz)_client_name)r clientrws rrzStashTable.__init__^s  rc|jS)z Table name.)rr6s rrwzStashTable.namebszzrc|jj|j|}|1|jj|j|s t ||Srz)rrOrroKeyError)r r(rJs r __getitem__zStashTable.__getitem__gsH!!$**c2 ><<&&tzz37sm# rcR|jj|j||yrz)rr\r)r r(rGs r __setitem__zStashTable.__setitem__os S%0rcf|jj|j|s t|yrz)rrbrrr r(s r __delitem__zStashTable.__delitem__rs)||""4::s33- 4rcN|jj|j|Srz)rrorrs r __contains__zStashTable.__contains__vs||""4::s33rc^t|jj|jSrz)iterrrfrr6s r__iter__zStashTable.__iter__ys DLL%%djj122rcp|jj|j}|jddS)Nsizer)rrkrrO)r rks r__len__zStashTable.__len__|s+||  ,xx""rNcP|jj|j||S)zGet value with default.)rrOr)r r(r_s rrOzStashTable.gets|| C99rcP|jj|j|S)z-Get all keys, optionally filtered by pattern.re)rrfr)r rHs rrfzStashTable.keyss ||  W ==rcN|jj|jy)zDelete all entries.N)rrirr6s rrizStashTable.clears 4::&rc#K|jj|jD],}||jj|j|f.yw)z Iterate over (key, value) pairs.NrrfrrOrs ritemszStashTable.itemssH<<$$TZZ0 9Ct||'' C88 8 9sAAc#K|jj|jD]*}|jj|j|,yw)zIterate over values.Nrrs rvalueszStashTable.valuessC<<$$TZZ0 4C,,""4::s3 3 4sAArz)rrrrrpropertyrwrrrrrrrOrfrirrrrrrvrvNsW 1 43#:>'9 4rrvc|ay)z@Set the global stash socket path (called during initialization).N)_stash_socket_path)paths rset_stash_socket_pathrs rcvddl}t*|jjd}|r|St dtS)zGet the stash socket path.rNGUNICORN_DIRTY_SOCKETz\Stash socket path not configured. Make sure dirty_workers > 0 and dirty_apps are configured.)osrenvironrOr)rrs rget_stash_socket_pathrsA !zz~~56 K I   rclttdd}| t}t|}|t_|S)z*Get or create a thread-local stash client. stash_clientN)getattr _thread_localrr*r)rr,s r _get_clientrs4 ]ND 9F ~+- [)%+ " Mrc:tj|||y)zStore a value in a table.N)rr\)rTr(rGs rr\r\sMeS%(rc8tj|||S)zRetrieve a value from a table.)rrO)rTr(r_s rrOrOs =  UC 11rc6tj||S)zDelete a key from a table.)rrbrTr(s rrbrb =  s ++rc6tj||S)zGet all keys in a table.)rrf)rTrHs rrfrfs =  eW --rc6tj|y)zDelete all entries in a table.N)rrirTs rririsMrc4tj|S)zGet information about a table.)rrkrs rrkrks =  e $$rc6tj|y)zEnsure a table exists.N)rrmrs rrmrmsMrc6tj||S)zCheck if a table or key exists.)rrors rrororrc6tj|y)zDelete an entire table.N)rrqrs rrqrqsMu%rc2tjS)zList all tables.)rrtrrrrtrts =   !!rc4tj|S)z%Get a dict-like interface to a table.)rrT)rws rrTrTs =  t $$rrz)(rr/r4errorsrprotocolrrrrr r r r r rrrrrr&r*rvlocalrrrrrr\rOrbrfrirkrmrorqrtrTrrrrs *X      //AjABJBqqhF4F4\ !  ) 2 , .  %  , & " %r