server/tools: add generate-thumb

1 files changed, 27 insertions, 0 deletions

diff --git a/server/generate-thumb b/server/generate-thumb
new file mode 100755
index 0000000..87d7e02
--- /dev/null
+++ b/server/generate-thumb
@@ -0,0 +1,27 @@
+#!/usr/bin/env python3
+
+'''
+Generates thumbnails for posts from CLI. Useful for testing changes to
+thumbnail generators, and for weird inputs.
+'''
+
+import argparse
+import os.path
+import sys
+from szurubooru.func import posts
+
+def main():
+    parser = argparse.ArgumentParser('Starts szurubooru using waitress.')
+    parser.add_argument('post_id', metavar='POST', help='post to generate thumbnail for')
+    args = parser.parse_args()
+
+    try:
+        post = posts.get_post_by_id(args.post_id)
+        posts.generate_post_thumbnail(post)
+    except posts.PostNotFoundError:
+        pass
+    except:
+        raise
+
+if __name__ == '__main__':
+    main()