diff --git a/app/orm.py b/app/orm.py index 387c828..6f4858f 100644 --- a/app/orm.py +++ b/app/orm.py @@ -57,12 +57,12 @@ class Origin(Base): session.close() @staticmethod - def delete(engine: Engine, origins: ["Origin"] = None) -> int: + def delete(engine: Engine, origin_refs: [str] = None) -> int: session = sessionmaker(bind=engine)() - if origins is None: + if origin_refs is None: deletions = session.query(Origin).delete() 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.close() return deletions