B Zcw@shddlZddlmZddlZddlmZddlmZddlm Z ddl m Z ddl m Z ddl mZdd lmZdd lmZdd lmZdd lmZmZd dZGdddeZGddde jZddZddZddZddZddZ ddZ!d9dd Z"d!d"Z#d#d$Z$d%d&Z%d'd(Z&d)d*Z'd+d,Z(d-d.Z)d/d0Z*d1d2Z+d3d4Z,Gd5d6d6e-Z.d7d8Z/dS):N)wraps) countries)settings)DjangoFilterBackend)viewsets)filters)SessionAuthentication)get_system_setting) SystemSetting)Company) DefaultSchemaDefaultPaginationBasecCs(d||}dt|}|S)zN authorization format: 'Basic base64.encode(client_id:client_secret)' z{}:{}zBasic {})formatbase64Z b64encodeencodedecode) client_id client_secretZ"client_id_and_client_secret_format authorizationr+G:\easytimepro\master/mysite/cloud\utils.pyhttp_basic_auths rc@seZdZddZdS)CsrfExemptSessionAuthenticationcCsdS)Nr)selfrequestrrr enforce_csrf!sz,CsrfExemptSessionAuthentication.enforce_csrfN)__name__ __module__ __qualname__rrrrrrsrc@s(eZdZdZeZeZee j e j fZ dS) CloudViewSetN) rrrZmodelr Zschemar Zpagination_classrrZ SearchFilterZOrderingFilterZfilter_backendsrrrrr%s rcCs ddlm}|}||}|S)Nr) CurrencyCodes)Zforex_python.converterr Z get_symbol)currency_alpha_3r Z currency_codecurrency_symbolrrrget_currency_symbol.s  r#cCst|}dj|t|d}|S)Nz{currency_symbol}{price})r"price)r#rstr)r$r!r"Zprice_with_currency_symbolrrrget_price_with_currency_symbol7sr&cCsRddl}tt|}|dkr d}|jj|d}|jj|jd}|sHd}n|j}|S)NrzUnited States of Americaz United States)name)numericZUSD) pycountrydictrgetZ currenciesr(Zalpha_3) country_coder) country_namecountryZcurrencyr!rrrget_country_currency>s r/cCst|}|S)N)rZby_name)r-r,rrrconvert_country_name2codeNs r0cCstt|}|S)N)r*r)r,r-rrrconvert_country_code2nameTs r1cCsddl}||j}|S)Nr) ipaddress ip_address is_private)r3r2r4rrrcheck_ip_is_privateZs r5c Csddl}ddl}ddlm}ddlm}m}t}|d |d}|sfyd} d} |sj|j j |d }|r|j r|j js|j j |jd } | r| jr| jr| j} | j} nf||} | jd} | jd} y"|j j||j| | d d Wn$tk r} zWdd} ~ XYnX| |d <| |d <|jd || | gd dWnd|d <d|d <YnXn|d|d <|d|d <|S)Nr)cache)TerminalLocationTerminalz!{}_geolocation_latitude_longitude)r3)Z terminal_id)latitude longitude)ZterminalZgps_timedefaultsr;r<i)timeout)geocoderdatetimedjango.core.cacher6Zmysite.iclock.modelsr7r8r*r+robjectsfilterfirstZareaZ is_defaultidr;r<ipZlatlngZupdate_or_createZnow Exceptionset)r3devicer?r@r6r7r8geolocation_infoZgeolocation_cacher;r<Zterminal_location geolocationerrrconvert_ip2geolocationasB    "  rMc Csddl}ddlm}ddlm}t}|d|d}|syf||}|jrX|j|d<n2| |}|j } | j |d<| j dd|_| | d||dWqtk rd|d<YqXn|d||d<|S)Nr)r6) getDevicez{}_device_ip_to_addressaddress(-)r?rAr6Zmysite.iclock.utilsrNr*r+rZ gps_locationrFZcurrent_resultrOZsaverH TypeError) r3Zsnr?r6rNZdevice_address_infoZdevice_address_cacherIrKresultrrrconvert_ip2addresss(     rTcCsddl}ddlm}ddlm}|jr.|j}n|j}t}|dksF|srd|d<d|d<d |d <d |d <d |d <n~| d |d}|st j j |dj }td|} | rd| krd| kr| d} | d} | d} | d} | d}n$y||}||}|ddd}|d} |d} |dd}t|}|d} |d}|d } |d }| r| | d<| | d<| | d<|| d<| | d<|| d<tj j|ddjt| dt j j|d j||| d!d"|jdt|f}||Wn*tk rd} d} d } d } d }YnX| |d<| |d<| |d <| |d <||d <|d || |d# || |d$ || |d% || |d& ||nd| d ||d<| d# ||d<| d$ ||d <| d% ||d <| d& ||d <|S)'Nr)r6)rZDefaultgP fS@r;gbu)@r<ZIndiar.zNot setprovincecityz{}_geolocation_info_latitude)Z name__iexactZcompany_settingcompany_latitudecompany_longituder-company_province company_cityZgeometrylocationZlatZlngaddress_componentscountry_long_namecountry_short_namecompany_country) company_idr')value)rE)r.rVrUz%s_%s_%sz{}_geolocation_info_longitudez{}_geolocation_info_countryz{}_geolocation_info_provincez{}_geolocation_info_city) googlemapsrAr6 django.confrZ CLOUD_VERSIONZINTELLIGENTTIME_MAPS_APIZEASYTIMEPRO_MAPS_APIr*r+rr rBrEr ZClientZgeocodecompany_addressr rCupdatejsondumpsZUNITr%deleterGrH)Z company_namerOrbr6rZ MAPS_API_KEYrJZgeolocation_info_cacher`Zcompany_settingsrWrXr_rYrZZ google_mapsZgeocode_resultr[r\Zcompany_locationZcompany_country_short_nameZ cache_keyrrrconvert_address2geolocations           ricCsvi}xl|D]d}|dddkr6|d|d<|d|d<|dddkrR|d|d <|ddd kr |d|d <q W|S) Ntypesrr.Z long_namer]Z short_namer^Zadministrative_area_level_1rUZlocalityrVr)r\Z address_dictZ componentsrrrrds    rdcsDddlmddlmtfdd}j|_j|_|S)Nr) Membership)HttpResponseForbiddencsH|j}|jrDjj|dd}|s.|jdkr>|f||SSdS)N)userroleadmin)rnZis_authenticatedrBrCexistsusername)rargskwargsrnZmembership_exist)rlrkviewrrwrappers z0admin_and_vendor_login_required..wrapper)mysite.base.modelsrk django.httprlr__doc__r)rurvr)rlrkruradmin_and_vendor_login_requireds    rzcs6ddlmtfdd}j|_j|_|S)Nr)rlcs&|j}|jr|f||SSdS)N)rnis_super_admin)rrsrtrn)rlrurrrv&sz%admin_login_required..wrapper)rxrlrryr)rurvr)rlruradmin_login_required#s  r|cs<ddlmmtfdd}j|_j|_|S)Nr)rlHttpResponseNotFoundcs:|j}|js0|js|jr(|f||SSnSdS)N)rnZ is_anonymousZis_vendor_displayr{)rrsrtrn)rlr}rurrrv7s  z/admin_or_vendor_login_required..wrapper)rxrlr}rryr)rurvr)rlr}ruradmin_or_vendor_login_required4s  r~c Csddlm}ddlm}|j}|jj|dd}|jj|d}g}g}x,|D]$}|t |j |t |j qNWxtt |dkr|d} |jj| d} d d | D} x*| D]"} | |kr|| || PqW|d| krx| qxW|S) u Get all companies id list to which the current vendor belongs :param request: :return: belongs company id list The company structure can be converted to a multi-tree  Use non-recursive method of depth-first traversal Traverse the current vendor and the company that belongs to this vendor as a sub-vendor the company that the sub-vendor belongs to, and so on ... E.g vendor_1 / / / / / / / / / / / / / / company_1 company_2 company_6 / / / / company_3 company_4 company_5 company_7 Company_ * of any node login to the backend dashboard and iterate to find all company lists of its subordinates If I login in in "vendor_1", and the "all company count" is 7(company_1, company_2, company_3, company_4, company_5, company_6, company_7) If I login in in "company_2", and the "all company count" is 3(company_4, company_5, company_7) If I login in in "company_1", and the "all company count" is 1(company_3) r)rk)r )rncompany)id__in)vendor_company_idcSsg|] }|jjqSr)rrE).0mrrr {sz0get_all_belongs_companies_id..) rwrkmysite.cloud.modelsr rnrBrC values_listappendr%rElenpop) rrkr current_userZcurrent_user_company_id_listZcurrent_user_company_querycompany_id_listZbelongs_company_id_listcr`Zcurrent_belongs_companyZsub_belongs_company_listZsub_belongs_companyrrrget_all_belongs_companies_idHs,!        rcCsbddlm}|jj}|jjr(|j}n6|jjrPt|}|jj |dj |d}n|jj |d}|S)z get company query by three types: super admin's company/superuser extra's company/current company :param request: :return: company query r)r )r)rE) rr rnZcurrent_companyr{rBallZis_superuser_extrarrCZexclude)rr current_company_id company_queryall_belongs_companies_idrrrget_company_querys  rcCs:ddlm}|jjr |j}nt|}|jj|d}|S)z get company query by two types: super admin's company/other user's company for order and payment record ViewSet :param request: :return: company query r)r )r)rr rnr{rBrrrC)rr rrrrr)get_company_query_without_superuser_extras   rc Csddlm}ddlm}|j}|j}g}|jj|d d}x,|D]$}|jj|d }|rF| |qFW|j dkr|jj|d} n"|jjdd  d} |jj| d} | S) u Get the vendor query The company structure can be converted to a multi-tree E.g vendor_1 / / / / / / / / / / / / / / company_1 company_2 company_6 / / / / company_3 company_4 company_5 company_7 Vendor mean：Find the number of non-leaf nodes with a multi-tree depth of 1 If I login in in "vendor_1", and the "vendor count" is 2(company_1, company_2) If I login in in "company_1", and the "vendor count" is 0 If I login in in "company_2", and the "vendor count" is 1(company_5) :param request: :return: vendor query r)rk)r )rrrp)rrm)ro) rwrkrr rnZget_current_companyrErBrCrrqrrr) rrkr rrZ"vendor_first_depth_company_id_listZfist_depth_company_id_listr`Z node_exitsrrrrrget_vendor_querys     rc@sreZdZdZejejfddZdddZddd Z dd d Z d d ej ej ej dddfddZdejdfddZdS) AWSBoto3APIz :type aws_access_key_id: string :param aws_access_key_id: AWS access key ID :type aws_secret_access_key: string :param aws_secret_access_key: AWS secret access key cCs*ddl}||_||_||j|j|_dS)Nr)boto3aws_access_key_idaws_secret_access_keyZSession_session)rrrrrrr__init__szAWSBoto3API.__init__acmcCs0|jj|tjd}|jdddgd}|d}|S)a* docs: https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/acm.html#ACM.Client.list_certificates :param server_name: aws server name :return: Returns a dictionary (e.g., [ {'CertificateArn': 'arn:aws:acm:us-east-2:402510031561:certificate/6c37409d-57ea-4f7e-92c5-8028316828cd', 'DomainName': 'biotime.mx'}, {'CertificateArn': 'arn:aws:acm:us-east-2:402510031561:certificate/c8ddddc1-fee3-446f-a158-4f7ad33b988d', 'DomainName': '*.robin-vendor.biotime.mx'} ]) ) region_nameZPENDING_VALIDATIONZISSUEDZINACTIVE)ZCertificateStatusesZCertificateSummaryList)rclientrAWS_REGION_NAMEZlist_certificates)r server_name acm_clientresponseZcertificate_summary_listrrracm_list_certificatess z!AWSBoto3API.acm_list_certificatesNcCsF|jj|tjd}|j|dddid|dddgd }|d }|S) a| docs: https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/acm.html#ACM.Client.request_certificate :param server_name: aws server name :param domain_name: the domain name which need to add :return: Returns a certificate arn (e.g., arn:aws:acm:us-east-2:402510031561:certificate/b59c71a2-0456-4aa3-a5c9-f362fdea1674). )rZDNSZ(CertificateTransparencyLoggingPreferenceZDISABLEDzvendor-company-subdomainz*.r)ZKeyValue)Z DomainNameZValidationMethodZOptionsZTagsCertificateArn)rrrrZrequest_certificatesplit)rr domain_namerrcertificate_arnrrracm_request_certificate sz#AWSBoto3API.acm_request_certificatecCs2|jj|tjd}|j|d}|ddd}|S)a docs: https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/acm.html#ACM.Client.describe_certificate :param server_name: aws server name :param certificate_arn: the certificate arn :return: Return a dict (e.g., {'DomainName': '*.robin-vendor.biotime.mx', 'ValidationDomain': '*.robin-vendor.biotime.mx', 'ValidationStatus': 'SUCCESS', 'ResourceRecord': {'Name': '_d0a9d5454bbffb5c6ea9d3df796d5d66.robin-vendor.biotime.mx.', 'Type': 'CNAME', 'Value': '_704209c9b88b9f8204b46ef27789a2c7.qqqfmgwtgn.acm-validations.aws.'}, 'ValidationMethod': 'DNS'}). )r)rZ CertificateZDomainValidationOptionsr)rrrrZdescribe_certificate)rrrrrZdomain_validation_optionsrrracm_describe_certificate$s z$AWSBoto3API.acm_describe_certificateZroute53ZCREATEc Cs|jj|tjd} |||d|ddddgd|d} ||d |igd d d dgd|d} | j|| d | j|| d dS)a docs: https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/route53.html#Route53.Client.change_resource_record_sets :param server_name: aws server name :param action_type: CREATE :param alias_dns_name: load balancer instance's DNS name of EC2 :param alias_hosted_zone_id: load balancer instance's hosted zone id of EC2 :param route53_hosted_zone_id: route 53 hosted zone id :param domain_name: domain name :param resource_record_name: the resource record name :param resource_record_value: the resource record value :return: True )rF)ZDNSNameZEvaluateTargetHealth HostedZoneIdA)NameZ AliasTargetType)ZActionZResourceRecordSetzSubdomain for {})ZChangesCommentr<ZCNAME)rZResourceRecordsZTTLr)rZ ChangeBatchT)rrrrrZchange_resource_record_sets) rrZ action_typeZalias_dns_nameZalias_hosted_zone_idZroute53_hosted_zone_idrZresource_record_nameZresource_record_valueZroute53_clientZchange_batch_aZchange_batch_cnamerrr#route53_change_resource_record_sets8s*  z/AWSBoto3API.route53_change_resource_record_setsZelbv2cCs*|jj|tjd}|j|d|igd}|S)a} docs: https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/elbv2.html#ElasticLoadBalancingv2.Client.add_listener_certificates :param server_name: aws server name :param listener_arn: load balancer instance's 443 listener id's arn of EC2 :param certificate_arn: acm certificate arn :return: Return a dictionary )rr)Z ListenerArnZ Certificates)rrrrZadd_listener_certificates)rrZ listener_arnrZ elbv2_clientrrrrelbv2_add_listener_certificatess z+AWSBoto3API.elbv2_add_listener_certificates)r)rN)rN)rrrryrZAWS_ACCESS_KEY_IDZAWS_SECRET_ACCESS_KEYrrrrZAWS_EC2_ALIAS_DNS_NAMEZAWS_EC2_ALIAS_HOSTED_ZONE_IDZAWS_ROUTE53_HOSTED_ZONE_IDrZAWS_EC2_LISTENER_ARNrrrrrrs     Crcs@ddlddlmtfdd}j|_j|_|S)Nr) urlencodec sy|jd}Wntk rtj}tj}tj}|jdd}|st||}d|d}d|i} | } jtj || d } | d}| d} ||jd<|j | YnX|f||S)N access_tokenz!application/x-www-form-urlencoded)z Content-TypeZ Authorization grant_type)Zheadersdata expires_in) ZsessionKeyErrorrZ GRANT_TYPEZ CLIENT_IDZ CLIENT_SECRETr+rZpostZ TOKEN_URLrfZ set_expiry) rrsrtrrrrrZauthorization_headersZbodyZx_www_form_urlencoded_type_bodyZauthorization_contentr)requestsrrurrrvs(  z2get_or_update_oauth2_access_token..wrapper)rZ urllib.parserrryr)rurvr)rrrur!get_or_update_oauth2_access_tokens  4r)N)0r functoolsrrfZdjango_countriesrrcrZdjango_filters.rest_frameworkrZrest_frameworkrrZrest_framework.authenticationrZ mysite.utilsr Zmysite.base.models.sys_settingr rr Zmysite.api.utils_classr r rrZGenericViewSetrr#r&r/r0r1r5rMrTrirdrzr|r~rrrrobjectrrrrrrsF              ,X@48