o I<V, @sddlmZmZddlmZddlmZddlmZddl m Z ddl m Z ddl mZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZm Z m!Z!ddl"m#Z#ddl$m%Z%ddl&m'Z'm(Z(m)Z)ddl*m+Z+m,Z,m-Z-m.Z.m/Z/ddl0m1Z1ddl2m3Z3zddl4m5Z5Wne6yddl4m7Z5Ynwej8dej9dej:dej;dejGd"d#d#ej?Z@e(ed$d$d%d&ZAe(ed$d'd(d)ZBe(ed*d+d,d+ZCe(ed-d.d/d.ZDe(ed0d0d1d0ZEGd2d3d3e jFZGGd4d5d5eGZHGd6d7d7eGZIe jJKe+eHe jJKe,eIGd8d9d9eZLGd:d;d;ejMZNGdd?d?e jFZQe jJKe.e jJKe/e jJKe-eQd@S)A)absolute_importunicode_literals)loads)forms)settings)admin)helpers)main)Select)render_to_response)RequestContext)escape) ugettext_lazy) current_app)states) broadcastrevoke rate_limit)cached_property)abbrtask)action display_field fixedwidth) TaskState WorkerState PeriodicTaskIntervalScheduleCrontabSchedule) naturaldate)is_database_scheduler) force_text) force_unicodegreenredmagentayelloworangeRECEIVEDbluegray)ONLINEOFFLINEcseZdZfddZZS) MonitorListcs$tt|j|i||jj|_dSN)superr-__init__ model_adminlist_page_titletitle)selfargskwargs __class__@/var/www/html/env/lib/python3.10/site-packages/djcelery/admin.pyr00szMonitorList.__init__)__name__ __module__ __qualname__r0 __classcell__r9r9r7r:r-.sr-statecCs$t|j}t|jd}d||S)Nblack+{1})r r?TASK_STATE_COLORSgetformat)taskr?colorr9r9r: colored_state5s  rGlast_heartbeatcCs$|rdpd}t|}d||S)Nr+r,rA)is_aliveNODE_STATE_COLORSrD)noder?rFr9r9r: node_state<s rLETAetacCs|jsdSt|jS)Nz&none)rNr rEr9r9r:rNCs whentstampcCs dtt|jtt|jS)Nz
{1}
)rDr strrQrrOr9r9r:rQJsnamecCs"t|jd}dt|jt|S)Nz!
{1}
)rrSrDr )rE short_namer9r9r:rSQs csHeZdZdZdZddZd fdd Zd fdd Zfd d ZZ S) ModelMonitorFcKstSr.)r-)r4requestr6r9r9r:get_changelist]zModelMonitor.get_changelistNcs,|pi}|d|jtt|j|||dS)Nr3) extra_context) setdefault detail_titler/rV change_view)r4rW object_idrZr7r9r:r]`s  zModelMonitor.change_viewcs|jsdStt|||SNF) can_deleter/rVhas_delete_permission)r4rWobjr7r9r:ragsz"ModelMonitor.has_delete_permissioncs|jsdStt||Sr_)can_addr/rVhas_add_permission)r4rWr7r9r:rdlszModelMonitor.has_add_permissionr.) r;r<r=rcr`rXr]rardr>r9r9r7r:rVYsrVcseZdZedZedZdZdZddddfd d d d ffZe d eddde e e ddde ddde e dfZdZdZdZgdZGdddZeedddZeedd d!Zeed"d#d$Zeed%d&d'Zfd(d)Zfd*d+ZZS), TaskMonitorz Task detailTasksz djcelery/confirm_rate_limit.htmlrQN) r?task_idrSr5r6rNruntimeworkerrQ) extraprettyfieldsclassesDetails)collapserj)result tracebackexpires)rmrlrgUUID)rSptr5T)prettyr6ri) r?rgrSr5r6rNrhrirprqrrrQ)r?rSrQrNri)rSrgr5r6worker__hostname) revoke_tasksterminate_tasks kill_tasksrate_limit_tasksc@seZdZddiZdS)zTaskMonitor.Mediaall)zdjcelery/style.cssN)r;r<r=cssr9r9r9r:Medias r~zRevoke selected taskscCsFt}|D] }t|j|dqWddS1swYdS)N connectionrdefault_connectionrrgr4rWquerysetrr?r9r9r:rxs "zTaskMonitor.revoke_taskszTerminate selected taskscCsHt}|D] }t|j|ddqWddS1swYdS)NT)r terminaterrr9r9r:rys "zTaskMonitor.terminate_taskszKill selected taskscCsJt}|D] }t|j|dddqWddS1swYdS)NTKILL)rrsignalrrr9r9r:rzs "zTaskMonitor.kill_taskszRate limit selected tasksc Cstdd|D}|jj}|j}|jdr>|jd}t}|D] }t|||dq"WddS1s7wYdSt d|t |j t j ||d} t|j| t|dS) NcSg|]}|jqSr9)rS).0rEr9r9r: z0TaskMonitor.rate_limit_tasks..postrrzRate limit selection)r3r object_nameaction_checkbox_nameopts app_label)context_instance)setmodel_metarPOSTrCrrr_r! verbose_namerACTION_CHECKBOX_NAMEr rate_limit_confirmation_templater ) r4rWrtasksrrrater task_namecontextr9r9r:r{s0     zTaskMonitor.rate_limit_tasksc tt||}|dd|SNdelete_selected)r/re get_actionspopr4rWactionsr7r9r:r zTaskMonitor.get_actionscstt||}|dS)Nri)r/re get_querysetselect_relatedr4rWqsr7r9r:rs zTaskMonitor.get_queryset)r;r<r=rr\r2rdate_hierarchy fieldsetsrrGrSrNrQ list_displayreadonly_fields list_filter search_fieldsrr~rrxryrzr{rrr>r9r9r7r:rersL             recseZdZdZedZedZdefZdZ gdZ e eddd Z e ed d d Z e ed ddZfddZZS) WorkerMonitorTz Node detailz Worker Nodeshostname)rH)shutdown_nodes enable_eventsdisable_eventszShutdown selected worker nodescCtddd|DddS)NshutdowncSrr9rrnr9r9r:rrz0WorkerMonitor.shutdown_nodes.. destinationrr4rWrr9r9r:rszWorkerMonitor.shutdown_nodesz%Enable event mode for selected nodes.cCr)NrcSrr9rrr9r9r:rrz/WorkerMonitor.enable_events..rrrr9r9r:r  zWorkerMonitor.enable_eventsz&Disable event mode for selected nodes.cCr)NrcSrr9rrr9r9r:rrz0WorkerMonitor.disable_events..rrrr9r9r:rrzWorkerMonitor.disable_eventscrr)r/rrrrr7r9r:rrzWorkerMonitor.get_actions)r;r<r=rcrr\r2rLrrrrrrrrr>r9r9r7r:rs      rc@sBeZdZeZdZddZeddZej ddZe ddZ dS) TaskSelectWidgetNcCs2|j}ttdd|jjD}dtt||S)Ncss|] }|ds|VqdS)zcelery.N) startswith)rrSr9r9r: sz4TaskSelectWidget.tasks_as_choices..))r)_moduleslistsorted celery_apprtuplezip)r4rrr9r9r:tasks_as_choicessz!TaskSelectWidget.tasks_as_choicescCs|jdur ||_|jSr.)_choicesrr4r9r9r:choicess  zTaskSelectWidget.choicescCsdSr.r9)r4rr9r9r:rscCs|jjdSr.)rloaderimport_default_modulesrr9r9r:r szTaskSelectWidget._modules) r;r<r=rrrrpropertyrsetterrrr9r9r9r:rs  rc@seZdZeZddZdS)TaskChoiceFieldcCsdS)NTr9)r4valuer9r9r: valid_valuerYzTaskChoiceField.valid_valueN)r;r<r=rwidgetrr9r9r9r:rs rcsfeZdZeedddZejeddddZGdddZ fd d Z d d Z d dZ ddZ ZS)PeriodicTaskFormzTask (registered)F)labelrequiredz Task (custom))rr max_lengthc@seZdZeZdZdS)zPeriodicTaskForm.Metar9N)r;r<r=rrexcluder9r9r9r:MetasrcsTtt|}|d}|r||d<|ds(ttd}||j|j d<||S)NregtaskrEzNeed name of task) r/rcleanrCrValidationErrorr error_classmessages_errors)r4datarexcr7r9r:r"s zPeriodicTaskForm.cleanc CsD|j|}zt|W|Sty!}z ttd|d}~ww)NzUnable to parse JSON: %s) cleaned_datar ValueErrorrrr)r4fieldrrr9r9r: _clean_json-s   zPeriodicTaskForm._clean_jsoncC |dS)Nr5rrr9r9r: clean_args7 zPeriodicTaskForm.clean_argscCr)Nr6rrr9r9r: clean_kwargs:rzPeriodicTaskForm.clean_kwargs)r;r<r=rrrr CharFieldrErrrrrr>r9r9r7r:rs    rcsjeZdZeZeZdZddddfddddfdd d dfd d d dffZdfd d Z fddZ Z S)PeriodicTaskAdmin) __unicode__enabledN)rSrrEr)rjwiderkSchedule)intervalcrontab Arguments)r5r6)rjrrozExecution Options)rrqueueexchange routing_keycs4|pi}ttdd}t| |d<tt|||S)NCELERYBEAT_SCHEDULERwrong_scheduler)getattrrr r/rchangelist_view)r4rWrZ schedulerr7r9r:rUs   z!PeriodicTaskAdmin.changelist_viewcstt||}|ddS)Nrr)r/rrrrr7r9r:r\s zPeriodicTaskAdmin.get_querysetr.) r;r<r=rformrrrrrrr>r9r9r7r:r>s.rN)R __future__rranyjsonrdjangor django.confrdjango.contribrdjango.contrib.adminrdjango.contrib.admin.viewsr main_viewsdjango.forms.widgetsr django.shortcutsr django.templater django.utils.htmlr django.utils.translationrrceleryrrcelery.task.controlrrr celery.utilsrcelery.utils.textr admin_utilsrrrmodelsrrrrrhumanizerutilsr django.utils.encodingr! ImportErrorr"SUCCESSFAILUREREVOKEDSTARTEDRETRYrBrJ ChangeListr-rGrLrNrQrS ModelAdminrVrersiteregisterr ChoiceFieldr ModelFormrrr9r9r9r:sv                            `& #