def handle(self, *args, **options):
seed = self.seeder
gen_dict = self.seed_dict
invalid_randomizers = seed.clean(gen_dict)
#generate numentries records in the database
for _ in range(options['numentries']):
#separate document for non-random fields in seeding process
Document = {
'twitter': " ",
'instagram': " ",
'external_delivery_link': " ",
'rating': "0.00"
}
rand_Document = seed.gen_rand_dict(gen_dict)
Document.update(rand_Document)
Restaurant.insert(Document)
def insert_restaurant_page(request):
"""Insert new restaurant into database"""
validate(instance=request.body, schema=restaurant_schema)
body = json.loads(request.body)
invalid = Restaurant.field_validate(body)
if invalid:
return JsonResponse(invalid)
try:
restaurant = Restaurant.insert(body)
return JsonResponse(model_to_json(restaurant))
except ValueError:
return HttpResponseBadRequest('duplicate email')
def reassign_page(request):
"""
Page to change the role of a user provided the user email and new role (If upgraded to RO must
provide fields to make new restaurant instance
"""
validate(instance=request.body, schema=signup_schema)
body = json.loads(request.body)
user = SDUser.objects.get(pk=body['user_email'])
user.reassign_role(body['role'])
if body['role'] == "RO":
del body['user_email']
del body['role']
restaurant = Restaurant.insert(body)
user.restaurant_id = str(restaurant._id)
user.save(update_fields=["restaurant_id"])
return JsonResponse({"restaurant_id": str(restaurant._id)})
else:
user.restaurant_id = None
user.save(update_fields=["restaurant_id"])
return HttpResponse(status=200)
