f_ d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZmZ d dl m
Z
d dl m Z d dl
mZ d dlmZmZ d dlmZmZmZmZmZ d dlmZ d d lmZmZm Z m Z m Z d d
l m Z d d l!m"Z" G d d
e
Z#e e#j$ e e#j% iZ& e" s
e#j$ e&e< d Z'd Z(d Z)e G d d Z*e G d d Z+d Z,d Z-d Z.d Z/dS ) N) dataclassasdict)Enum)drop_privileges)cpinfo)ListDict)get_allowed_modulesget_admin_suites_configwrite_suites_allowed
ALL_SUITESget_allowed_suites) CDNSuitePro)OBJECT_CACHE_FEATURECRITICAL_CSS_FEATUREIMAGE_OPTIMIZATION_FEATURE CDN_FEATUREFeature)
UserConfig)is_wp2_environment_safec e Zd ZdZdZdZdZdS )BillingFeaturezA
Backwards-compatible list of features that we bill for.
zAccelerateWP PremiumzAccelerateWP CDN FreezAccelerateWP CDN ProN)__name__
__module__ __qualname____doc__ACCELERATE_WP_PREMIUMACCELERATE_WP_CDNACCELERATE_WP_CDN_PRO py/clwpos/billing.pyr r ! s. 3/ 2r! r c | t k rt | S t j |v r t
j S t | S )N)r FEATURE_TO_BILLING_FEATUREgetr namer r )featureallowed_suitess r" billing_feature_by_awp_featurer) 2 sL + )--g666>)) 33 % ) )' 2 22r! c 8 | dS t j d| duS )z-
Checks that string has uuid4 format
NFzK[0-9a-fA-F]{8}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}-[0-9a-fA-F]{12})rematch)values r"
is_valid_uuidr. ; s(
}u
8ach i iqu uur! c R t j rt S t j | }t |j }|j }|t | s@t t j |_ t |j |j
| |j S )z
We need some unique identifier which user can
use as his token on our provisioning server.
We use uuid4 to make it somehow hard to bruteforce
and still unique (hopefully as we don't check that :P)
)osgeteuid get_unique_identifier_as_userpwdgetpwnamr pw_uid unique_idr. struuiduuid4r pw_gid)usernamepwconfigr6 s r" get_or_create_unique_identifierr> D s
z|| /,... h B
$RY
/
/F I
i 8 8 tz||,, RY 6::: r! c 8 e Zd ZU eed< eed< eed< eed< dS )
FeatureRecordr&
purchase_date
attributesactiveN)r r r r7 __annotations__r boolr r! r" r@ r@ Y s:
III LLLLLr! r@ c D e Zd ZU eed< eed< eed< ee ed< dS )
UserRecordr; primary_domainidfeaturesN)r r r r7 rD r r@ r r! r" rG rG b sA MMM GGG=!!!!!!r! rG c j g } t d D ]\ }} t j | j }n%# t $ r t j d| Y Aw xY w t ||| }n%# t $ r t j d| Y yw xY w|
t | | S )z
Collect report about billable users.
Look for all users with allowed feature and add them
to list for the further processing on CLN side.
legacy argument changes format so it
is accepted by older CLN versions
)cplogindns)keylszHUser %s does not have system uid; Malformed control panel configuration?z%CLN billing report for user %s failed) r r3 r4 r5 KeyErrorloggingwarning_single_user_report Exception exceptionappendr )reportuserdomainuid user_records r"
get_reportr[ j s F %7888 + + f ,t$$+CC O fhl m m m H -c4@@KK Et L L L H
f[))**** Ms! 3 AAA++ B
B
c t | }t | }t | t | } fd|D } fd|D }t }|D ]} t | | }
|
|
j |vrt |
j | | rt || n>t t j
d | | i | |v ||
j < ||
j j p| |v ||
j _ t ||t! | t# | S )Nc p i | ]2}t | j D ] }|j | 3S r )r
rJ purchase_datesr% .0suiter' suites_admin_configs r"
z'_single_user_report.. s\ !%(1
$377>> r! c i | ]?}t | j D ]*}|j |t +@S r )r
rJ rB r% dictr_ s r" rc z'_single_user_report.. sb !%(1
$/33E466BB r! )day)r& rA rB rC )r; rH rI rJ )r
_get_enabled_modulesr r re r) r- r@ r% r7 datetimedatetodayreplacerC rG r> listvalues) rY r; rX rJ enabled_featuresr( feature_purchase_datesfeature_attributesbillable_featuresr' billing_featurerb s @r" rR rR s 1# 6 6H+H551#66',,N # # a a8.QQ " (9 9 97D$* .11'::>C.w7888
++--55!5<<==-11'2>> "22 8 8 8 o 3 4 4$ "/"78?`GO_D_
o 3 4 ; ; !.x88+224455
r! c t | 5 d t | D cd d d S # 1 swxY w Y d S )Nc h | ]\ }}}| S r r )r` _modules r" z'_get_enabled_modules.. s* . . .<1a . . .r! )r; )r r enabled_modules)rW s r" rh rh s . .. .:, , ,*?,,. . .. . . . . . . . . . . . . . . . . .s ,A A
A
c N ddl m} ddlm} | d|j i d }|S )z4
Get unique identifier for current end-user
r )daemon_communicate)
WposDaemoncommandr6 ) clwpos.utilsr{
clwpos.daemonr| DAEMON_GET_UNIQUE_ID_COMMAND)r{ r| r6 s r" r2 r2 sW 0/////((((((""::$ I r! )0ri rP r0 r3 r+ r8 dataclassesr r enumr clcommon.clpwdr clcommon.cpapir typingr r clwpos.feature_suitesr
r r r
r r clwpos.optimization_featuresr r r r r clwpos.user.configr r~ r r r r r$ r) r. r> r@ rG r[ rR rh r2 r r! r" r s
) ) ) ) ) ) ) ) * * * * * * ! ! ! ! ! ! . - - - - - * ) ) ) ) ) 0 0 0 0 0 03 3 3 3 3T 3 3 3 D 1
\7E7[ 343 3 3v v v * " " " " " " " " 25
5
5
n. . .
r!