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 export_buildings():
i = 0
blist = []
bitems = buildings.items()
log("processing %s buildings"%(len(bitems),), 1)
for b_id, bdata in bitems:
addr = bdata["address"]
building = Building.query(Building.address == addr).get()
owner = Owner.query(Owner.name == bdata["owner"]).get()
byear = bdata["year"]
btype = bdata["building_type"]
if not building:
log("Can't find building '%s' -- creating new entry"%(addr,), 2)
building = Building(address=addr)
if owner:
building.owner = owner.key
if byear:
building.year = byear
if btype:
building.building_type = btype
if b_id:
building.building_id = b_id
if not building.zipcode:
zc = _zip(bdata["zipcode"], addr)
if zc:
building.zipcode = zc
if not building.latitude or not building.longitude:
building.latitude, building.longitude = address2latlng(building.address)
blist.append(building)
i += 1
if not i % 100:
log("processed %s buildings"%(i,), 2)
log("saving buildings", 1)
db.put_multi(blist)
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 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 export_owners():
i = 0
olist = []
oitems = owners.items()
log("processing %s owners"%(len(oitems),), 1)
for name, odata in oitems:
if not name:
log("no name for owner: %s"%(json.dumps(odata),), important=True)
elif not Owner.query(Owner.name == name).get():
log("Can't find owner '%s' -- creating new entry"%(name,), 2)
owner = Owner(name=name)
if odata["address"]:
owner.address, zc = odata["address"].rsplit(" ", 1)
zcode = _zip(zc, owner.address)
if zcode:
owner.zipcode = zcode
else:
log("no address for %s!"%(name,), important=True)
olist.append(owner)
i += 1
if not i % 100:
log("processed %s owners"%(i,), 2)
log("saving %s owners"%(len(olist),), 1)
db.put_multi(olist)
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()
def get(self, ownerUID, format="json"):
if len(ownerUID) == 0:
simplewebapp.formatResponse(format, self, "FAILED")
return
o = Owner.all().filter("uid =", ownerUID).fetch(1)
if len(o) != 1:
simplewebapp.formatResponse(format, self, "FAILED")
return
owner = o[0]
q = self.request.get("q").lower()
# VEERY SLOW ! OPTIMIZE!
ownEntries = db.GqlQuery("SELECT * FROM Entry WHERE owners = :1", owner).run()
tagsRaw = getTagTerms(q)
results = []
if len(q) != 0:
results = findEntries(ownEntries, tagsRaw)
simplewebapp.formatResponse(format, self, results)
# 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()
