def get(self):
dir = "model/data"
with open('%s/User.csv'%(dir)) as csvfile:
data = csv.reader(csvfile)
for row in data:
User.create(
auth_id = row[0],
password_raw = row[1],
name = row[2],
email_address = row[3],
lang = row[4],
country = row[5],
role = row[6]
)
with open('%s/Pet.csv'%(dir)) as csvfile:
data = csv.reader(csvfile)
entities = []
for row in data:
pet = Pet(
name=row[0],
pet_type=ndb.Key(PetType, row[1]),
pet_category=ndb.Key(PetCategory, row[2]),
avatar=row[4],
birth=row[5],
sex=row[6],
owner=ndb.Key(User,row[7]),
status=row[8],
user= [
ndb.Key(User,row[9])
]
)
if row[10]:
pet.user += [ndb.Key(User,row[10]),]
if row[3]:
pet.breed=ndb.Key(Breed, row[3])
entities += [pet,]
ndb.put_multi(entities)
def update(self):
name = self.request.get('name')
pet_type = self.request.get('pet_type')
pet_category = self.request.get('pet_category')
breed = self.request.get('breed')
avatar = self.request.get('avatar')
birth = self.request.get('birth')
sex = self.request.get('sex')
user = self.user
owner_id = user.get_id()
pet = Pet(
name = name,
pet_type = pet_type,
pet_category = pet_category,
breed = breed,
avatar = avatar,
birth = birth,
sex = sex,
owner_id = owner_id
)
pet.update()
def get(self, *args, **kwargs):
pet_id = kwargs['pet_id']
data = {}
data["pet"] = Pet.get(pet_id)
self.SendJson(data)
