B?Z J d Z ddlZddlZddlmZmZ ddlmZmZ ddl m
Z
ddl m Z ddl
mZ ddlmZmZ dd lmZ dd
lmZmZ dd lmZ dd lmZ ej e Z ed
d G d d Z G d de Z G d de Z!de"de#fdZ$dS )u
This program is free software: you can redistribute it and/or modify it under
the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License,
or (at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
See the GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see .
Copyright © 2019 Cloud Linux Software Inc.
This software is also available under ImunifyAV commercial license,
see
N) dataclassfields)AnySelf)urljoin)Request)
alru_cache)APIAPIError)
LicenseCLN)IAIDTokenErrorIndependentAgentIDAPI) HostingPanel)ImunifyPatchSubscriptionTF)frozeneqc b e Zd ZU dZeed< dZedz ed< ede ee
f de fd Z dS )PurchaseEligibilityFeligibleN purchase_urldatareturnc d t | D fd| D } | di |S )Nc h | ] }|j
S )name).0fs c/opt/imunify360/venv/lib/python3.11/site-packages/imav/malwarelib/api/imunify_patch_subscription.py z0PurchaseEligibility.from_dict..2 s 4441444 c $ i | ] \ }}|v ||
S r r )r kv allowed_keyss r
z1PurchaseEligibility.from_dict..3 s) LLL$!Q!|:K:KA:K:K:Kr! r )r items)clsr
filtered_datar% s @r from_dictz PurchaseEligibility.from_dict0 sS 44s 444 LLLL$**,,LLL
s##]###r! )
__name__
__module__ __qualname__r bool__annotations__r str classmethoddictr r r* r r! r r r + sn Hd #L#*###$T#s(^ $ $ $ $ [$ $ $r! r c e Zd ZdS )PurchaseEligibilityErrorN)r+ r, r- r r! r r4 r4 7 s Dr! r4 c e Zd Z eej d Z eej d ZdZe de
e de
e dz fd Z e de
fd Ze ee de
fd
Zed
d e dz de dz fd ZdS )ImunifyPatchSubscriptionAPIz /api/patch/subscriptions/checkz//api/products/patch/agent/generate-product-datai idsr Nc | K |sg S | d d{V x}sdS t | j t j d|i |dd } | | d{V }n3# t $ r&}t d| Y d}~dS d}~ww xY w|d S ) zrGet active subscriptions for specific ids.
Return None if interaction with the remote API failed.
z9Can't get iaid token: %s. Return empty subscription list.Nusersapplication/jsonzX-Authz Content-Typer headerszHFailed to get subscriptions details: %s. Return empty subscription list. patchActive)
_get_tokenr SUBSCRIPTION_URLjsondumpsencode
async_requestr loggererror)r( r7 tokenrequestresultexcs r get_subscriptionsz-ImunifyPatchSubscriptionAPI.get_subscriptionsB s$
I >>K
E
4 WcN++2244 $6HII
,,W55555555FF LL2
44444
m$$s 'B
B3
B..B3c x K | d {V S # t $ r t cY S w xY wN)_get_purchase_eligibilityr4 r )r( s r get_purchase_eligibilityz4ImunifyPatchSubscriptionAPI.get_purchase_eligibilityb s\ ) 6688888888 8' ) ) )&(( ( ( ( )s 99)ttlc K | d{V x}st t t d{V t j d}t | j t j
| |dd } | | d{V }n;# t $ r.}t d| t d}~ww xY wd|vsd|vr)t d | t t | S )
a
Asynchronously retrieves the customer's host purchase eligibility status
for Imunify Patch.
Sends an authenticated request to the configured external service
to determine whether the system is eligible for a purchase offer
and, if so, obtains the associated purchase URL.
Returns:
PurchaseEligibility: An instance of PurchaseEligibility containing
the eligibility flag and optional purchase URL.
In case of any of the following, the PurchaseEligibilityError is raised:
- Missing or invalid authentication token
- Failure to make a request (e.g. APIError)
- Malformed or incomplete response (missing required keys)
N)r9 custom_reseller_configuredr: r; r< z Failed to get purchase URL: %s.r r z Bad response from server %s.)r? r4 lenr get_usersr is_custom_reseller_configuredr ELIGIBILITY_URLrA rB rC rD r rE rF r r* )r( rG r rH rI rJ s r rN z5ImunifyPatchSubscriptionAPI._get_purchase_eligibilityi s} , #~~ / //////// -*,, , |~~ 7 7 9 9999999::*4*R*T*T
D!!((** $6HII
- ,,W55555555FF - - - LL:C @ @ @*,, , -
& ( (j.F.F LL7 @ @ @*,, ,",,V444s 3C
D)DD
error_messagec K t j d {V S # t $ r(}t | pd| Y d }~d S d }~ww xY w)NzCan't get IAID Token: %s.)r get_tokenr
rE rF )rW rJ s r r? z&ImunifyPatchSubscriptionAPI._get_token s| .8:::::::: : LL )/B ) ) ) [) ZI *5 *5 *5 *5 [*5X d
cDj \ r! r6 _r c L t j S rM )r selectexists)r^ s r has_imunify_patch_subscriptionsrb s # * , , 3 3 5 55r! )%__doc__rA logging dataclassesr r typingr r urllib.parser urllib.requestr async_lrur defence360agent.api.serverr
r !defence360agent.contracts.licenser defence360agent.internals.iaidr
r +defence360agent.subsys.panels.hosting_panelr imav.malwarelib.modelr getLoggerr+ rE r Exceptionr4 r6 r0 r. rb r r! r rq s * ) ) ) ) ) ) ) ) " " " " " " 4 4 4 4 4 4 4 4 8 8 8 8 8 8 E D D D D D : : : : : : 8 $ $ $5!!!$ $ $ $ $ $ $ "!$ y e e e e e# e e eP6s 6t 6 6 6 6 6 6r!