3 g8Q@s^dZddlZddlZyddlZdZWnek r<dZYnXyddlZdZWnek rfdZYnXddlm Z ddl m Z ddl m Z mZmZmZmZddlmZdd lmZdd lmZdd lmZee d d ZGdddZGdddeeZGdddeeZGddde ZGdddeZ Gddde!Z"GdddZ#Gddde#eZ$dS)zo Django Extensions additional model fields Some fields might require additional dependencies to be installed. NTF)settings)ImproperlyConfigured) DateTimeField CharField SlugFieldQUniqueConstraint) LOOKUP_SEP)slugify)get_random_string) force_strZ$EXTENSIONS_MAX_UNIQUE_QUERY_ATTEMPTSdc@s0eZdZddZeddZddZddZd S) UniqueFieldMixincCs"tt||tstdj|dS)Nz#'{}' argument must be True or False) isinstancegetattrbool ValueErrorformat)selfattrnamerb/var/www/tester-filtro-web/env/lib/python3.6/site-packages/django_extensions/db/fields/__init__.py check_is_bool%szUniqueFieldMixin.check_is_boolcsfddjjDS)Ncs>g|]6}|j s"|js"|jr|jr||jkr4|jndfqS)N)Z is_relationZ one_to_oneZ many_to_oneZ related_modelmodel).0f) model_clsrr ,sz0UniqueFieldMixin._get_fields..)_metaZ get_fields)rr)rr _get_fields)szUniqueFieldMixin._get_fieldscCs8x,|j|D]\}}|r ||kr |jjSq W|jjS)N)rZ_default_managerall)rr slug_fieldfieldrrrr get_queryset0s zUniqueFieldMixin.get_querysetcsjj|}jr"|jjd}i}x8jjD],}j|kr0x|D]}t|d||<qDWq0Wt} tjdd} | rt dd| } x<| D]4} j| j krfdd| j D} | tf| M} qWt |}||j<x,| s|j | f|rt |}||j<qWt j||S)N)pk constraintscSs t|tS)N)rr)crrrIsz.UniqueFieldMixin.find_unique..cs$i|]}|jkrt|d|qS)N)attnamer)rr")model_instancerrr Msz0UniqueFieldMixin.find_unique..) r# __class__r$excluderunique_togetherr(rrfilterfieldsnextsetattr)rr)r"iteratorargsZquerysetkwargsparamsparamqueryr%Zunique_constraintsZunique_constraint conditionnewr)r)rr find_unique6s2        zUniqueFieldMixin.find_uniqueN)__name__ __module__ __qualname__r staticmethodrr#r:rrrrr#s rcsheZdZdZfddZddZeddZdd Zd d Z d d Z ddZ ddZ fddZ ZS) AutoSlugFielda AutoSlugField By default, sets editable=False, blank=True. Required arguments: populate_from Specifies which field, list of fields, or model method the slug will be populated from. populate_from can traverse a ForeignKey relationship by using Django ORM syntax: populate_from = 'related_model__field' Optional arguments: separator Defines the used separator (default: '-') overwrite If set to True, overwrites the slug on every save (default: False) slugify_function Defines the function which will be used to "slugify" a content (default: :py:func:`~django.template.defaultfilters.slugify` ) It is possible to provide custom "slugify" function with the ``slugify_function`` function in a model class. ``slugify_function`` function in a model class takes priority over ``slugify_function`` given as an argument to :py:class:`~AutoSlugField`. Example .. code-block:: python # models.py from django.db import models from django_extensions.db.fields import AutoSlugField class MyModel(models.Model): def slugify_function(self, content): return content.replace('_', '-').lower() title = models.CharField(max_length=42) slug = AutoSlugField(populate_from='title') Inspired by SmileyChris' Unique Slugify snippet: https://www.djangosnippets.org/snippets/690/ cs|jdd|jdd|jdd}|dkr6tdn||_t|svt|ttfsX|f}tdd|Dsvt d ||jd t |_ |jd d |_ |jd d|_ |jd |jdd|_|jd|jdd|_|jd|jdt|_tj||dS)NblankTeditableF populate_fromz missing 'populate_from' argumentcss|]}t|tVqdS)N)rstr)rerrr sz)AutoSlugField.__init__..zB'populate_from' must be str or list[str] or tuple[str], found `%s`slugify_function separator- overwriteoverwrite_on_addallow_duplicatesmax_unique_query_attempts) setdefaultpopr_populate_fromcallablerlisttupler TypeErrorr rFrGrIrrJrKMAX_UNIQUE_QUERY_ATTEMPTSrLsuper__init__)rr3r4rB)r+rrrVs*        zAutoSlugField.__init__cCs:dtj|j}tjd||j|}tjd||fd|S)a  Clean up a slug by removing slug separator characters that occur at the beginning or end of a slug. If an alternate separator is used, it will also replace any instances of the default '-' separator with the new separator. z(?:-|%s)z%s+z ^%s+|%s+$)reescaperGsub)rvalueZre_seprrr _slug_stripszAutoSlugField._slug_stripcCs|r ||SdS)NrWr)contentrFrrr slugify_funcszAutoSlugField.slugify_funcccs|Vxtt||jD]d}|}d|j|f}t|}|jrft|||jkrf|d|j|}|j|}d||f}|VqWtd||jfdS)Nz%s%sz&max slug attempts for %s exceeded (%s))rangerLrGlenslug_lenr\ RuntimeError)r original_slugstartislugendZend_lenrrrslug_generators   zAutoSlugField.slug_generatorc stj}d}|r j r d}jr.|r.d}|r6|Sj}t|ttfsP|f}jj j}tdj fdd}j j t ||}d}|j_jr|dj}j|}|} jrtj||Sj|j| |S)NFTrFcsjj|dS)N)rF)r^get_slug_fields) lookup_value)r)rrFrrr's z+AutoSlugField.create_slug..)rr(rIrJrOrrQrRr get_fieldrFrGjoinmap max_lengthrar\rKr1r:rh) rr)addrfZuse_existing_slugrBr!Zslug_for_fieldrdrcr)r)rrFr create_slugs4    zAutoSlugField.create_slugc Cs~t|rd||S|jt}|}xD|D]<}yt||}Wq(tk rbtdj||||Yq(Xq(Wt|rzd|S|S)Nz%szbvalue {} in AutoSlugField's 'populate_from' argument {} returned an error - {} has no attribute {})rPsplitr rAttributeErrorr)rr)rjZlookup_value_pathattrelemrrrris    zAutoSlugField.get_slug_fieldscCst|j||}|S)N)r rq)rr)rpr[rrrpre_saveszAutoSlugField.pre_savecCsdS)Nrr)rrrrget_internal_typeszAutoSlugField.get_internal_typecs`tj\}}}}|j|d<|jdks0|j|d<|jdk rBd|d<|jdk rTd|d<||||fS)NrBrHrGFTrIrK)rU deconstructrOrGrIrK)rnamepathr3r4)r+rrrxs     zAutoSlugField.deconstruct)r;r<r=__doc__rVr\r>r^rhrqrirvrwrx __classcell__rr)r+rr?]s6   -r?csLeZdZdZfddZddZddZdd Zd d Zfd d Z Z S)RandomCharFielda2 RandomCharField By default, sets editable=False, blank=True, unique=False. Required arguments: length Specifies the length of the field Optional arguments: unique If set to True, duplicate entries are not allowed (default: False) lowercase If set to True, lowercase the alpha characters (default: False) uppercase If set to True, uppercase the alpha characters (default: False) include_alpha If set to True, include alpha characters (default: True) include_digits If set to True, include digit characters (default: True) include_punctuation If set to True, include punctuation characters (default: False) keep_default If set to True, keeps the default initialization value (default: False) cs |jdd|jdd|jdd|_|jdkr8td|j|d<|jdd|_|jd|jd d|_|jd |jr|jrtd |jd d|_|jd |jd d|_|jd |jd d|_ |jdd|_ |jd |jdt |_ d|krd|d<t j||dS)Nr@TrAFlengthzmissing 'length' argumentro lowercase uppercasez@the 'lowercase' and 'uppercase' arguments are mutually exclusiveinclude_digits include_alphainclude_punctuation keep_defaultrLunique)rMrNr~rrrrrrrrrTrLrUrV)rr3r4)r+rrrVGs.          zRandomCharField.__init__ccs:x&t|jD]}djt|j|Vq Wtd|jdS)NrWz+max random character attempts exceeded (%s))r_rLrmr r~rb)rcharsrerrrrandom_char_generatoresz%RandomCharField.random_char_generatorcCs$x|jjD]}|j|kr dSq WdS)NTF)rr-r()rr)r5rrrin_unique_togetherjs z"RandomCharField.in_unique_togethercCs| s |jr(t||jdkr(t||jSd}|jr`|jrD|tj7}n|jrV|tj7}n |tj 7}|j rp|tj 7}|j r|tj 7}|j|}|j r|j| rt|}t||j||S|j||jj|j|S)NrW)rrr(rrstringascii_lowercaserascii_uppercase ascii_lettersrdigitsr punctuationrrrr0r1r:rrl)rr)rp populationZ random_charsr9rrrrvps,        zRandomCharField.pre_savecCsdS)Nrr)rrrr internal_typeszRandomCharField.internal_typecstj\}}}}|j|d<|d=|jdkr6|j|d<|jdkrJ|j|d<|jdkr^|j|d<|jdkrr|j|d<|jdkr|j|d <|jdkr|j|d <||||fS) Nr~roTrrFrrrr) rUrxr~rrrrrr)rryrzr3r4)r+rrrxs              zRandomCharField.deconstruct) r;r<r=r{rVrrrvrrxr|rr)r+rr}$s! r}cs0eZdZdZddZddZfddZZS)CreationDateTimeFieldzc CreationDateTimeField By default, sets editable=False, blank=True, auto_now_add=True cOs:|jdd|jdd|jddtj|f||dS)NrAFr@T auto_now_add)rMrrV)rr3r4rrrrVs   zCreationDateTimeField.__init__cCsdS)Nrr)rrrrrwsz'CreationDateTimeField.get_internal_typecsTtj\}}}}|jdk r$d|d<|jdk r6d|d<|jdk rHd|d<||||fS)NFTrAr@r)rUrxrAr@r)rryrzr3r4)r+rrrxs   z!CreationDateTimeField.deconstruct)r;r<r=r{rVrwrxr|rr)r+rrsrcs<eZdZdZddZddZfddZfdd ZZS) ModificationDateTimeFieldz ModificationDateTimeField By default, sets editable=False, blank=True, auto_now=True Sets value to now every time the object is saved. cOs"|jddtj|f||dS)Nauto_nowT)rMrrV)rr3r4rrrrVs z"ModificationDateTimeField.__init__cCsdS)Nrr)rrrrrwsz+ModificationDateTimeField.get_internal_typecs0tj\}}}}|jdk r$d|d<||||fS)NFTr)rUrxr)rryrzr3r4)r+rrrxs z%ModificationDateTimeField.deconstructcs&t|ddst||jStj||S)NZupdate_modifiedT)rr(rUrv)rr)rp)r+rrrvs  z"ModificationDateTimeField.pre_save) r;r<r=r{rVrwrxrvr|rr)r+rrs  rc@s eZdZdS)UUIDVersionErrorN)r;r<r=rrrrrsrcsReZdZdZdZdfdd Zdd Zfd d Zfd d ZfddZ Z S)UUIDFieldMixina UUIDFieldMixin By default uses UUID version 4 (randomly generated UUID). The field support all uuid versions which are natively supported by the uuid python module, except version 2. For more information see: https://docs.python.org/lib/module-uuid.html $NTc szts td| jd|j|r8d|_d| d<| jdd||_||_||_||_||_ |p\||_ t j | d|i| dS)NzW'uuid' module is required for UUIDField. (Do you have Python 2.5 or higher installed ?)roFTr@rA verbose_name) HAS_UUIDrrMDEFAULT_MAX_LENGTHZempty_strings_allowedautoversionnode clock_seq namespace uuid_namerUrV) rrryrrrrrrr3r4)r+rrrVs  zUUIDFieldMixin.__init__cCs|j s|jdkrtjS|jdkr4tj|j|jS|jdkrHtdnB|jdkrbtj|j|j S|jdkr|tj |j|j Std|jdS)Nrrkz UUID version 2 is not supported.zUUID version %s is not valid.) ruuiduuid4uuid1rrruuid3rruuid5)rrrr create_uuids     zUUIDFieldMixin.create_uuidcshtj||}|jr>|r>|dkr>t|j}t||j||S|jrd| rdt|j}t||j||S)N)rUrvrr rr1r()rr)rpr[)r+rrrvs   zUUIDFieldMixin.pre_savec s|jr dStjf|S)N)rrU formfield)rr4)r+rrrszUUIDFieldMixin.formfieldcstj\}}}}|jdd|jkr*|d=|jdk r>|j|d<|jdkrR|j|d<|jdk rf|j|d<|jdk rz|j|d<|jdk r|j|d<|j dk r|j |d <||||fS) NroTrrrrrrr) rUrxgetrrrrrrrry)rryrzr3r4)r+rrrx#s             zUUIDFieldMixin.deconstruct)NNTrNNNN) r;r<r=r{rrVrrvrrxr|rr)r+rrs  rcs,eZdZdZdZfddZddZZS)ShortUUIDFieldz ShortUUIDFied Generates concise (22 characters instead of 36), unambiguous, URL-safe UUIDs. Based on `shortuuid`: https://github.com/stochastic-technologies/shortuuid cs,tj||tstd|jd|jdS)Nza'shortuuid' module is required for ShortUUIDField. (Do you have Python 2.5 or higher installed ?)ro)rUrVHAS_SHORT_UUIDrrMr)rr3r4)r+rrrVCszShortUUIDField.__init__cCs~|j s|jdkrtjS|jdkr,tjS|jdkr@tdn:|jdkrTtdn&|jdkrltj|jdStd |jdS) Nrrrkz UUID version 2 is not supported.rz UUID version 3 is not supported.r)ryzUUID version %s is not valid.)r shortuuidrrr)rrrrrIs      zShortUUIDField.create_uuid)r;r<r=r{rrVrr|rr)r+rr8s r)%r{rXrrr ImportErrorrrZ django.confrZdjango.core.exceptionsrZdjango.db.modelsrrrrrZdjango.db.models.constantsr Zdjango.template.defaultfiltersr Zdjango.utils.cryptor Zdjango.utils.encodingr rrTrr?r}rr Exceptionrrrrrrrs<         :HX