def create_user():
typeUser = request.json['user']
Session = sessionmaker(bind=engine)
session = Session()
if(typeUser == "client"):
new_client = Client(name= request.json["name"],email=request.json["email"],contact=request.json["contact"])
new_client.hash_password(request.json["password"])
try:
session.add(new_client)
session.commit()
except IntegrityError:
return jsonify({'message':'That account already exists'})
return jsonify({'message':'successful insertion of a client'})
elif(typeUser == "owner"):
new_owner = Owner(name= request.json["name"],email=request.json["email"],contact=request.json["contact"])
new_owner.hash_password(request.json["password"])
try:
session.add(new_owner)
session.commit()
except IntegrityError:
return jsonify({'message':'That account already exists'})
return jsonify({'message':'successful insertion of a owner'})
else:
return jsonify({'message':'You didnt choose a user type'})
def create_owner(email, first_name, last_name, password):
owner = Owner(email=email,
first_name=first_name,
last_name=last_name,
password=password)
db.session.add(owner)
db.session.commit()
return owner
def map_row(self, row, metadata=None):
owner = Owner()
owner.id = row[0]
owner.firstName = row[1]
owner.lastName = row[2]
owner.address = row[3]
owner.city = row[4]
owner.telephone = row[5]
return owner
def update_owner_table(self, item):
owner = self.db.query(Owner).filter(
Owner.id == item["repository"]["owner"]["id"]).first()
if not owner:
new_owner = Owner(
id=item["repository"]["owner"]["id"],
login=item["repository"]["owner"]["login"],
html_url=item["repository"]["owner"]["html_url"],
)
self.db.add(new_owner)
self.db.commit()
def load_owners():
"""Load pet owners into database."""
print "Loading owners..."
Owner.query.delete()
owners = [1, 2, 3, 4, 5, 6, 7, 11, 12, 14, 15, 16, 17, 19, 20, 21, 22, 23]
# add user_ids from owners list to db table
for user in owners:
user_id = user
owner = Owner(user_id=user_id)
db.session.add(owner)
db.session.commit()
def load_owners():
"""Load owners from owner.txt into database."""
print "Owners"
for i, row in enumerate(open("data/owner.txt")):
row = row.rstrip()
name, contact = row.split(", ")
owner = Owner(name=name, contact=contact)
# Add to the session.
db.session.add(owner)
# Commit the session/data to the dbase.
db.session.commit()
# Add puppies to the DB
db.session.add_all([rufus, tuna])
db.session.commit()
all_pups = Puppy.query.all()
print(all_pups)
tuna = Puppy.query.filter_by(name='Tuna').first()
print(tuna)
a = Puppy.query.filter_by(id=1).first()
print(a.id)
# Create Owner object
huzaifa = Owner('Huzaifa', tuna.id)
# Giving toys to Tuna
toy1 = Toy('Chew Toy', tuna.id)
toy2 = Toy('Ball', tuna.id)
toy3 = Toy('Roblox', tuna.id)
db.session.add_all([huzaifa, toy1, toy2, toy3])
db.session.commit()
# Grab Tuna again
tuna = Puppy.query.filter_by(name='Tuna').first()
print(tuna)
rufus = Puppy("Rufus")
fido = Puppy("Fido")
#Add puppies to db
db.session.add_all([rufus, fido])
db.session.commit()
print(Puppy.query.all())
rufus = Puppy.query.filter_by(name="Rufus").first()
# Create owner object
jose = Owner("Jose", rufus.id)
#Give Rufus some report_toys
toy1 = Toy("Chew Toy", rufus.id)
toy2 = Toy("Ball", rufus.id)
db.session.add_all([jose, toy1, toy2])
db.session.commit()
#GRAB RUFUS AFTER THOSE ADDITIONS
rufus = Puppy.query.filter_by(name="Rufus").first()
print(rufus)
print(rufus.report_toys)
from model import db, Puppy, Toy, Owner
rufus = Puppy("Rufus")
fido = Puppy("Fido")
db.session.add_all([rufus, fido])
db.session.commit()
puppy = Puppy.query.filter_by(name="Rufus").first()
saurabh = Owner("Saurabh", puppy.id)
toy1 = Toy("Chew", puppy.id)
toy2 = Toy("Phew", puppy.id)
db.session.add_all([saurabh, toy1, toy2])
db.session.commit()
rufus.report_toys()
