Skip to content

Commit 8208b8f

Browse files
improve script
1 parent 0ece3c9 commit 8208b8f

File tree

1 file changed

+9
-19
lines changed

1 file changed

+9
-19
lines changed

cads_broker/entry_points.py

+9-19
Original file line numberDiff line numberDiff line change
@@ -139,33 +139,23 @@ def delete_requests(
139139
minutes=minutes, seconds=seconds, hours=hours, days=days
140140
)
141141
with database.ensure_session_obj(None)() as session:
142-
database.logger.info(f"deleting old system_requests before {timestamp}.")
142+
database.logger.info(f"deleting {status} system_requests before {timestamp}.")
143143
statement = (
144-
sa.select(database.SystemRequest)
144+
sa.delete(database.SystemRequest)
145145
.where(database.SystemRequest.status == status)
146146
.where(database.SystemRequest.created_at < timestamp)
147147
)
148-
requests = session.scalars(statement).all()
149-
number_of_requests = len(requests)
148+
number_of_requests = session.execute(statement).rowcount
150149
if not skip_confirmation:
151-
typer.confirm(
152-
f"Deleting {number_of_requests} requests. Do you want to continue?",
150+
if not typer.confirm(
151+
f"Deleting {number_of_requests} {status} requests. Do you want to continue?",
153152
abort=True,
154153
default=True,
155-
)
156-
else:
157-
database.logger.info(f"Deleting {number_of_requests} requests.")
158-
for request in requests:
159-
database.logger.info(f"deleting {request.request_uid}...")
160-
database.set_request_status(
161-
request.request_uid, "dismissed", session=session
162-
)
163-
154+
):
155+
typer.echo("Operation cancelled.")
156+
return
164157
session.commit()
165-
166-
database.logger.info(
167-
f"{number_of_requests} requests successfully removed from the broker database."
168-
)
158+
typer.echo(f"{number_of_requests} requests successfully removed from the broker database.")
169159

170160

171161
@app.command()

0 commit comments

Comments
 (0)