o
    [hL                     @   sL  d Z ddlZddlZddlmZ ddlmZ ddlmZ ddl	m
Z
 ddlmZmZ ddlmZ dd	lmZ dd
lmZ ddlmZ ddlmZ ddlmZ ddlmZ G dd dejZG dd dejZG dd deZ G dd dej!Z"G dd de"Z#G dd dej$Z%G dd dej&Z'G dd dej(Z)dd  Z*G d!d" d"ej+Z,G d#d$ d$e,Z-G d%d& d&ej.Z/G d'd( d(ej0Z1G d)d* d*ej+Z2G d+d, d,ej3Z4G d-d. d.ej5Z6G d/d0 d0ej7Z8G d1d2 d2e8Z9G d3d4 d4ej+Z:d5d6 d7D Z;e;<d8d9d: d;d< Z=G d=d> d>Z>G d?d@ d@e>ej?Z@G dAdB dBe>ejZAdS )Cz8
Form Widget classes specific to the Django admin site.
    N)forms)settings)ValidationError)URLValidator)CASCADE	UUIDField)reverse)NoReverseMatch)smart_urlquote)	urlencode)	Truncator)get_language)gettextc                       s<   e Zd ZdZG dd dZd
 fdd	Z fdd	Z  ZS )FilteredSelectMultiplez
    A SelectMultiple with a JavaScript filter interface.

    Note that the resulting JavaScript assumes that the jsi18n
    catalog has been loaded in the page
    c                   @   s   e Zd Zg dZdS )zFilteredSelectMultiple.Media)zadmin/js/core.jszadmin/js/SelectBox.jszadmin/js/SelectFilter2.jsN__name__
__module____qualname__js r   r   /home/ubuntu/experiments/live_experiments/Pythonexperiments/Otree/venv/lib/python3.10/site-packages/django/contrib/admin/widgets.pyMedia   s    r   Nr   c                    s   || _ || _t || d S N)verbose_name
is_stackedsuper__init__)selfr   r   attrschoices	__class__r   r   r   %   s   zFilteredSelectMultiple.__init__c                    sj   t  |||}d|d d d< | jr|d d d  d7  < | j|d d d< t| j|d d d< |S )NZselectfilterwidgetr   classZstackeddata-field-namezdata-is-stacked)r   get_contextr   r   intr   namevaluer   contextr    r   r   r%   *   s   z"FilteredSelectMultiple.get_context)Nr   )r   r   r   __doc__r   r   r%   __classcell__r   r   r    r   r      s
    r   c                       ,   e Zd ZG dd dZd fdd	Z  ZS )BaseAdminDateWidgetc                   @      e Zd ZddgZdS )zBaseAdminDateWidget.Mediaadmin/js/calendar.js#admin/js/admin/DateTimeShortcuts.jsNr   r   r   r   r   r   5       r   Nc                    &   ddd|pi }t  j||d d S )NZ
vDateField10r#   sizer   formatr   r   r   r   r8   r    r   r   r   ;      zBaseAdminDateWidget.__init__NNr   r   r   r   r   r,   r   r   r    r   r.   4       r.   c                   @      e Zd ZdZdS )AdminDateWidgetzadmin/widgets/date.htmlNr   r   r   template_namer   r   r   r   r@   @       r@   c                       r-   )BaseAdminTimeWidgetc                   @   r/   )zBaseAdminTimeWidget.Mediar0   r1   Nr   r   r   r   r   r   E   r2   r   Nc                    r3   )NZ
vTimeField8r5   r7   r9   r:   r    r   r   r   K   r;   zBaseAdminTimeWidget.__init__r<   r=   r   r   r    r   rD   D   r>   rD   c                   @   r?   )AdminTimeWidgetzadmin/widgets/time.htmlNrA   r   r   r   r   rF   P   rC   rF   c                       s.   e Zd ZdZdZdddZ fddZ  ZS )	AdminSplitDateTimezF
    A SplitDateTime Widget that has some admin-specific styling.
    z!admin/widgets/split_datetime.htmlNc                 C   s   t tg}tj| || d S r   )r.   rD   r   ZMultiWidgetr   )r   r   Zwidgetsr   r   r   r   [   s   zAdminSplitDateTime.__init__c                    s,   t  |||}td|d< td|d< |S )NzDate:Z
