def location():
drop_all_tables(test_conn, test_cursor)
new_york = save(State(name='New York'), test_conn, test_cursor)
pennsylvania = save(State(name='Pennsylvania'), test_conn, test_cursor)
brooklyn = save(City(name='Brooklyn', state_id=new_york.id), test_conn,
test_cursor)
manhattan = save(City(name='Manhattan', state_id=new_york.id), test_conn,
test_cursor)
philadelphia = save(City(name='Philadelphia', state_id=pennsylvania.id),
test_conn, test_cursor)
south_philly_zip = save(Zipcode(code=19019, city_id=philadelphia.id),
test_conn, test_cursor)
chelsea_zip = save(Zipcode(code=10001, city_id=manhattan.id), test_conn,
test_cursor)
dumbo_zip = save(Zipcode(code=11210, city_id=brooklyn.id), test_conn,
test_cursor)
venue = save(Venue(name='Los Tacos Al Pastor', price=1), test_conn,
test_cursor)
location = save(
Location(longitude=40.7024,
latitude=-73.9875,
address='141 Front Street',
zipcode_id=dumbo_zip.id,
venue_id=venue.id), test_conn, test_cursor)
yield location
drop_all_tables(test_conn, test_cursor)
def build_category():
db.drop_all_tables(test_conn, test_cursor)
category = models.Category()
category.name = 'Taco Places'
db.save(category, test_conn, test_cursor)
yield
db.drop_all_tables(test_conn, test_cursor)
def tourist_spot():
db.drop_all_tables(test_conn, test_cursor)
new_york = db.save(models.State(name='New York'), test_conn, test_cursor)
pennsylvania = db.save(models.State(name='Pennsylvania'), test_conn,
test_cursor)
brooklyn = db.save(models.City(name='Brooklyn', state_id=new_york.id),
test_conn, test_cursor)
manhattan = db.save(models.City(name='Manhattan', state_id=new_york.id),
test_conn, test_cursor)
philadelphia = db.save(
models.City(name='Philadelphia', state_id=pennsylvania.id), test_conn,
test_cursor)
south_philly_zip = db.save(
models.Zipcode(code=19019, city_id=philadelphia.id), test_conn,
test_cursor)
chelsea_zip = db.save(models.Zipcode(code=10001, city_id=manhattan.id),
test_conn, test_cursor)
dumbo_zip = db.save(models.Zipcode(code=11210, city_id=brooklyn.id),
test_conn, test_cursor)
venue = db.save(
models.Venue(name='Los Tacos Al Pastor',
price=1,
foursquare_id='4bf58dd8d48988d151941735'), test_conn,
test_cursor)
grimaldis = db.save(models.Venue(name='Grimaldis', price=2), test_conn,
test_cursor)
pizza = db.save(models.Category(name='Pizza'), test_conn, test_cursor)
tourist_spot = db.save(models.Category(name='Tourist Spot'), test_conn,
test_cursor)
db.save(models.VenueCategory(venue_id=grimaldis.id, category_id=pizza.id),
test_conn, test_cursor)
db.save(
models.VenueCategory(venue_id=grimaldis.id,
category_id=tourist_spot.id), test_conn,
test_cursor)
db.save(
models.VenueCategory(venue_id=venue.id, category_id=tourist_spot.id),
test_conn, test_cursor)
grimaldi_location = db.save(
models.Location(longitude=40.7024,
latitude=-73.9875,
address='1 Front Street',
zipcode_id=dumbo_zip.id,
venue_id=grimaldis.id), test_conn, test_cursor)
taco_location = db.save(
models.Location(longitude=40.7024,
latitude=-73.9875,
address='141 Front Street',
zipcode_id=dumbo_zip.id,
venue_id=venue.id), test_conn, test_cursor)
yield tourist_spot
db.drop_all_tables(test_conn, test_cursor)
def test_conn():
# reference in user and pasword from .env/settings.py
test_conn = psycopg2.connect(dbname='mobilegaming_test',
user='******',
password='******')
cursor = test_conn.cursor()
db.drop_all_tables(test_conn, cursor)
db.reset_all_primarykey(test_conn, cursor)
yield test_conn
db.drop_all_tables(test_conn, cursor)
db.reset_all_primarykey(test_conn, cursor)
def clean_tables():
db.drop_all_tables(test_conn, test_cursor)
yield
db.drop_all_tables(test_conn, test_cursor)
"""
seed.py file used to seed data into psql tables.
fixed drop_all_tables() -> drop_records() using TRUNCATE to delete table with foreign key
added reset_all_primarykey to change primary keys back to 1 every time seed.py is run
"""
import psycopg2
import api.src.db as db
import api.src.models as models
# delete/truncate all tables and reset primary keys
db.drop_all_tables(db.conn, db.cursor)
db.reset_all_primarykey(db.conn, db.cursor)
# seeding games
amongus = db.save(models.Game(name = 'Among Us', platform = 'android', publisher = 'Innersloth LLC',
release_date = '2018-07-25', genre = 'action', game_engine = 'Unity'), db.conn, db.cursor)
roblox = db.save(models.Game(name = 'Roblox', platform = 'android', publisher = 'Roblox Corporation',
release_date = '2019-12-31', genre = 'building', game_engine = 'Something Else'), db.conn, db.cursor)
candy = db.save(models.Game(name = 'Candy Crush Saga', platform = 'android', publisher = 'King',
release_date = '2015-1-1', genre = 'casual', game_engine = 'Candy Factory'), db.conn, db.cursor)
amongus_ios = db.save(models.Game(name = 'Among Us', platform = 'iOS', publisher = 'Innersloth LLC',
release_date = '2018-11-2', genre = 'action', game_engine = 'Unity'), db.conn, db.cursor)
# breakpoint()
# seeding ratings
no1 = db.save(models.Rating(game_id = amongus.id, metacritic = 84, TS_rating = 4.465,