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