implemented delete origin endpoint for frontend

This commit is contained in:
Oscar Krause 2023-01-04 09:52:51 +01:00
parent 5035dd4947
commit 41cb7ae14f

View File

@ -56,12 +56,12 @@ class Origin(Base):
session.close() session.close()
@staticmethod @staticmethod
def delete(engine: Engine, origins: ["Origin"] = None) -> int: def delete(engine: Engine, origin_ref: str = None) -> int:
session = sessionmaker(bind=engine)() session = sessionmaker(bind=engine)()
if origins is None: if origin_ref is None:
deletions = session.query(Origin).delete() deletions = session.query(Origin).delete()
else: else:
deletions = session.query(Origin).filter(Origin.origin_ref in origins).delete() deletions = session.query(Origin).filter(Origin.origin_ref == origin_ref).delete()
session.commit() session.commit()
session.close() session.close()
return deletions return deletions