legacy/pairent_backend/pairent_app/factories.py

130 lines
4.3 KiB
Python

import factory
import random
import time, datetime
from django.db import models
from pairent_app.models import Apartament, User
factory.Faker.override_default_locale('ru_RU');
class OpenID_Address(factory.declarations.BaseDeclaration):
def evaluate(self, instance, step, extra):
return ''.join(random.choices(list('abcdef12345678990'), k=6)) + "@vvsu.ru";
class PhoneNumber(factory.declarations.BaseDeclaration):
def evaluate(self, instance, step, extra):
return '+799' + str(random.randint(0, 99999999)).zfill(8);
class CSV(factory.declarations.BaseDeclaration):
def __init__(self, min, max, count_min, count_max):
self.min = min; self.max = max;
self.count_min = count_min;
self.count_max = count_max;
super().__init__();
def evaluate(self, instance, step, extra):
len_ = random.randint(self.count_min, self.count_max);
out = [];
for i in range(len_): out.append(str(random.randint(self.min, self.max)));
return ','.join(out);
class Random(factory.declarations.BaseDeclaration):
def __init__(self, min, max = None, multiply = 1, list = None):
if (max == None):
self.min = 0;
self.max = min;
else:
self.min = min;
self.max = max;
self.multiply = multiply;
self.list = list;
super().__init__();
def evaluate(self, instance, step, extra):
if (type(self.list) == list):
return random.choice(self.list);
if (self.multiply != None):
return random.randint(self.min, self.max) * self.multiply;
return random.randint(self.min, self.max);
class Date(factory.declarations.BaseDeclaration):
def __init__(self, min_year = 2022, max_year = datetime.date.today().year):
self.min_year = min_year;
self.max_year = max_year;
super().__init__();
def evaluate(self, instance, step, extra):
return datetime.datetime(random.randint(self.min_year, self.max_year), random.randint(1,12), random.randint(1,27)).strftime("%G-%m-%d");
class ApartmentFactory(factory.django.DjangoModelFactory):
class Meta:
model = Apartament
# Base data
price = Random(3,10,5000);
lastPrice = Random(3,10,5000);
bail = Random(2,10,1000);
agencyCommission = Random(3,5,2500);
utilitiesPrice = Random(4,8,1000)
minimumLeasePeriod = Random(1,2,6);
address = factory.Faker('address', locale='ru_RU');
description = factory.Faker('sentence', locale='ru_RU');
perimetrs = Random(15,40);
rooms = Random(0,4);
ceilingHeight = Random(3,5);
floorHouse = Random(2,20);
floor = Random(2,20);
phoneNumber = PhoneNumber()
timeToBus = Random(1, 12, 5);
timeToTrain = Random(1, 12, 5);
# Apartment props
isFurniture = Random(0, 1);
isAnimal = Random(0, 1);
isTelevision = Random(0, 1);
isChild = Random(0, 1);
isInternet = Random(0, 1);
isBathroom = Random(0, 1);
isRefrigerator = Random(0, 1);
isWasher = Random(0, 1);
isAirConditioning = Random(0, 1);
isFreshRepair = Random(0, 1);
# House props
isElevator = Random(0, 1);
isParking = Random(0, 1);
isGarbageChute = Random(0, 1);
isConcierge = Random(0, 1);
# views = [] # factory.Faker('random_number')
dateCreate = Date()
class UserFactory(factory.django.DjangoModelFactory):
class Meta:
model = User
favorites_apartments = CSV(1, 100, 1, 16);
comparison_apartments = CSV(1, 100, 1, 5);
openid_addr = OpenID_Address();
name = factory.faker.Faker('name');
date_of_birth = Date(1980, 2006);
about_me = factory.faker.Faker('sentence');
gender = Random(0,0,0, ['f', 'm', 'n', '?'])
phone = PhoneNumber()
email = OpenID_Address()
telegram = '@uwu'
discord = '@uwu' # они поменяли формат
city = Random(0,0,0, ['Владивосток', 'Хабаровск', 'Урюпинск', 'Мухосранск', 'Нью-Йорк'])
role = 's'