a gh$@sxddlmZddlmZmZmZmZmZmZmZm Z ddl m Z ddl m Z ddlmZddlmZGddde Zd S) )datetime) StringField EmailField DateTimeField BooleanField ListFieldReferenceFieldCASCADEEmbeddedDocumentField)bcrypt)TimestampedMixin)Role)Profilecs|eZdZdZedddZeddddZeddd Zed d dZ edd gd d dZ e dddZ e dddZ e dddZe dddZe dddZe dddZe dddZe dddZedddZeejdddZedddZeeddZeeed d!Z d"d#Z!e"d$d%d&Z#e"d$d'd(Z$d)d*Z%d+d,Z&d-d.Z'd/d0Z(d1d2Z)d3d4Z*d5d6Z+d7d8Z,d9d:Z-d;d<Z.fd=d>Z/Z0S)?Usera Professional User model for managing user information. Attributes: username (str): The unique username of the user (required). email (str): The email address of the user (required, unique). password (str): The user's password (required). first_name (str): The user's first name. last_name (str): The user's last name. is_verified (bool): A flag indicating if the user's email is verified. is_superuser (bool): A flag indicating if the user is a superuser. is_admin (bool): A flag indicating if the user is an admin. is_active (bool): A flag indicating if the user account is active. is_staff (bool): A flag indicating if the user is staff. last_login (datetime): The date and time of the user's last login. date_joined (datetime): The date and time of user registration (auto-generated). roles (list of Role): User roles or permissions (related to Role model). last_seen_at (datetime): The date and time when the user was last seen. notif_enabled (bool): A flag indicating if notifications are enabled for the user. is_online (bool): A flag indicating if the user is currently online. is_blocked (bool): A flag indicating if the user is blocked. Methods: __str__(): Return the username as a string representation of the user. set_password(password: str): Set the user's password (typically after hashing). check_password(password: str): Check if the provided password matches the user's password. add_role(role: Role): Add a role to the user's list of roles. remove_role(role: Role): Remove a role from the user's list of roles. mark_as_online(): Mark the user as online. mark_as_offline(): Mark the user as offline. block_user(): Block the user, preventing certain actions. unblock_user(): Unblock the user, allowing them to resume normal activities. update_last_seen(): Update the user's last seen timestamp to the current time. 2z The unique username of the user.) max_length help_textTzThe email address of the user.)requireduniquerzThe user's password.)rrzThe user's Company name.)zAnonymous CompanyzBrands CompanyzAd Agency CompanyzTV Channels CompanyzThe user's Company type.)rrchoicesrFz2A flag indicating if the user's email is verified.)defaultrz-A flag indicating if the user is a superuser.z*A flag indicating if the user is an admin.z0A flag indicating if the user account is active.z'A flag indicating if the user is staff.z2A flag indicating if the user is currently online.z)A flag indicating if the user is blocked.z|js:tjdd}|s,tdd}|||_|dS)Nadminr')r,r objectsfirstr*)rZ client_rolerrrensure_client_roles zUser.ensure_client_rolecCs|j |_|dS)z Toggle the active status of the user. If the user is currently active, this function will deactivate the user. If the user is currently inactive, this function will activate the user. N) is_activer*rrrrtoggle_active_statuss zUser.toggle_active_statuscCs d|_dS)zM Mark the user as online. Returns: None TN is_onlinerrrrmark_as_onlineszUser.mark_as_onlinecCs d|_dS)zN Mark the user as offline. Returns: None FNr6rrrrmark_as_offlineszUser.mark_as_offlinecCs d|_dS)zj Block the user, preventing them from certain actions. Returns: None TN is_blockedrrrr block_userszUser.block_usercCs d|_dS)zq Unblock the user, allowing them to resume normal activities. Returns: None FNr:rrrr unblock_userszUser.unblock_usercCst|_dS)zo Update the user's last seen timestamp to the current time. Returns: None N)rutcnow last_seen_atrrrrupdate_last_seenszUser.update_last_seencCst|_dS)zp Update the user's last login timestamp to the current time. Returns: None N)rr> last_loginrrrrupdate_last_loginszUser.update_last_logincstt|}|dd|dd|dd|ddsV|dd|dd||d|d<||d |d <||d |d <|d d|j|d <|S) z7 Convert the document to a dictionary. idNr r7deletedFZ deleted_at date_joinedrAr?_clsr,)superrto_dictpopgetZ_format_datetimer,)rZ user_dict __class__rrrHs       z User.to_dict)1__name__ __module__ __qualname____doc__rrremailr Z company_nameZ company_typerZ is_verified is_superuseris_adminr4is_staffr7r;Z notif_enabledrrArr>rEr?r rprofilerr r r,rstrr$r&r-r/r3r5r8r9r<r=r@rBrH __classcell__rrrKrr s$           rN)r mongoenginerrrrrrr r app.config.extensionsr app.models.baser app.models.roler app.models.profilerrrrrrs (