30 lines
855 B
Python
30 lines
855 B
Python
|
import random
|
||
|
|
|
||
|
|
from data.ClientParams import ClientParams
|
||
|
|
from data.enum.CompanySize import CompanySize
|
||
|
|
|
||
|
|
class ClientParamsFactory:
|
||
|
|
|
||
|
|
def __init__(self) -> None:
|
||
|
|
super().__init__()
|
||
|
|
|
||
|
|
def get_client_params(self) -> ClientParams:
|
||
|
|
payment_delay = random.randint(0, 14)
|
||
|
|
payed = random.getrandbits(1)
|
||
|
|
net_worth = random.randint(0, 100)
|
||
|
|
is_skarbowka = random.getrandbits(1)
|
||
|
|
membership = random.getrandbits(1)
|
||
|
|
infuelnce_rate = random.randint(0, 100)
|
||
|
|
is_hat = random.getrandbits(1)
|
||
|
|
company_size = random.randint(1,6)
|
||
|
|
|
||
|
|
return ClientParams(
|
||
|
|
payment_delay,
|
||
|
|
net_worth,
|
||
|
|
infuelnce_rate,
|
||
|
|
bool(payed),
|
||
|
|
bool(is_skarbowka),
|
||
|
|
bool(membership),
|
||
|
|
bool(is_hat),
|
||
|
|
CompanySize(company_size)
|
||
|
|
)
|