"""Load locations from top_locations.csv into boulders db"""

# open file

with open('data_seed_files/top_locations.csv', 'r,') as csvfile:

#checks to see if the file has a header

has_header = csv.Sniffer().has_header(csvfile.read(1024))

#resets the file back to the start

csvfile.seek(0)

# read in file

my_file = csv.reader(csvfile, delimiter=',')

#skips the header

if has_header:

next(my_file) #skip header row

# loop through the file

for row in my_file:

# unpack each row

parent_name, name, lat_long, desc, directions, sub_locations = row

# print parent_name

# print name

# print lat_long

# print desc

# print directions

# print sub_locations

lat, lon = lat_long.split(', ')

# print lat

# print lon

# create a new location object

location = Location(location_name=name,

location_directions=directions,

location_description=desc,

latitude=lat,

longitude=lon)

# add it to db

db.session.add(location)

# commit the db when done.

"""Load sub_locations from sub_locations.csv into boulders db"""

# open file

with open('data_seed_files/sub_locations.csv', 'r,') as csvfile:

#checks to see if the file has a header

has_header = csv.Sniffer().has_header(csvfile.read(1024))

#resets the file back to the start

csvfile.seek(0)

# read in file

my_file = csv.reader(csvfile, delimiter=',')

#skips the header

if has_header:

next(my_file) #skip header row

# loop through the file

for row in my_file:

# unpack each row

parent_name, name, lat_long, desc, directions, sub_loc = row

# print parent_name

# print name

# print lat_long

# print desc

# print directions

# print sub_loc

if lat_long:

lat, lon = lat_long.split(', ')

else:

lat = ""

lon = ""

# print temp

# print type(temp)

# lat= temp[0] #WHAT THE HELL? WHY? ASK FOR HELP IN AM

# lon=temp[-1]

# print lat

# print lon

#use flask-sqlalchemy to query the db

# query the db to get the location_id of a matching location name

parent = Location.query.filter_by(location_name=parent_name).first()

# print parent

parent_id = parent.location_id

# print parent_id

# create a new sub_location object

sub = Sub_location(sub_location_name=name,

sub_location_directions=directions,

sub_location_description=desc,

sub_latitude=lat,

sub_longitude=lon,

location_id=parent_id)

# add it to db

db.session.add(sub)

# commit the db when done.

"""Load boulders from boulders.csv"""

# open file

with open('data_seed_files/boulders.csv', 'r,') as csvfile:

#checks to see if the file has a header

has_header = csv.Sniffer().has_header(csvfile.read(1024))

#resets the file back to the start

csvfile.seek(0)

# read in file

my_file = csv.reader(csvfile, delimiter=',')

#skips the header

if has_header:

next(my_file) #skip header row

# loop through the file

for row in my_file:

# unpack each row

parent_name, name, lat_long, desc, directions, routes = row

parent_name = parent_name.strip()

# print "\n\n"

# print parent_name

# print name

# print lat_long

# print desc

# print directions

# print routes

#checks if lat_long is empty string

if lat_long:

lat, lon = lat_long.split(', ')

else:

lat = ""

lon = ""

# query the db to get the sub_location_id of a matching location name

parent = Sub_location.query.filter_by(sub_location_name=parent_name).first()

# print "sub location look", parent

# if yes get the sub_location_id and the sub_locations location_id

if parent:

parent_id = parent.sub_location_id

g_parent_id = parent.location_id

# print "parent: %i g_parent: %i" % (parent_id, g_parent_id)

# create a new boulder object

rock = Boulder(boulder_name=name,

boulder_description=desc,

boulder_directions=directions,

boulder_latitude=lat,

boulder_longitude=lon,

location_id=g_parent_id,

sub_location_id=parent_id)

#add to db

db.session.add(rock)

# if there is no matching sub_location, query the db for a matching location

# name and retreive the location_id

if not parent:

parent = Location.query.filter_by(location_name=parent_name).first()

# print "location look", parent

parent_id = parent.location_id

# print "parent: %i " % (parent_id)

# create a new boulder object

rock = Boulder(boulder_name=name,

boulder_description=desc,

boulder_directions=directions,

boulder_latitude=lat,

boulder_longitude=lon,

location_id=parent_id)

#add to db

db.session.add(rock)

# commit the db when done.

"""Load routes from bouldering_routes2.csv, bouldering_routes_1.csv into db"""

# print the_file

# print "\n\n"

# open file

with open(the_file, 'r,') as csvfile:

#checks to see if the file has a header

has_header = csv.Sniffer().has_header(csvfile.read(1024))

#resets the file back to the start

csvfile.seek(0)

# read in file

my_file = csv.reader(csvfile, delimiter=',')

#skips the header

if has_header:

next(my_file) #skip header row

# loop through the file

for row in my_file:

# unpack each row

parent_name, name, difficulty, desc, directions, protection = row

parent_name = parent_name.strip()

# print parent_name

# print name

# print difficulty

# print desc

# print directions

# print protection

# print "\n\n"

# query the db to get the boulder_id of a matching boulder name

parent = Boulder.query.filter_by(boulder_name=parent_name).first()

# print type(parent)

# print parent

parent_id=parent.boulder_id

# print parent_id

# create a new route object

route = Route(route_name=name,

difficulty_rate=difficulty,

route_directions=directions,

route_description=desc,

route_protection=protection,

boulder_id=parent_id)

# add it to db

db.session.add(route)

# commit the db when done.