from sqlalchemy import and_ from orm.rating import is_negative, is_positive from orm.reaction import Reaction, ReactionKind from orm.shout import Shout from services.db import local_session from utils.diff import apply_diff, get_diff def handle_proposing(kind: ReactionKind, reply_to: int, shout_id: int): with local_session() as session: if is_positive(kind): replied_reaction = ( session.query(Reaction).filter(Reaction.id == reply_to, Reaction.shout == shout_id).first() ) if replied_reaction and replied_reaction.kind is ReactionKind.PROPOSE.value and replied_reaction.quote: # patch all the proposals' quotes proposals = ( session.query(Reaction) .filter( and_( Reaction.shout == shout_id, Reaction.kind == ReactionKind.PROPOSE.value, ) ) .all() ) # patch shout's body shout = session.query(Shout).filter(Shout.id == shout_id).first() body = replied_reaction.quote Shout.update(shout, {body}) session.add(shout) session.commit() # реакция содержит цитату -> обновляются все предложения # (proposals) для соответствующего Shout. for proposal in proposals: if proposal.quote: proposal_diff = get_diff(shout.body, proposal.quote) proposal_dict = proposal.dict() proposal_dict["quote"] = apply_diff(replied_reaction.quote, proposal_diff) Reaction.update(proposal, proposal_dict) session.add(proposal) if is_negative(kind): # TODO: rejection logic pass