SI_InteligentnyWozekWidlowy/ClientParamsFactory.py

30 lines
855 B
Python
Raw Normal View History

2022-05-09 15:49:11 +02:00
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)
)