cli-delete-posts
Added the possibility to delete single posts or a range of multiple posts at once using the command as following:<br/> `docker-compose run server ./szuru-admin --delete posts 35 36 40-45`<br/> Each space represents a single post id, using a `-` between two ids will delete all posts within range, including the first and the last one
This commit is contained in:
parent
6c3b50d287
commit
3fbe3a0b93
1 changed files with 51 additions and 1 deletions
|
@ -13,7 +13,7 @@ from getpass import getpass
|
|||
from sys import stderr
|
||||
|
||||
from szurubooru import config, db, errors, model
|
||||
from szurubooru.func import files, images
|
||||
from szurubooru.func import files, images, snapshots
|
||||
from szurubooru.func import posts as postfuncs
|
||||
from szurubooru.func import users as userfuncs
|
||||
|
||||
|
@ -100,6 +100,48 @@ def regenerate_thumbnails() -> None:
|
|||
pass
|
||||
|
||||
|
||||
def delete_posts(parameters: list) -> None:
|
||||
verification: str = "y"
|
||||
|
||||
while "" == verification:
|
||||
verification = input("Do you really want to delete all posts with the given ID's [y/n]: ")
|
||||
|
||||
if "y" != verification.lower():
|
||||
return
|
||||
|
||||
def delete_one_post(post_id: int) -> None:
|
||||
print("Deleting post %d" % post_id)
|
||||
|
||||
try:
|
||||
post: model.Post = postfuncs.get_post_by_id(post_id)
|
||||
except postfuncs.PostNotFoundError:
|
||||
print("Post with ID %d not found" % post_id)
|
||||
return
|
||||
|
||||
postfuncs.delete(post)
|
||||
snapshots.delete(post, None)
|
||||
|
||||
def delete_multiple_posts(start_id: int, end_id: int) -> None:
|
||||
for post_id in range(start_id, end_id + 1):
|
||||
delete_one_post(post_id)
|
||||
|
||||
for parameter in parameters:
|
||||
try:
|
||||
if "-" not in parameter:
|
||||
delete_one_post(int(parameter))
|
||||
continue
|
||||
|
||||
post_range: list = [int(number) for number in parameter.split("-", 2)]
|
||||
delete_multiple_posts(*post_range)
|
||||
except ValueError:
|
||||
print("One of the specified parameters is not a number")
|
||||
return
|
||||
|
||||
db.get_session().commit()
|
||||
|
||||
print("All posts were deleted")
|
||||
|
||||
|
||||
def main() -> None:
|
||||
parser_top = ArgumentParser(
|
||||
description="Collection of CLI commands for an administrator to use",
|
||||
|
@ -129,6 +171,12 @@ def main() -> None:
|
|||
help="regenerate the thumbnails for posts if the "
|
||||
"thumbnail files are missing",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--delete-posts",
|
||||
metavar="<post_ids>",
|
||||
nargs='+',
|
||||
help="Will delete all posts with the given id's"
|
||||
)
|
||||
command = parser_top.parse_args()
|
||||
|
||||
try:
|
||||
|
@ -140,6 +188,8 @@ def main() -> None:
|
|||
reset_filenames()
|
||||
elif command.regenerate_thumbnails:
|
||||
regenerate_thumbnails()
|
||||
elif command.delete_posts:
|
||||
delete_posts(command.delete_posts)
|
||||
except errors.BaseError as e:
|
||||
print(e, file=stderr)
|
||||
|
||||
|
|
Loading…
Reference in a new issue