From 369ddaf2f86f1a596716012d295910f38f859bcc Mon Sep 17 00:00:00 2001
From: Shyam Sunder <sgsunder1@gmail.com>
Date: Tue, 3 Sep 2019 14:35:57 -0400
Subject: [PATCH] server/tools: add tool to change post filenames due to
 changed secret

---
 server/szuru-admin | 57 ++++++++++++++++++++++++++++++++++++++--------
 1 file changed, 48 insertions(+), 9 deletions(-)

diff --git a/server/szuru-admin b/server/szuru-admin
index 5fe54c16..429dc831 100755
--- a/server/szuru-admin
+++ b/server/szuru-admin
@@ -5,13 +5,16 @@ Collection of CLI commands for an administrator to use
 '''
 
 import logging
+import os
+import re
 import time
 from argparse import ArgumentParser
 from getpass import getpass
 from sys import stderr
 
-from szurubooru import db, errors, model
-from szurubooru.func import files, images, posts as postfuncs, users as userfuncs
+from szurubooru import config, db, errors, model
+from szurubooru.func import files, images, \
+    posts as postfuncs, users as userfuncs
 
 
 def reset_password(username: str) -> None:
@@ -30,10 +33,10 @@ def reset_password(username: str) -> None:
 
 def check_audio() -> None:
     post_list = (db.session
-        .query(model.Post)
-        .filter(model.Post.type == model.Post.TYPE_VIDEO)
-        .order_by(model.Post.post_id)
-        .all())
+                 .query(model.Post)
+                 .filter(model.Post.type == model.Post.TYPE_VIDEO)
+                 .order_by(model.Post.post_id)
+                 .all())
 
     for post in post_list:
         print('Checking post %d ...' % post.post_id, end='\r')
@@ -43,7 +46,8 @@ def check_audio() -> None:
         try:
             has_sound_data = images.Image(content).check_for_sound()
         except errors.ProcessingError:
-            print('Post %d caused an error when checking for sound' % post.post_id)
+            print('Post %d caused an error when checking for sound' %
+                  post.post_id)
 
         if has_sound_data and not has_existing_flag:
             print('Post %d has sound data but is not flagged' % post.post_id)
@@ -51,15 +55,48 @@ def check_audio() -> None:
             print('Post %d has no sound data but is flagged' % post.post_id)
 
 
+def reset_filenames() -> None:
+    regex = re.compile(r'(\d+)_[0-9a-f]{16}\.(\S+)')
+
+    def convert_to_new_filename(old_name: str) -> str:
+        matches = regex.match(old_name)
+        if not matches:
+            raise ValueError('Not a valid filename')
+        post_id = int(matches.group(1))
+        post_ext = matches.group(2)
+        return '%d_%s.%s' % \
+            (post_id, postfuncs.get_post_security_hash(post_id), post_ext)
+
+    def rename_in_dir(dir: str) -> None:
+        for file in os.listdir(config.config['data_dir'] + dir):
+            try:
+                old_path = file
+                new_path = convert_to_new_filename(file)
+            except ValueError:
+                continue
+            if old_path != new_path:
+                print('%s -> %s' % (dir + old_path, dir + new_path))
+                os.rename(config.config['data_dir'] + dir + old_path,
+                          config.config['data_dir'] + dir + new_path)
+
+    rename_in_dir('posts/')
+    rename_in_dir('generated-thumbnails/')
+    rename_in_dir('posts/custom-thumbnails/')
+
+
 def main() -> None:
     parser_top = ArgumentParser(
         description='Collection of CLI commands for an administrator to use',
         epilog='Look at README.md for more info')
     parser = parser_top.add_mutually_exclusive_group(required=True)
     parser.add_argument('--change-password', metavar='<username>',
-        help='change the password of specified user')
+                        help='change the password of specified user')
     parser.add_argument('--check-all-audio', action='store_true',
-        help='check the audio flags of all posts, noting discrepancies, without modifying posts')
+                        help='check the audio flags of all posts, '
+                        'noting discrepancies, without modifying posts')
+    parser.add_argument('--reset-filenames', action='store_true',
+                        help='reset the content and thumbnail filenames'
+                        'caused by a lost/changed secret key')
     command = parser_top.parse_args()
 
     try:
@@ -67,6 +104,8 @@ def main() -> None:
             reset_password(command.change_password)
         elif command.check_all_audio:
             check_audio()
+        elif command.reset_filenames:
+            reset_filenames()
     except errors.BaseError as e:
         print(e, file=stderr)