def db_spammer(amount, teg_list=None): if teg_list is None: teg_list = ['Science', 'Math', 'Style', 'Drugs', 'Tech', 'Linux'] nickname = [ 'FightForDobro', 'iLikeIceCream', 'Yoda', 'Albert Hoffman', 'Alexander Theodore Shulgin', 'Timothy Francis Leary' ] if len(User.objects().to_json()) == 2: for i in nickname: user_dict = { 'nickname': i, 'name': get_first_name(), 'surname': get_last_name() } User(**user_dict).save() for _ in range(amount): teg_obj = Teg(teg_title=choice(teg_list)) post_dict = { 'post_title': lorem.sentence(), 'content': lorem.paragraph(), 'teg': teg_obj, } user = User.objects(nickname=choice(nickname)).get() user.create_post(**post_dict)
def delete(self, p_id): user_data = request.json[0].pop('author') user = User.objects(id=user_data['id']).get() user.delete_post(p_id) return {'Post deleted successful': request.json[0]}
def post(self): user_data = request.json[0].pop('author') user = User.objects(id=user_data['id']).get() user.create_post(**request.json[0]) return {'Post create successful': request.json[0]}
def get(self, nickname=None): if nickname: user = User.objects(nickname=nickname).get() return PostScheme().dump(Post.objects(author=user), many=True) objects = User.objects return UserScheme().dump(objects, many=True)
def delete(self, nickname): obj = User.objects(nickname=nickname).get() obj.delete() return {nickname: 'DELETED'}
def put(self, nickname): obj = User.objects(nickname=nickname).get() obj.update(**request.json) return UserScheme().dump(obj.reload())
def post(self): obj = User(**request.json) obj.save() return UserScheme().dump(obj.reload())