def delete_msg_file_relation(session: DBSession, user_id: int, msg_id: int, file_id: int) -> None:
db_msgfile = session.get_msgfile_by_msgfile_ids(msg_id, file_id)
if db_msgfile is None:
raise DBMsgFileNotExistsException
db_file = session.get_file_by_id(file_id)
if db_file.sender_id != user_id:
raise DBResourceOwnerException
db_msgfile.is_delete = True
def create_msg_file_relation(session: DBSession, user_id: int, msg_id: int, file_id: int) -> DBMsgFile:
db_file = session.get_file_by_id(file_id)
if db_file is None:
raise DBFileNotExistsException
if db_file.sender_id != user_id:
raise DBResourceOwnerException
msg_file = DBMsgFile(
msg_id=msg_id,
file_id=file_id,
)
session.add_model(msg_file)
return msg_file
def get_file_by_id(session: DBSession, file_id: int) -> DBFile:
file = session.get_file_by_id(file_id)
if file is None:
raise DBFileNotExistsException
return file