date_labelzTime:Z
time_label)r   r%   _r'   r    r   r   r%   a   s   zAdminSplitDateTime.get_contextr   )r   r   r   r+   rB   r   r%   r,   r   r   r    r   rG   T   s
    
rG   c                   @   r?   )AdminRadioSelectzadmin/widgets/radio.htmlNrA   r   r   r   r   rI   h   rC   rI   c                   @   r?   )AdminFileWidgetz'admin/widgets/clearable_file_input.htmlNrA   r   r   r   r   rJ   l   rC   rJ   c                 C   s~   i }| r=t | dr=|  D ]/\}}t|r| }t|ttfr*ddd |D }nt|tr4d| }nt|}|||< q|S )z
    Convert the type of lookups specified in a ForeignKey limit_choices_to
    attribute to a dictionary of query parameters
    items,c                 s       | ]}t |V  qd S r   str.0xr   r   r   	<genexpr>{       z.url_params_from_lookup_dict.<locals>.<genexpr>)01)	hasattrrK   callable
isinstancetuplelistjoinboolrO   )Zlookupsparamskvr   r   r   url_params_from_lookup_dictp   s   


ra   c                       sJ   e Zd ZdZdZd fdd	Z fddZdd	 Zd
d Zdd Z	  Z
S )ForeignKeyRawIdWidgetzj
    A Widget for displaying ForeignKeys in the "raw_id" interface rather than
    in a <select> box.
    z%admin/widgets/foreign_key_raw_id.htmlNc                    s"   || _ || _|| _t | d S r   )rel
admin_sitedbr   r   )r   rc   rd   r   usingr    r   r   r      s   zForeignKeyRawIdWidget.__init__c           	         s   t  |||}| jj}| j|rStd|jj|jj	f | jj
d}|  }|r0|dt| 7 }||d< td|d< d}t| j trH|d7 }|d	 d
 d| nd |d< |d	 d rj| |\|d< |d< |S d |d< |S )Nzadmin:%s_%s_changelist)current_app?related_urlLookupZ
link_titleZvForeignKeyRawIdAdminFieldz vUUIDFieldr"   r   r#   r)   Z
link_labelZlink_url)r   r%   rc   modelrd   is_registeredr   _meta	app_label
model_namer(   url_parametersr   rH   rY   get_related_fieldr   
setdefaultlabel_and_url_for_value)	r   r(   r)   r   r*   Zrel_tori   r^   Z	css_classr    r   r   r%      s8   	z!ForeignKeyRawIdWidget.get_contextc                 C   s   | j j}t|r| }t|S r   )rc   limit_choices_torX   ra   )r   rt   r   r   r   base_url_parameters   s   z)ForeignKeyRawIdWidget.base_url_parametersc                 C   s.   ddl m} |  }||| j ji |S )Nr   )TO_FIELD_VAR)django.contrib.admin.views.mainrv   ru   updaterc   rq   r(   )r   rv   r^   r   r   r   rp      s   z$ForeignKeyRawIdWidget.url_parametersc              
   C   s   | j  j}z| j jj| jjdi ||i}W n t| j jj	t
fy)   Y dS w ztd| jj|jj|jjf |jfd}W n tyJ   d}Y nw t|d|fS )N rz   z%s:%s_%s_change)argsrz      r   )rc   rq   r(   rk   Z_default_managerrf   re   get
ValueErrorZDoesNotExistr   r   rd   rm   rn   ro   pkr	   r   words)r   r)   keyobjurlr   r   r   rs      s(   &
	z-ForeignKeyRawIdWidget.label_and_url_for_valuer<   )r   r   r   r+   rB   r   r%   ru   rp   rs   r,   r   r   r    r   rb      s    "rb   c                       sD   e Zd ZdZdZ fddZdd Zdd Zd	d
 Zdd Z	  Z
