j5bdZddlZddlZddlmZddlmZmZmZm Z m Z m Z m Z m Z mZmZmZmZGddeZGdd eZGd d eZGd d ZGddZejZdadZdZdZdZddZdZddZ dZ!dZ"dZ#ddZ$dZ%dZ&dZ'dS)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_messageceZdZdZdS) StashErrorz$Base exception for stash operations.N)__name__ __module__ __qualname____doc__G/var/www/html/venv/lib/python3.11/site-packages/gunicorn/dirty/stash.pyrrEs....rrc"eZdZdZfdZxZS)StashTableNotFoundErrorz#Raised when a table does not exist.c^||_td|dS)NzStash table not found: ) table_namesuper__init__)selfr __class__s rrz StashTableNotFoundError.__init__Ls1$ ?:??@@@@@rrrrrr __classcell__r!s@rrrIsG--AAAAAAAAArrc"eZdZdZfdZxZS)StashKeyNotFoundErrorz,Raised when a key does not exist in a table.cr||_||_td|d|dS)NzKey not found in z: )rkeyrr)r rr(r!s rrzStashKeyNotFoundError.__init__Ts@$ @Z@@3@@AAAAArr"r$s@rr&r&QsG66BBBBBBBBBrr&ceZdZdZddZdZdZdZddZd Z dd Z d Z dd Z d Z dZdZddZdZdZdZdZdZdZdS) StashClientzl Client for stash operations. Communicates with the arbiter which stores all tables in memory. >@c`||_||_d|_tj|_dS)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-'  ^%% rcBttjS)zGenerate a unique request ID.)struuiduuid4r s r_get_request_idzStashClient._get_request_idns4:<<   rcNddl}|jdS ||j|j|_|j|j|j|jdS#|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 r:es r_connectzStashClient._connectrs : ! F Jv~v7IJJDJ J ! !$, / / / J  t/ 0 0 0 0 0 g& J J JDJAaAABB I JsA#A44B$BB$c||j4 |jn#t$rYnwxYwd|_dSdS)zClose the connection.N)r/close Exceptionr7s r_closezStashClient._closesY : !    """"    DJJJ " !s # 00Nc |j5|j||}t ||||||} t j|j|t j|j}|d} | t j kr!|dcdddS| t j kr||di} | dd} | dd } | d krt|| d krt||t| td | #t$rB} |t!| trtd | | d} ~ wwxYw#1swxYwYdS)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: )r2r/rBr8rr write_message read_messagegetMSG_TYPE_RESPONSEMSG_TYPE_ERRORrr&rrErF isinstance)r optabler(rHrI request_idrMresponsemsg_type error_inforL error_msgrAs r_executezStashClient._executes)Z" H" Hz! --//J(BugG  H+DJ@@@(5djAA#<<//}>>>#<<11!" H" H" H" H" H" H" H" H"!===!)gr!:!:J!+ l!K!KJ *y/ J JI!%>>>5e<<<!%<<<3E3???$Y///$%L(%L%LMMM H H H a,, !?A!?!?@@aG  H=" H" H" H" H" H" H" H" H" H" Hs8AF4A,E%BE%% F1/=F,,F11F44F8;F8cB|t|||dS)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(rHN)r[r)r rUr(rHs rputzStashClient.puts$ lEs% @@@@@rcb |t||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()r[rr&)r rUr(defaults rrPzStashClient.getsC ==u#=>> >$   NNN s  ..c<|t||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_)r[rr rUr(s rdeletezStashClient.delete}}_e}===rc<|t||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 rI)r[r )r rUrIs rkeyszStashClient.keyss}}]E7}CCCrc<|t|dS)z] Delete all entries in a table. Args: table: Table name N)r[r r rUs rclearzStashClient.clears ne,,,,,rc8|t|S)z Get information about a table. Args: table: Table name Returns: Dict with table info (size, etc.) )r[r ris rinfozStashClient.infos}}]E222rc<|t|dS)z Ensure a table exists (create if not exists). This is idempotent - calling it multiple times is safe. Args: table: Table name N)r[r ris rensurezStashClient.ensure s ou-----rc<|t||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_)r[rrbs rexistszStashClient.existsrdrc<|t|dS)zV Delete an entire table. Args: table: Table name N)r[r ris r delete_tablezStashClient.delete_table$s +U33333rc8|tdS)zT List all tables. Returns: List of table names )r[rr7s rtableszStashClient.tables-s}}_b111rc"t||S)z Get a dict-like interface to a table. Args: name: Table name Returns: StashTable instance ) StashTable)r names rrUzStashClient.table6s$%%%rcn|j5|ddddS#1swxYwYdS)zClose the client connection.N)r2rFr7s rrDzStashClient.closeBs{ Z   KKMMM                  s *..c|SNrr7s r __enter__zStashClient.__enter__Gs rc.|dSr{)rD)r exc_typeexc_valexc_tbs r__exit__zStashClient.__exit__Js r)r+)NNNr{)rrrrrr8rBrFr[r]rPrcrgrjrlrnrprrrurUrDr|rrrrr*r*Zsc & & & &!!! J J J0H0H0H0Hl A A A" > > > D D D D--- 3 3 3 . . . > > > >444222 & & & rr*cteZdZdZdZedZdZdZdZ dZ dZ d Z dd Z dd Zd ZdZdZd S)rwa2 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"||_||_dSr{)_client_name)r clientrxs rrzStashTable.__init__^s  rc|jS)z Table name.)rr7s rrxzStashTable.namebs zrc|j|j|}|/|j|j|st ||Sr{)rrPrrpKeyError)r r(rKs r __getitem__zStashTable.__getitem__gsM!!$*c22 ><&&tz377 $smm# rcH|j|j||dSr{)rr]r)r r(rHs r __setitem__zStashTable.__setitem__os$ S%00000rcd|j|j|st|dSr{)rrcrrr r(s r __delitem__zStashTable.__delitem__rs4|""4:s33 3--   rcB|j|j|Sr{)rrprrs r __contains__zStashTable.__contains__vs|""4:s333rcZt|j|jSr{)iterrrgrr7s r__iter__zStashTable.__iter__ys"DL%%dj11222rcl|j|j}|ddS)Nsizer)rrlrrP)r rls r__len__zStashTable.__len__|s-|  ,,xx"""rNcD|j|j||S)zGet value with default.)rrPr)r r(r`s rrPzStashTable.gets| C999rcD|j|j|S)z-Get all keys, optionally filtered by pattern.rf)rrgr)r rIs rrgzStashTable.keyss|  W ===rcD|j|jdS)zDelete all entries.N)rrjrr7s rrjzStashTable.clears  4:&&&&&rc#K|j|jD]&}||j|j|fV'dS)z Iterate over (key, value) pairs.NrrgrrPrs ritemszStashTable.itemssY<$$TZ00 9 9Ct|'' C888 8 8 8 8 9 9rc#K|j|jD]$}|j|j|V%dS)zIterate over values.Nrrs rvalueszStashTable.valuessT<$$TZ00 4 4C,""4:s33 3 3 3 3 4 4rr{)rrrrrpropertyrxrrrrrrrPrgrjrrrrrrwrwNs  X111   444333###::::>>>>'''999 44444rrwc |adS)z@Set the global stash socket path (called during initialization).N)_stash_socket_path)paths rset_stash_socket_pathrsrcddl}t-|jd}|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.)osrenvironrPr)rrs rget_stash_socket_pathrsR III!z~~566  K I    rcttdd}|)t}t|}|t_|S)z*Get or create a thread-local stash client. stash_clientN)getattr _thread_localrr*r)rr-s r _get_clientrs= ]ND 9 9F ~+-- [))%+ " MrcLt|||dS)zStore a value in a table.N)rr])rUr(rHs rr]r]s$MMeS%(((((rcHt|||S)zRetrieve a value from a table.)rrP)rUr(r`s rrPrPs ==  UC 1 11rcFt||S)zDelete a key from a table.)rrcrUr(s rrcrc ==  s + ++rcFt||S)zGet all keys in a table.)rrg)rUrIs rrgrgs ==  eW - --rcHt|dS)zDelete all entries in a table.N)rrjrUs rrjrjs MMrcDt|S)zGet information about a table.)rrlrs rrlrls ==  e $ $$rcHt|dS)zEnsure a table exists.N)rrnrs rrnrns MMrcFt||S)zCheck if a table or key exists.)rrprs rrprprrcHt|dS)zDelete an entire table.N)rrrrs rrrrrs MMu%%%%%rcBtS)zList all tables.)rrurrrrurus ==   ! !!rcDt|S)z%Get a dict-like interface to a table.)rrU)rxs rrUrUs ==  t $ $$rr{)(rr0r5errorsrprotocolrrrrr r r r r rrrrrr&r*rwlocalrrrrrr]rPrcrgrjrlrnrprrrurUrrrrs **X                              ////////AAAAAjAAABBBBBJBBBqqqqqqqqhF4F4F4F4F4F4F4F4\ !!    ))) 2222 ,,, ....  %%%    ,,,, &&& """ %%%%%r