a gh2@sVddlZddlmZddlmZGdddejZGdddeZGdd d eZdS) N)datetime)dbcsXeZdZdZdddgdZejee ddddZ ddZ fd d Z d d Z ZS) UniqueIDMixina A mixin class that provides a unique identifier for the document. Base class for MongoDB documents with an automatically generated unique public ID. Attributes: public_id (str): A unique identifier for the document. Methods: generate_public_id(): Generates a random UUID and assigns it to the public_id field. to_dict(): Converts the document to a dictionary for serialization. T public_id)allow_inheritanceabstractindexesz#Unique identifier for the document.)defaultuniquerequired help_textcCstt|_dS)zQ Generates a random UUID and assigns it to the public_id field. N)struuiduuid4rselfr/app/app/models/base.pygenerate_public_id#sz UniqueIDMixin.generate_public_idcs(tt|j|i||js$|dS)z Initialize a new ExtendedModel instance. Args: *args: Additional positional arguments. **kwargs: Additional keyword arguments. N)superr__init__rrrargskwargs __class__rrr*s zUniqueIDMixin.__init__cCs6|}t|j|d<|dd|dd|S) Converts the document to a dictionary for serialization. Returns: dict: A dictionary representation of the document. id_idN_cls)to_mongoto_dictr rpoprZ document_dictrrrr!8s    zUniqueIDMixin.to_dict)__name__ __module__ __qualname____doc__metar StringFieldr rrrrrr! __classcell__rrrrrs  rcseZdZdZejdddZejej ddZ ejddZ ejdd dZ fd d Z d d ZddZddZfddZddZeddZdddgdZZS)TimestampedMixinzN A mixin class that adds timestamps for document creation and update. Fz%Indicates if the document is deleted.)r r z Timestamp for document creation.z'Timestamp for the last document update.)r Nz Timestamp for document deletion.cstj|i|dS)z% Saves the document. N)rsaverrrrr,^szTimestampedMixin.savecCst|_|dS)zh Updates the document and the document's `updated_at` field with the current timestamp. N)rnow updated_atr,rrrrupdateds zTimestampedMixin.updatecCsd|_t|_|dS)z Soft delete the document by setting the 'deleted' flag and timestamp. Archives the document by setting the `deleted` field to True and updating the `deleted_at` field with the current timestamp. TN)deletedrr- deleted_atr,rrrrarchiveks zTimestampedMixin.archivecCsd|_d|_|dS)z Restore a soft-deleted document. Restores the document by setting the `deleted` field to False and clearing the `deleted_at` field. FN)r0r1r,rrrrrestoretszTimestampedMixin.restorecsZtt|}||d|d<||d|d<||d|d<|dd|S)z7 Convert the document to a dictionary. created_atr.r1rN)rr+r!_format_datetimegetr"r#rrrr!}s  zTimestampedMixin.to_dictcCs|rt|tr|SdS)zO Helper method to format a datetime object to ISO 8601 string. N) isinstancer isoformat)rdtrrrr5sz!TimestampedMixin._format_datetimecCsbtd|}ttd|d}|jdd}|jdd|d||}|||dd|DdS)z4 Get a paginated list of documents. dF)r0cSsg|] }|qSr)r!).0docrrr z2TimestampedMixin.get_paginated..)totalpageper_pageitems)maxminobjectscountskiplimit)clsrArBr@ documentsrrr get_paginateds   zTimestampedMixin.get_paginatedTz -created_atz -updated_at)rordering)r$r%r&r'r BooleanFieldr0 DateTimeFieldrr-r4r.r1r,r/r2r3r!r5 classmethodrLr(r*rrrrr+Hs6    r+cs>eZdZdZeeZddZddZ fddZ Z S) TaggedMixinax Base class for MongoDB documents with tags. Attributes: tags (list): A list of tags associated with the document. Methods: add_tag(tag): Adds a tag to the document if it doesn't already exist. remove_tag(tag): Removes a tag from the document if it exists. to_dict(): Converts the document to a dictionary for serialization. cCs||jvr|j|dS)z Adds a tag to the document if it doesn't already exist. Args: tag (str): The tag to add. N)tagsappendrtagrrradd_tags zTaggedMixin.add_tagcCs||jvr|j|dS)zy Removes a tag from the document if it exists. Args: tag (str): The tag to remove. N)rRremoverTrrr remove_tags zTaggedMixin.remove_tagcstt|}|dd|S)rrN)rrQr!r"r#rrrr!s zTaggedMixin.to_dict) r$r%r&r'r ListFieldr)rRrVrXr!r*rrrrrQs    rQ)rrapp.config.extensionsrDynamicDocumentrr+rQrrrrs   A`