legacy/pairent_backend/pairent_app/factories.py

import factory
import random
import time, datetime, uuid

from django.db import models

from pairent_app.models import Apartament, User

factory.Faker.override_default_locale('ru_RU');

class UUID(factory.declarations.BaseDeclaration):
    def evaluate(self, instance, step, extra):
        return str(uuid.uuid4()).upper();

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);
    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'

    openid_addr = OpenID_Address();
    openid_id = UUID();
    photo_provider = Random(0,0,0, ['VVSU', 'GRAVATAR']);

def fake(size: int):
    """
        This method is designed to be conveniently used within a
        django shell, like this:
        >>> from pairent_app.factories import fake;
        >>> fake(42);
    """
    ApartmentFactory.create_batch(size);
    UserFactory.create_batch(size);