]j&dZddlZddlZddlmZddlmZddlmZddl Z ddl Z ddl m Z ddl mZddl mZdd lmZdd lmZdd lmZej,eZd d Z eZej7ej9d dZdZdedefdZ de!fdZ"dZ#dZ$d!dZ%dZ&dZ'dddZ(dZ)dZ*dZ+dZ,y#ej9ddZYNxYw)"zThis file and its contents are licensed under the Apache License 2.0. Please see the included NOTICE for copyright information and LICENSE for a copy of the license. N) timedelta)partial)Any)CurrentContext)settings)get_connection)send_stop_job_command)InvalidJobOperation)StartedJobRegistryc^ fd fd}|Dcgc] }|| }}|jDcic]\}}|dk7s |||}}}g} |r| jd||r| jd|| rdj| SdScc}wcc}}w#t$rYywxYw) NcNt|}|dt|kDrdzSdzS)Nz...)reprlen)valuev_repr max_lengths A/root/env/lib/python3.12/site-packages/label_studio/core/redis.py_truncate_scalarz4_truncate_args_for_logging.._truncate_scalars2%[F+:&3v;3K%T TQST Tct|trVg}|jD]*\}}|jt |d|,ddj |zdzS|S)Nz: {, }) isinstancedictitemsappendrjoin)rpartsdkdvrs r_truncate_top_levelz7_truncate_args_for_logging.._truncate_top_levelsu%&#kkmHFBLLDH:R0@0D/E!FGHTYYu--33#E* *r on_failurezargs: zkwargs: rz no argumentszfailed to format arguments)rrr Exception) argskwargsrr#argtruncated_argskvtruncated_kwargsresultrs ` @r_truncate_args_for_loggingr.s, U +?CCs-c2CCBH,,.f$!QTUYeTeA2155ff  MMF>"23 4  MMH%5$67 8$*tyy >>Df ,+,s9B BB B BAB B  B B,+B,z#=> Redis is connected successfully.z=> Redis is not connected.ctsy tjtjdy#tj j $r$}tjd|dYd}~yd}~wtj j$r$}tjd|dYd}~yd}~wtj j$r$}tjd|dYd}~yd}~wwxYw)NFzRedis client is alive!Tz/Redis healthcheck failed with ConnectionError: )exc_infoz,Redis healthcheck failed with TimeoutError: zRedis healthcheck failed: ) _redispingloggerdebugredis exceptionsConnectionErrorerror TimeoutError RedisError)excs rredis_healthcheckr<>s     -.    + + FseLW[ \    ( ( CC5ITX Y    & & 1#7$ Gs-3C2A// C2B.. C2C--C2c8tjr tSy)NF)r REDIS_ENABLEDr<rrredis_connectedr@Qs "" rrreturnc jt|ttttt t tdfS)z3Check if a value can be serialized for job context.N)rstrintfloatboollistrtype)rs r_is_serializablerIWs" ec3tT4dL MMrcdi}tjx}r|j|d<tjx}r||d<|s)|r't |dr|j r|j |d<tj }|jD]\}}|dvs t|s|||<|S)z| Capture the current context for passing to a job. Returns a dictionary of context data that can be serialized. user_idorganization_idactive_organization_id)userrequest) rget_useridget_organization_idhasattrrM get_job_datarrI) context_datarNorg_idjob_datakeyrs r_capture_contextrY\s L&&((t("&'' Y 3355v5*0 &' dwt-EF4KfKf*.*E*E &'**,Hnn&& U ) ).>u.E %L & rcBtsytj|SN)r<r1getrXs r redis_getr^ys   ::c?rcDtsytj||Sr[)r<r1hget)key1key2s r redis_hgetrcs   ;;tT ""rcHtsytj|||S)N)ex)r<r1set)rXrttls r redis_setrhs   ::c5S: ))rcFtsytj|||Sr[)r<r1hset)rarbrs r redis_hsetrks   ;;tT5 ))rcBtsytj|Sr[)r<r1deleter]s r redis_deleterns   == r) in_secondsc ddlm}txr|jdd}|jdd}d|vr|d=d|vr|d=d}d|vr|d}|d=d}d |vr!|d }|d =t |t r || }|j d d} |r t} |jd i} | j| | |d < t||} tjd|jd|d| dtj|} | j }|dkDr t#| j$t'|}||g|i||t(j*|| d}|S ||i|}|S#t$r(tjd |jd|YwxYw#t$r)tjd|jd|dYwxYw#t$r!t-j.}| r | |g|wxYw)a Start job async with redis or sync if redis is not connected. Automatically preserves context for async jobs and clears it after completion. :param job: Job function :param args: Function arguments :param in_seconds: Job will be delayed for in_seconds :param retry: RQ Retry object or int (max retries). Only used in async mode. :param kwargs: Function keywords arguments :return: Job or function result r)Retryr5T queue_namedefaultN job_timeoutretry)maxr$metaz"Failed to capture context for job z on queue zStart async job z with .)seconds)rt failure_ttlrur$)rqrqr@r\rrDpoprYupdater%r3info__name__r. django_rq get_queueenqueuer enqueue_inrrRQ_FAILED_JOB_TTLsysr0)jobror&r'rqr5rrrtrur$rUrw args_infoqueueenqueue_methodr-r0s rstart_job_async_or_syncrsC   ;&**Wd";EL)4Jv < & 7OK]+ = ! E&w 7O eS !e$EL$/J  c+-L::fb)D KK %!F6N R24@I KK*3<<. :,fU^T__`a b##J/ >$U%5%5y7TUN    $ 22!   $)&)FM5 c KKc2t|||}t|S)z Checks if func_name with kwargs[meta] is in queue (doesn't check workers) :param queue: queue object :param func_name: function name :param meta: job meta information :return: True if job in queue )get_jobs_by_metaany)r func_namerwjobss ris_job_in_queuers E9d 3D t9rc|syt|t}t|tr|j n|}|j j |j|duS)z Checks if job id is on workers :param job_id: Job ID :param queue_name: Queue name :return: True if job on worker F) connectionN)r r1rrCencoderzscorerX)job_idrrregistrymembers ris_job_on_workerrsP !*@H *63 7V]]_VF    % %hllF ;4 GGrc P|j|}|atjd|d|jd |j |j tj d|dytjd|d |jd tt|tj d |d y#t$rtj d|dYywxYw#t$r.}tj d |d t|Yd}~yd}~wwxYw)zk Delete job by id from queue @param queue: Queue on redis to delete from @param id: Job id Nz Stopping job z from queue rxz Fetched job z and stopped.zJob z was already cancelled.z on worker from queue zSend stop job z to redis worker.z Redis job z was not found: ) fetch_jobr3r~namecancelrmr4r r r1r%rC)rrQres rdelete_job_by_idrs //" C  mB4|EJJ"->? @# = LL4t#:; < = D LL:bT)9#a&B C C Ds)9C)C."C+*C+. D%7$D  D%cfd|jD}|Dcgc]!}t|ds|j|k(s |#c}Scc}w)z Get jobs from queue by func_name and meta data :param queue: Queue on redis to check in :param func_name: Started function name :param meta: meta dict :return: Job list c3VK|] }|jjk(s|"ywr[)funcr).0rrs r z#get_jobs_by_meta..*s" NCsxx/@/@I/MC Ns))rw)get_jobsrSrw)rrrwrrs ` rrr!s? O5>>+ ND MC73#7CHHrs  / $,-*   8 $,@  F KKM LL67 & NCNDN $: # * *  45L^ H&D4 NQ LL-. Fs (B??C