lidddlmZddlmZddlmZmZddlmZGddeZ Gdde Z y ) )List)Protocol) PayloadDictPayloadDictList)LoggercLeZdZdZdedefdZdefdZdededdfd Z de fd Z y) EventStorezC EventStore protocol. For buffering events in the Emitter. payloadreturncy)z Add PayloadDict to buffer. Returns True if successful. :param payload: The payload to add :type payload: PayloadDict :rtype bool Nselfr s a/var/www/html/strategist-ai/venv_dbt/lib/python3.12/site-packages/snowplow_tracker/event_store.py add_eventzEventStore.add_events cy)zd Get a list of all the PayloadDicts in the buffer. :rtype PayloadDictList Nr rs rget_events_batchzEventStore.get_events_batch' rbatch need_retryNcy)aL Removes sent events from the event store. If events need to be retried they are re-added to the buffer. :param batch: The events to be removed from the buffer :type batch: PayloadDictList :param need_retry Whether the events should be re-sent or not :type need_retry bool Nr )rrrs rcleanupzEventStore.cleanup/s rcy)Q Returns the number of events in the buffer :rtype int Nr rs rsizezEventStore.size:rr) __name__ __module__ __qualname____doc__rboolrrrrintrr rrr r sL    /   _  $  4   c rr cneZdZdZddededdfdZdedefdZ de fd Z d e d eddfd Z defd Z defdZy)InMemoryEventStorezg Create a InMemoryEventStore object with custom buffer capacity. The default is 10,000 events. loggerbuffer_capacityr Nc.g|_||_||_y)a :param logger: Logging module :type logger: Logger :param buffer_capacity: The maximum capacity of the event buffer. When the buffer is full new events are lost. :type buffer_capacity int N) event_bufferr'r&)rr&r's r__init__zInMemoryEventStore.__init__Hs02. rr c|jr|jjdy|jj |y)z| Add PayloadDict to buffer. :param payload: The payload to add :type payload: PayloadDict z%Event buffer is full, dropping event.FT)_buffer_capacity_reachedr&errorr)appendrs rrzInMemoryEventStore.add_eventTs=  ( ( * KK  E F   )rc,|j}g|_|S)zk Get a list of all the PayloadDicts in the in the buffer. :rtype PayloadDictList )r))rrs rrz#InMemoryEventStore.get_events_batchbs !! rrrc\|sy|D]$}||jvs|j|r$yy)aZ Removes sent events from the InMemoryEventStore buffer. If events need to be retried they are re-added to the buffer. :param batch: The events to be removed from the buffer :type batch: PayloadDictList :param need_retry Whether the events should be re-sent or not :type need_retry bool N)r)r)rrrevents rrzInMemoryEventStore.cleanupls6  ED---~~e, rc,t|jS)r)lenr)rs rrzInMemoryEventStore.size}s 4$$%%rc<|j|jk\S)zR Returns true if buffer capacity is reached :rtype: bool )rr'rs rr,z+InMemoryEventStore._buffer_capacity_reacheds yy{d2222r)i')rrr r!rr#r*rr"rrrrrr,r rrr%r%Csr v      /_$4"&c&3$3rr%N) typingrtyping_extensionsrsnowplow_tracker.typingrrloggingrr r%r rrr9s.$&@( ( VH3H3r