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()])
