o di!@sddlZddlmZddlmZddlmZddlmZddl m Z ddl m Z Gdd d eZ Gd d d ejZGd d d eZGdddeZGdddeZGdddeZGdddeZdS)N)settings)BaseUserManager)ObjectDoesNotExist)models)timezone)send_opt_verification_mailcsHeZdZdZddZd ddZd ddZfd d Zed d Z Z S) UserManagera4 Custom manager for the User model. Methods: create_user(email, password, **extra_fields): Creates and saves a regular user. create_superuser(email, password, **extra_fields): Creates and saves a superuser. lowercase_email(email): Converts an email address to lowercase. cKs:||}|jdd|i|}|||j|jd|S)a Creates and saves a User with the given email and password. Args: email (str): The user's email address. password (str): The user's password. **extra_fields: Additional fields for the user. Returns: User: The created User instance. This method is used to create and save a regular user with the given email and password. email)usingN)lowercase_emailmodel set_passwordsave_db)selfr password extra_fieldsuserr r ]/var/www/html/evchargy.com/django-station-ev/station_ev/apps/account/managers/user_manager.py _create_users  zUserManager._create_userNcKs,|dd|dd|j||fi|S)a Creates and saves a regular user. Args: email (str): The user's email address. password (str): The user's password. **extra_fields: Additional fields for the user. Returns: User: The created regular User instance. This method is used to create and save a regular user with the given email, password, and optional extra fields. By default, regular users are not staff members or superusers. is_staffF is_superuser) setdefaultrrr rrr r r create_user+s  zUserManager.create_usercKsb|dd|dd|ddurtd|ddur"td||}|j||fi|S)a Creates and saves a superuser. Args: email (str): The superuser's email address. password (str): The superuser's password. **extra_fields: Additional fields for the superuser. Returns: User: The created superuser User instance. This method is used to create and save a superuser with the given email, password, and optional extra fields. Superusers have staff and superuser privileges. rTrz"Superuser must have is_staff=True.z&Superuser must have is_superuser=True.)rget ValueErrorr rrr r rcreate_superuser?s   zUserManager.create_superuserctjddS)a  Override the default queryset to filter out removed users. Returns: QuerySet: The filtered queryset of non-removed users. This method overrides the default queryset to filter out users who have been removed (soft deleted). FZ is_removedsuper get_querysetfilterr __class__r rr#Zs zUserManager.get_querysetcCs0|pd}z |}W|StyY|Sw)a; Convert an email address to lowercase. Args: email (str): The email address to convert. Returns: str: The lowercase email address. This method is used to convert an email address to lowercase to ensure case-insensitive matching when users log in. )striplowerr)r r r rr es zUserManager.lowercase_email)NN) __name__ __module__ __qualname____doc__rrrr# staticmethodr __classcell__r r r&rr s    rc@seZdZdZddZdS)UserCodeManagerz Custom manager for the UserCode model. Methods: create_or_update_code(user): Create or update a UserCode instance for the given user. cCsxt}z|j|d}Wnty|j|dd}Ynw|r:tjr0dtj ddd}||_ nd|_ ||_ | |S)a Create or update a UserCode instance for the given user. Args: user (User): The user for whom the verification code is created or updated. Returns: UserCode: The UserCode instance associated with the user. This method is used to create or update a UserCode instance for a user. It generates a verification code and updates the timestamp. )ri@)rcoder( 1234567890)k) rnowrrcreater IS_PRODUCTIONjoinrandomchoicesr2 timestampr)rrZ date_updated user_codeZ code_valuer r rcreate_or_update_codes  z%UserCodeManager.create_or_update_codeN)r+r,r-r.r>r r r rr1{s r1c eZdZdZfddZZS)UserVehicleManagerz Custom manager for the UserVehicle model. Methods: get_queryset(): Override the default queryset to filter out removed users. cr)a` Override the default queryset to filter out removed users. Returns: QuerySet: The filtered queryset of user vehicle records. This method is used to customize the default queryset for the UserVehicle model. It filters out user vehicle records associated with users marked as removed (is_removed=True). Fr r!r%r&r rr#s zUserVehicleManager.get_querysetr+r,r-r.r#r0r r r&rr@sr@cr?)UserVehicleAssetManagera Custom manager for the UserVehicleAsset model. This manager provides additional methods for querying UserVehicleAsset instances. Methods: get_queryset: Returns a queryset of UserVehicleAsset instances with 'is_removed' set to False. cr)z Get a queryset of UserVehicleAsset instances with 'is_removed' set to False. Returns: QuerySet: The queryset of UserVehicleAsset instances. Fr r!r%r&r rr#z$UserVehicleAssetManager.get_querysetrAr r r&rrB rBcr?)ContactUsManagerz Custom manager for the ContactUs model. This manager provides additional methods for querying ContactUs instances. Methods: get_queryset: Returns a queryset of ContactUs instances with 'is_removed' set to False. cr)z Get a queryset of ContactUs instances with 'is_removed' set to False. Returns: QuerySet: The queryset of ContactUs instances. Fr r!r%r&r rr#rCzContactUsManager.get_querysetrAr r r&rrErDrEcr?)ContactUsAssetsManagera Custom manager for the ContactUsAssets model. This manager provides additional methods for querying ContactUsAssets instances. Methods: get_queryset: Returns a queryset of ContactUsAssets instances with 'is_removed' set to False. cr)z Get a queryset of ContactUsAssets instances with 'is_removed' set to False. Returns: QuerySet: The queryset of ContactUsAssets instances. Fr r!r%r&r rr#rCz#ContactUsAssetsManager.get_querysetrAr r r&rrFrDrFcr?)NewsLetterSubscriberManagera Custom manager for the NewsLetterSubscriber model. This manager provides additional methods for querying NewsLetterSubscriber instances. Methods: get_queryset: Returns a queryset of NewsLetterSubscriber instances with 'is_removed' set to False. cr)z Get a queryset of NewsLetterSubscriber instances with 'is_removed' set to False. Returns: QuerySet: The queryset of NewsLetterSubscriber instances. Fr r!r%r&r rr# rCz(NewsLetterSubscriberManager.get_querysetrAr r r&rrGrDrG)r: django.confrdjango.contrib.auth.modelsrdjango.core.exceptionsr django.dbr django.utilsrZ#apps.account.services.mail_servicesrrManagerr1r@rBrErFrGr r r rs      o4