From d158cc260b0facf558590f5d67000aafbf323f40 Mon Sep 17 00:00:00 2001 From: Felipe Luis Quezada Valenzuela Date: Fri, 31 Jan 2025 09:53:27 -0300 Subject: [PATCH] #edit defs --- db_setup.py | 3 ++- defs.py | 4 ++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/db_setup.py b/db_setup.py index 769153a..30412e3 100644 --- a/db_setup.py +++ b/db_setup.py @@ -8,8 +8,9 @@ class ModificadosEv(Base): id = Column(Integer, primary_key=True, autoincrement=True) evento_uuid = Column(String, nullable=False) publicado_fecha = Column(DateTime, nullable=False) + attribute_count = Column(Integer, nullable=False) __table_args__ = ( - UniqueConstraint('evento_uuid', 'publicado_fecha', name='ev_pub_fecha'), + UniqueConstraint('evento_uuid', 'attribute_count', name='ev_attr'), ) def get_engine(db_path: str): diff --git a/defs.py b/defs.py index bbe743e..3131147 100644 --- a/defs.py +++ b/defs.py @@ -25,7 +25,7 @@ class MISPProcessor: # Para publicar publica = False - stmt = select(ModificadosEv).where(ModificadosEv.evento_uuid == event_uuid) + stmt = select(ModificadosEv).where(ModificadosEv.evento_uuid == event_uuid, ModificadosEv.attribute_count == int(event_attribute_count)) existe = session.scalars(stmt).first() if existe: @@ -72,7 +72,7 @@ class MISPProcessor: if a["disable_correlation"]: self.misp.direct_call(f"attributes/edit/{a['id']}", {"disable_correlation": False}) publica = True - new_ev = ModificadosEv(evento_uuid=event_uuid, publicado_fecha=datetime.now(timezone.utc)) + new_ev = ModificadosEv(evento_uuid=event_uuid, publicado_fecha=datetime.now(), attribute_count=int(event_attribute_count)) session.add(new_ev) try: session.commit()