def post(self): parser = reqparse.RequestParser() parser.add_argument('message', type=str, required=True, help='Message cannot be blank!' ) parser.add_argument('duration', type=int, required=True, help='Duration cannot be blank!' ) parser.add_argument('message_category', type=str, required=True, help='Message category cannot be blank!' ) args = parser.parse_args() message = MessageModel(message=args['message'], duration=args['duration'], creation_date=datetime.now(utc), message_category=args['message_category'] ) message_manager.insert_message(message) return message, status.HTTP_201_CREATED
def mutate(self, info, input): data = input data['date'] = datetime.datetime.utcnow() message = MessageModel(**data) db_session.add(message) db_session.commit() return SendMessage(message=message)
def mutate(self, info, input): data = input_to_dictionary(input) data['created'] = datetime.utcnow() data['edited'] = datetime.utcnow() message = MessageModel(**data) db_session.add(message) db_session.commit() #middleware 처리할 return CreateMessage(message=message)
def _create_message(self, guest_id, host_id, message): user = "******" if message.user_id() == host_id else "guest" new_msg = MessageModel( guest_id=guest_id, user=user, message=message.message(), channel="airbnb", sent=message.sent(), ) self.db.add_message(host_id, new_msg)
def test_create_message_by_guest(sync, mock_message_one): expected_message = MessageModel( guest_id="002", sent=mock_message_one.sent(), message=mock_message_one.message(), user="******", channel="airbnb", ) sync._create_message("002", "001", mock_message_one) sync.db.add_message.assert_called_once_with("001", expected_message)
def test_update_message_existing_different_message(sync, mock_message_one, mock_message_two): existing_msg = MessageModel( guest_id="002", sent=mock_message_one.sent(), message=mock_message_one.message(), user="******", channel="airbnb", ) sync.db.messages_by_host_guest.return_value = [existing_msg] sync._update_message("002", "001", mock_message_two) expected_message = MessageModel( guest_id="002", sent=mock_message_two.sent(), message=mock_message_two.message(), user="******", channel="airbnb", ) sync.db.add_message.assert_called_once_with("001", expected_message)
def test_update_message_existing_same_message(sync, mock_message_one): existing_msg = MessageModel( guest_id="002", sent=mock_message_one.sent(), message=mock_message_one.message(), user="******", channel="airbnb", ) sync.db.messages_by_host_guest.return_value = [existing_msg] sync._update_message("002", "001", mock_message_one) assert not (sync.db.add_message.called)
def test_update_message_empty_db(sync, mock_message_one): sync.db.messages_by_host_guest.return_value = [] sync._update_message("002", "001", mock_message_one) expected_message = MessageModel( guest_id="002", sent=mock_message_one.sent(), message=mock_message_one.message(), user="******", channel="airbnb", ) sync.db.add_message.assert_called_once_with("001", expected_message)
def messages(self): result = {} data = self._query_table("msg") data = [(MessageModel(**item["itemData"]), item["itemID"]) for item in data] for message, key in data: host_id, guest_id, *_ = key.split("#") if host_id not in result: result[host_id] = {} if guest_id not in result[host_id]: result[host_id][guest_id] = [] result[host_id][guest_id].append(message) return result
def add(): "添加日程" if request.method == "GET": g.title = u"添加日程" g.submit = u"添加" return render_template("notes/edit.html") else: remindDate = request.form.get("remindDate", None) message = request.form.get("message", None) uid = session.get("uid", None) dao = MessageModel(uid, message, remindDate) dbSession = sessionMaker() dbSession.add(dao) try: dbSession.commit() r = True except: dbSession.rollback() r = False finally: dbSession.close() return redirect(url_for("notes.index"))
def messages_by_host_guest(self, host_id: str, guest_id: str): data = self._query_table("msg", f"{host_id}#{guest_id}", ["itemData"]) data = [item["itemData"] for item in data] data = [MessageModel(**message) for message in data] return data