a ld4@s<ddlmZmZddlmZddlmZddlmZmZm Z ddl Z ddl m Z ddl mZddlmZdd lmZmZmZmZmZmZmZdd lmZdd lmZdd lmZdd lm Z m!Z!m"Z"m#Z#ddl$m%Z%ddl&m'Z'm(Z(m)Z)ddl*m+Z+ddl,m-Z-ddl.m/Z/ddl0m1Z1ddl2m3Z3ddl4m5Z5ddl6m7Z7ddl8m9Z9m:Z:ddl;mZ>m?Z?ddl@mAZAddlBmCZCmDZDddlEmFZFddlGmHZHddlImJZJmKZKmLZLmMZMmNZNmOZOddlPmQZQmRZRmSZSmTZTddlUmVZVmWZWdd lXmYZYeZZeNeOeKeJgZ[ed!e+e[d"d#Gd$d%d%e3Z\Gd&d'd'e3ee?Z]dS)()datetime timedelta) strtobool)BytesIO)IterableList OrderedDictN)settings)get_user_model)EmailMultiAlternatives) BooleanFieldCaseCountFQValueWhen) HttpResponse)timezone)method_decorator)ResetPasswordToken clear_expiredget_password_reset_lookup_field$get_password_reset_token_expiry_time)reset_password_token_created)HTTP_IP_ADDRESS_HEADERHTTP_USER_AGENT_HEADER_unicode_ci_compare)swagger_auto_schema)action) IsAdminUser)Response) ModelViewSet) celery_app)DEFAULT_FROM_EMAIL)CompanyExportExcelSerializerCompanyUpdateSerializer)BaseCompanyHistoryViewSetBaseCompanyListViewSetBaseCompanyRetrieveMixinBaseCompanyViewSet)ExportUserIntoExcelTask)Company Subscription) ReviewSource)UserResultsSetPagination) assign_param company_paramend_date_paramstart_date_param user_paramusername_param)AdminUpdateProfileSerializerAdminUpdateUserSerializerCreateUserSerializerUserSerializer)SealWidget)export_company_tasklist)manual_parameters)name decoratorcseZdZeZejdZ e gZ e Z fddZedeegdedgdddd d Zfd d Zfd dZfddZZS)AdminUserViewSetprofilec st}|jdkr|jj}|tjd}|rDt|d}| |}|t jd}t t |t jd}|tjd}|r|dur|rtddt|dt|dBt|dB@t|d B}ntddt|d B}| |jtttd |tdd td td ddd}|S|r>|s>|s>| t|dt|dB}|dur|s|r~tddt|dt|dBt|dB@}n tdd}| |d}|S|}|dS|dS)Nr=idfalseT)Zcompanies__isnull)Zusername__icontains)Zname__icontains)Zemail__icontains) companies__idrD)Zcompanies__user_idrFthenF)default output_field)is_ownerz -is_ownerz-id)super get_querysetrrequest query_paramsgetr4r?rfilterr1boolrr0r5annotater rrrr order_bydistinct) selfquerysetrNZ user_filterZ user_queryZcompany_filterZ assign_filter name_filterfilter_ __class__D/var/www/html/Ranjet/backend/verify_trusted/users/api/views/admin.pyrLVsx         zAdminUserViewSet.get_querysetGET)methodr>FZ dashboard)methodsdetailurl_pathc Cs|jj}|tjdd}|tjdd}t|d}|durLt|dnt}t dt | | dd}t dt d| ||gdd}t dt d | ||gdd}t d t ||gd d}t j } | j|||d } tj j|t d d } | d| d| d| d| dd} t| S)Nz 00:00:00z 23:59:59z%Y-%m-%d %H:%M:%S company_id)Z due_date__lt due_date__gte)rPT)Zpayment__isnullrcZcreate_date__rangeFrD)Zdate_joined__range)count_user_non_payingcount_user_trialcount_user_paid)count_user_signed_up_setcount_total_userrdrerfrgrh)rdrerfrgrh)rMrNrOr3r?r2rstrptimenowrrdater-objectsall aggregateUserr!) rUrMrNstart_date_filterend_date_filterrdrerfrgdatauser response_datar[r[r\admin_dashboardsh   z AdminUserViewSet.admin_dashboardcs4|jdkrtS|jdvrtS|jdkr*tStS)Ncreate)updatepartial_updateZupdate_profile)rr8r7r6rKget_serializer_classrUrYr[r\rys   z%AdminUserViewSet.get_serializer_classcs(tjj|djdddt|dS)N)rsF) is_verifiedrs)r,rlrPrwrKperform_destroyrUinstancerYr[r\r|sz AdminUserViewSet.perform_destroyc stj|g|Ri|}tjjfidt|jdi}t }t t |d}t ||rt|jt|tr|jdkr|jd}n&tjj||jtd|jtdd}tj|j||d|S)Nz {}__iexactemail)hoursr)rs user_agent ip_address)senderr~reset_password_token)rKrvrorlrPformatrrrfirstrrrjrreligible_for_resetrrgetattrpassword_reset_tokensrmcountrMETArOrrrsendrZ) rUrMargskwargs create_userrsZ$password_reset_token_validation_timeZnow_minus_expiry_timetokenrYr[r\rvs6     zAdminUserViewSet.create)__name__ __module__ __qualname__r9serializer_classrorlprefetch_relatedrmrVr permission_classesr/pagination_classrLrr3r2rruryr|rv __classcell__r[r[rYr\rALs C ; rAcsbeZdZegZfddZeddgdddZdd Ze gd ed d gd dddZ Z S)AdminCompanyViewSetcs|jdkrtStS)Nrw)rr&rKryrzrYr[r\rys z(AdminCompanyViewSet.get_serializer_classTPATCH)r`r_cOs(|}d|_d|_|tddiS)NFsuccess) get_objectrsr{saver!)rUrMrrr~r[r[r\ remove_users zAdminCompanyViewSet.remove_usercCsTtjj|jdtjj|jdtjj|jdtjj|jddS)N)rbrC)r;rlrPrDdeleter:r.r,r}r[r[r\r|sz#AdminCompanyViewSet.perform_destroy)paginator_inspectorsFr]z export-excel)r`r_racOs$|jj}t|}|tddS)N)status)rMrNr<s apply_asyncr!)rUrMrrrNtaskr[r[r\ export_excelas z AdminCompanyViewSet.export_excel) rrrr rryrrr|rrrr[r[rYr\rs   Fr)^rrZdistutils.utilriortypingrrrZ xlsxwriter django.confr django.contrib.authr django.core.mailr django.db.modelsr r rrrrr django.httpr django.utilsrdjango.utils.decoratorsr django_rest_passwordreset.modelsrrrr!django_rest_passwordreset.signalsrZdjango_rest_passwordreset.viewsrrrdrf_yasg.utilsrrest_framework.decoratorsrrest_framework.permissionsr rest_framework.responser!rest_framework.viewsetsr"configr#Zconfig.settings.baser$(verify_trusted.companies.api.serializersr%r&Z%verify_trusted.companies.api.viewsetsr'r(r)r*Zverify_trusted.companies.exportr+verify_trusted.companies.modelsr,r-verify_trusted.reviews.modelsr.$verify_trusted.users.api.paginationsr/verify_trusted.users.api.paramsr0r1r2r3r4r5$verify_trusted.users.api.serializersr6r7r8r9Zverify_trusted.widgets.modelsr:r;Zverify_trusted.companies.tasksr<ro search_paramsrArr[r[r[r\sd     $                6