S )ManyToManyRawIdWidgetzv
    A Widget for displaying ManyToMany ids in the "raw_id" interface rather than
    in a <select multiple> box.
    z&admin/widgets/many_to_many_raw_id.htmlc                    s4   t  |||}| j| jjrd|d d d< |S )NZvManyToManyRawIdAdminFieldr"   r   r#   )r   r%   rd   rl   rc   rk   r'   r    r   r   r%      s   z!ManyToManyRawIdWidget.get_contextc                 C   s   |   S r   )ru   r   r   r   r   rp      s   z$ManyToManyRawIdWidget.url_parametersc                 C   s   dS )Nry   r   r   r)   r   r   r   rs      s   z-ManyToManyRawIdWidget.label_and_url_for_valuec                 C   s   | |}|r|dS d S )NrL   )r}   split)r   datafilesr(   r)   r   r   r   value_from_datadict   s   

z)ManyToManyRawIdWidget.value_from_datadictc                 C   s   |rd dd |D S dS )NrL   c                 s   rM   r   rN   rQ   r`   r   r   r   rS      rT   z5ManyToManyRawIdWidget.format_value.<locals>.<genexpr>rz   )r\   r   r   r   r   format_value   s   z"ManyToManyRawIdWidget.format_value)r   r   r   r+   rB   r%   rp   rs   r   r   r,   r   r   r    r   r      s    r   c                   @   s   e Zd ZdZdZ				dddZdd Zed	d
 Zedd Z	edd Z
e
jdd Z
dd Zdd Zdd Zdd Zdd ZdS )RelatedFieldWidgetWrapperzd
    This class is a wrapper to a given widget to add the add icon for the
    admin interface.
    z)admin/widgets/related_widget_wrapper.htmlNFc           
      C   s   |j | _ |j| _|| _|| _|d u r||j}|| _t|dd}t|t	s+d| jd< | o/|| _
t|dd tu }	| o@|	 o@|| _| oF|| _|| _d S )Nallow_multiple_selectedFzavailable-sourcezdata-contextZ	on_delete)Zneeds_multipart_formr   r"   rc   rl   rk   can_add_relatedgetattrrY   AutocompleteMixincan_change_relatedr   can_delete_relatedcan_view_relatedrd   )
r   r"   rc   rd   r   r   r   r   ZmultipleZcascader   r   r   r      s   



z"RelatedFieldWidgetWrapper.__init__c                 C   s4   t  | }t | j||_| jj|_||t| < |S r   )copydeepcopyr"   r   id)r   memor   r   r   r   __deepcopy__  s
   

z&RelatedFieldWidgetWrapper.__deepcopy__c                 C      | j jS r   )r"   	is_hiddenr   r   r   r   r   $     z#RelatedFieldWidgetWrapper.is_hiddenc                 C   r   r   )r"   mediar   r   r   r   r   (  r   zRelatedFieldWidgetWrapper.mediac                 C   r   r   r"   r   r   r   r   r   r   ,  r   z!RelatedFieldWidgetWrapper.choicesc                 C   s   || j _d S r   r   r   r   r   r   r   0  s   c                 G   s   t d||f  | jj|dS )Nzadmin:%s_%s_%s)rg   r{   )r   rd   r(   )r   infoactionr{   r   r   r   get_related_url4  s
   z)RelatedFieldWidgetWrapper.get_related_urlc                 C   s   ddl m}m} | jjj}|j|jf}| j j	}d
dd ||f|dffD }	| j|||| j||	|j|j| j| j| j| j| jjd}
| jrQ| |d|
d	< | jr]| |d
d|
d< | jsc| jru| d| |
d< | |dd|
d< |
S )Nr   )IS_POPUP_VARrv   &c                 s   s    | ]}d | V  qdS )z%s=%sNr   )rQ   paramr   r   r   rS   A  s
    
z8RelatedFieldWidgetWrapper.get_context.<locals>.<genexpr>   )Zrendered_widgetr   r(   
url_paramsrk   ro   r   r   r   r   Zmodel_has_limit_choices_toaddZadd_related_urldeleteZ__fk__Zdelete_related_template_url=Zview_related_url_paramsZchangeZchange_related_template_url)rw   r   rv   rc   rk   rm   rn   ro   rq   r(   r\   r"   renderr   r   r   r   r   r   rt   r   )r   r(   r)   r   r   rv   Zrel_optsr   Zrelated_field_namer   r*   r   r   r   r%   ;  sB   

z%RelatedFieldWidgetWrapper.get_contextc                 C      | j |||S r   )r"   r   r   r   r   r(   r   r   r   r   b     z-RelatedFieldWidgetWrapper.value_from_datadictc                 C   r   r   )r"   value_omitted_from_datar   r   r   r   r   e  r   z1RelatedFieldWidgetWrapper.value_omitted_from_datac                 C   s   | j |S r   )r"   id_for_label)r   Zid_r   r   r   r   h  s   z&RelatedFieldWidgetWrapper.id_for_label)NFFF)r   r   r   r+   rB   r   r   propertyr   r   r   setterr   r%   r   r   r   r   r   r   r   r      s,    




'r   c                          e Zd Zd fdd	Z  ZS )AdminTextareaWidgetNc                       t  jddi|p	i d d S )Nr#   ZvLargeTextFieldr   r9   r   r   r    r   r   r   m     zAdminTextareaWidget.__init__r   r   r   r   r   r,   r   r   r    r   r   l      r   c                       r   )AdminTextInputWidgetNc                    r   Nr#   Z
vTextFieldr   r9   r   r    r   r   r   r  r   zAdminTextInputWidget.__init__r   r   r   r   r    r   r   q  r   r   c                       r   )AdminEmailInputWidgetNc                    r   r   r9   r   r    r   r   r   w  r   zAdminEmailInputWidget.__init__r   r   r   r   r    r   r   v  r   r   c                       s2   e Zd ZdZdef fdd	Z fddZ  ZS )AdminURLFieldWidgetzadmin/widgets/url.htmlNc                    s&   t  jddi|p	i d | | _d S )Nr#   Z	vURLFieldr   )r   r   	validator)r   r   Zvalidator_classr    r   r   r   ~  s   zAdminURLFieldWidget.__init__c                    s   z|  |r|nd d}W n ty   d}Y nw t |||}td|d< td|d< |r7t|d d	 nd|d d
< ||d< |S )Nrz   TFz
Currently:Zcurrent_labelzChange:Zchange_labelr"   r)   href	url_valid)r   r   r   r%   rH   r
   )r   r(   r)   r   r   r*   r    r   r   r%     s   
zAdminURLFieldWidget.get_context)r   r   r   rB   r   r   r%   r,   r   r   r    r   r   {  s    r   c                       s"   e Zd ZdZd fdd	Z  ZS )AdminIntegerFieldWidgetZvIntegerFieldNc                    s    t  jd| ji|p
i d d S )Nr#   r   )r   r   
class_namer   r    r   r   r     s    z AdminIntegerFieldWidget.__init__r   )r   r   r   r   r   r,   r   r   r    r   r     s    r   c                   @   r?   )AdminBigIntegerFieldWidgetZvBigIntegerFieldN)r   r   r   r   r   r   r   r   r     rC   r   c                       r   )AdminUUIDInputWidgetNc                    r   )Nr#   Z
vUUIDFieldr   r9   r   r    r   r   r     r   zAdminUUIDInputWidget.__init__r   r   r   r   r    r   r     r   r   c                 C   s   i | ]}|  |qS r   )lowerrP   r   r   r   
<dictcomp>  s    r   )-arazbgcacsdadeeleneseteufafifrglhehihrhur   isitjakmkoltlvmkmsnbnlplzpt-BRptroruskzsr-Cyrlsrsvthtrukvizzh-CNzzh-TW)zzh-hanszzh-hantc                  C   sn   t  } t| }|d u r5| d ur5d }| dd| }dkr5t| d |  }r*|S | dd| }dks|S )N-r   )r   SELECT2_TRANSLATIONSr}   rfind)Z	lang_codeZsupported_codeir   r   r   get_select2_language  s   
r   c                       sN   e Zd ZdZdZdddZdd Zd fd	d
	ZdddZe	dd Z
  ZS )r   z
    Select widget mixin that loads options from AutocompleteJsonView via AJAX.

    Renders the necessary data attributes for select2 and adds the static form
    media.
    z%s:autocompleteNr   c                 C   s:   || _ || _|| _|| _|d u ri n| | _t | _d S r   )fieldrd   re   r   r   r   r   	i18n_name)r   r   rd   r   r   rf   r   r   r   r     s   zAutocompleteMixin.__init__c                 C   s   t | j| jj S r   )r   url_namerd   r(   r   r   r   r   get_url  s   zAutocompleteMixin.get_urlc                    s   t  j||d}|dd |ddd|  | jjjj| jjjj	| jj
dt| j d| j|d |d r6dnd d d	 |S )
z
        Set select2's AJAX attributes.

        Attributes can be set using the html5 data attribute.
        Nested attributes require a double dash as per
        https://select2.org/configuration/data-attributes#nested-subkey-options
        )extra_attrsr#   rz   true   GETzadmin-autocomplete )zdata-ajax--cachezdata-ajax--delayzdata-ajax--typezdata-ajax--urlzdata-app-labelzdata-model-namer$   z
data-themezdata-allow-clearzdata-placeholderlangr#   )r   build_attrsrr   rx   r   r   rk   rm   rn   ro   r(   jsondumpsis_requiredr   )r   Z
base_attrsr   r   r    r   r   r    s,   

zAutocompleteMixin.build_attrsc              
      s  dg df}|g}d} fdd|D } j s' js'|d  |dddd  jjjj}t jjd|j	j
|j
 fd	d
 jj jjdi d |iD }	|	D ]+\}
}t|
|v oi|du pi j}||O }t|d }|d }| ||
||| qY|S )z9Return selected options based on the ModelChoiceIterator.Nr   Fc                    s&   h | ]}t | jjjvrt |qS r   )rO   r   r   Zempty_valuesr   r   r   r   	<setcomp>  s     z.AutocompleteMixin.optgroups.<locals>.<setcomp>r   rz   
field_namec                 3   s(    | ]}t | jj|fV  qd S r   )r   r   r   Zlabel_from_instance)rQ   r   r   Zto_field_namer   r   rS   )  s
    
z.AutocompleteMixin.optgroups.<locals>.<genexpr>z%s__inr   )r  r   appendZcreate_optionr   Zremote_fieldrk   rm   r   r   Zattname	get_fieldr   Zquerysetrf   re   filterrO   len)r   r(   r)   attrdefaultgroupsZhas_selectedZselected_choicesZremote_model_optsr   Zoption_valueZoption_labelselectedindexZsubgroupr   r  r   	optgroups  s>   



zAutocompleteMixin.optgroupsc                 C   sR   t jrdnd}| jrd| j fnd}tjd| d| f| d dd	| d
fidS )Nrz   z.minz"admin/js/vendor/select2/i18n/%s.jsr   z"admin/js/vendor/jquery/jquery%s.jsz)admin/js/vendor/select2/select2.full%s.js)zadmin/js/jquery.init.jszadmin/js/autocomplete.jsscreenz&admin/css/vendor/select2/select2%s.csszadmin/css/autocomplete.css)r   css)r   DEBUGr   r   r   )r   extraZ	i18n_filer   r   r   r   =  s&   
zAutocompleteMixin.media)Nr   Nr   )r   r   r   r+   r   r   r   r  r  r   r   r,   r   r   r    r   r     s    

#r   c                   @      e Zd ZdS )AutocompleteSelectNr   r   r   r   r   r   r   r  X      r  c                   @   r  )AutocompleteSelectMultipleNr  r   r   r   r   r  \  r  r  )Br+   r   r  djangor   Zdjango.confr   Zdjango.core.exceptionsr   Zdjango.core.validatorsr   Zdjango.db.modelsr   r   Zdjango.urlsr   Zdjango.urls.exceptionsr	   Zdjango.utils.htmlr
   Zdjango.utils.httpr   Zdjango.utils.textr   Zdjango.utils.translationr   r   rH   ZSelectMultipler   Z	DateInputr.   r@   Z	TimeInputrD   rF   ZSplitDateTimeWidgetrG   ZRadioSelectrI   ZClearableFileInputrJ   ra   Z	TextInputrb   r   ZWidgetr   ZTextarear   r   Z
EmailInputr   ZURLInputr   ZNumberInputr   r   r   r   rx   r   r   ZSelectr  r  r   r   r   r   <module>   sT    Tv2q