fixed Origin.delete()

This commit is contained in:
Oscar Krause 2023-01-23 07:12:02 +01:00
parent bed24b56ce
commit f7ef8d76b6

View File

@ -57,12 +57,12 @@ class Origin(Base):
session.close() session.close()
@staticmethod @staticmethod
def delete(engine: Engine, origins: ["Origin"] = None) -> int: def delete(engine: Engine, origin_refs: [str] = None) -> int:
session = sessionmaker(bind=engine)() session = sessionmaker(bind=engine)()
if origins is None: if origin_refs 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 in origin_refs).delete()
session.commit() session.commit()
session.close() session.close()
return deletions return deletions