seed_db.py

"""Script to seed database."""

import os
import json
from random import choice, randint
from datetime import datetime

import crud 
import model
import server

# os.system('dropdb testdb')
# os.system('createdb testdb')

model.connect_to_db(server.app)
# model.db.create_all() # <-- can un-comment later but good to get in the habit
# # of entering this into bash/terminal


# Create 10 test users
users_in_db = []

for n in range(10):
    email = f'user{n}@test.com'  # A unique email
    username = f'user_{n}'  # A unique username
    password = 'testpwd'

    user = crud.create_user(email, username, password)

    users_in_db.append(user)


# Create 10 test landlords
landlords_in_db = []

for n in range(10):
    landlord_name = f'Test Landlord {n}'
    street_num = randint(10, 2500)
    office_address = f'{street_num} Test Address Lane'

    landlord = crud.create_landlord(landlord_name, office_address)

    landlords_in_db.append(landlord)


# Create 5 test buildings
buildings_in_db = []
for n in range(10):
    building_housenumber = randint(10, 2500)
    building_streetname = "Renters Road"
    building_postcode = randint(10100, 99999)
    landlord_to_own = choice(landlords_in_db)
    landlord_id=landlord_to_own.landlord_id

    building = crud.create_building(building_housenumber, building_streetname, 
                                    building_postcode, landlord_id)

    buildings_in_db.append(building)


# Create 10 test reviews

# Example review creation from terminal:
# test_review=Review(review_body="This is a test review.",
#                   created_on=datetime.datetime(2020, 5, 23), 
#                   updated_at=datetime.datetime(2020, 5, 23), 
#                   user_id=test_user.user_id, 
#                   building_id=test_building.building_id)

for n in range(25):
    review_body = f'This is review number {n}. This landlord...'

    # create_yr = randint(2017, 2020)
    # create_mth = randint(1, 12)
    # create_day = randint(1, 28)
    # created_on = datetime(create_yr, create_mth, create_day)
    
    # updated_at = datetime.now()

    user_to_write = choice(users_in_db)
    user_id=user_to_write.user_id

    building_to_write = choice(buildings_in_db)
    building_id=building_to_write.building_id

    review = crud.create_review(review_body, user, building_to_write)