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)