def send_feed(uid: int, top: bool, content: str) -> int: """ 发送feed,返回feed_id """ feed = Feed(uid=uid, content=content, top=top) db.session.add(feed) db.session.commit() return feed.id
def add_message(): ref = request.referrer form = FeedPostForm() formModalMessage = PrivateMessageForm() if form.validate_on_submit(): fromUser = User.objects.get(username=session.get('username')) fetchNotifications(fromUser) toUser = User.objects.get(username=request.values.get('toUser')) image = None if request.files.get('image'): filename = secure_filename(form.image.data.filename) file_path = "images/posts/" + str(uuid.uuid4()) + filename file_pathB = "static/" + file_path form.image.data.save(file_pathB) image = str(file_path) print(image) print("Image gotten") print("AfterImage") print(image) post = form.post.data if toUser == fromUser: toUser = None message = Message( fromUser=fromUser, toUser=toUser, text=post, messageType=POST, ).save() feed = Feed(user=fromUser, message=message).save() if image: message.image = image message.save() process_message(message) if ref: return redirect(ref) else: return redirect(url_for('home_app.home')) else: return 'Error!'
def process_message(message): fromUser = message.fromUser friends = Relationship.objects.filter(fromUser=fromUser, rel_type=Relationship.FRIENDS, status=Relationship.APPROVED) for friend in friends: rel = Relationship.get_relationship(friend.toUser, message.toUser) if rel != "BLOCKED": feed = Feed(user=friend.toUser, message=message).save() return True
def main(): result = Feed(fetch_feed()) db = Sqlite3() db.create_table() db.insert_records([item.get_as_record() for item in result.get_entries()])