def _make_apartment(user, addr, rent, roomates, rooms, s_date):
apartmant = Apartment(owner=user,
city=city_model,
address=addr,
rent=rent,
num_of_roomates=roomates,
num_of_rooms=rooms,
start_date=s_date)
apartmant.save()
return apartmant
def apartment_model(db, city_model):
new_owner = get_user_model().objects.create_user(
'*****@*****.**', 'owner', 'own', '1900-01-01', 'password')
new_owner.save()
new_apartment = Apartment(
owner=new_owner,
city=city_model,
address='street',
rent=4500,
num_of_roomates=2,
num_of_rooms=3,
start_date='2021-1-1',
about='Hey!',
image_url='www.some-url.com',
)
new_apartment.save()
return new_apartment
def apart_success_search(db, city_model_Tel_Aviv):
new_owner = get_user_model().objects.create_user(
'*****@*****.**', 'owner', 'own',
'1900-01-01', 'password')
new_owner.save()
new_apartment = Apartment(
owner=new_owner,
city=city_model_Tel_Aviv,
address='street',
rent=2500,
num_of_roomates=2,
num_of_rooms=3,
start_date='2021-03-02',
about='Hey!',
image_url='www.some-url.com',
)
new_apartment.save()
return new_apartment
def generate_apartment_data(apps, schema_editor):
apt_test_data = [
(User.objects.get(email='*****@*****.**'),
City.objects.get(cityName='Tel Aviv'), 'Israelof 5', 2650, 2, 3,
'2020-01-01', "Join our cool aprt!"),
(User.objects.get(email='*****@*****.**'),
City.objects.get(cityName='Jerusalem'), 'Yoseftal 6', 3200, 3, 4,
'2020-02-02', "Good vibes only"),
]
with transaction.atomic():
for baseuser, city, address, rent, roomates, rooms, startdate, about in apt_test_data:
Apartment(
owner=baseuser,
city=city,
address=address,
rent=rent,
num_of_roomates=roomates,
num_of_rooms=rooms,
start_date=startdate,
about=about,
image_url=random.choice(APARTMENT_IMAGE_URL_LIST)).save()
def generate_more_apartment_data(apps, schema_editor):
users = User.objects.all()[100:]
cities = City.objects.all()
with transaction.atomic():
for user in users:
start_date = "2021-" + str(random.randint(1, 12)) + \
"-" + str(random.randint(1, 28))
address = random.choice(STREETS_LIST) + ", " + \
str(random.randint(1, 150))
num_of_roomates = random.randint(2, 4)
num_of_rooms = num_of_roomates + random.randint(0, 3)
Apartment(
owner=user,
city=cities[random.randint(0, 5)],
address=address,
rent=random.randint(1500, 5000),
num_of_roomates=num_of_roomates,
num_of_rooms=num_of_rooms,
start_date=start_date,
about=random.choice(APARTMENT_ABOUT),
image_url=random.choice(APARTMENT_IMAGE_URL_LIST)).save()