Update fix_container_permissions.py

miraheze/swift/fix_container_permissions.py

@@ -1,18 +1,24 @@
-import os
 import re
 import subprocess
-import sys
+import argparse
 
-if len(sys.argv) != 2:
-    print("Please use in form of 'python fix_containers.py foowiki'")
-    sys.exit(1)
 
-wiki = sys.argv[1]
+def fix_container_perms(wiki: str) -> None:
+    out = subprocess.run(['sudo', '-u', 'www-data', 'php', '/srv/mediawiki/1.41/maintenance/run.php', '/srv/mediawiki/1.41/extensions/CreateWiki/maintenance/setContainersAccess.php', '--wiki', wiki], capture_output=True, text=True)
+    matches = re.findall(r"Making sure 'mwstore:\/\/miraheze-swift\/([^']+)' [^\n]+\.failed\.", out.stdout)
+    for match in matches:
+        subprocess.run(['swift', 'post', '--read-acl', 'mw:media', '--write-acl', 'mw:media', f'miraheze-{wiki}-{match}'], check=True)
 
-out = subprocess.run(['sudo', '-u', 'www-data', 'php', '/srv/mediawiki/1.41/maintenance/run.php', '/srv/mediawiki/1.41/extensions/CreateWiki/maintenance/setContainersAccess.php', '--wiki', wiki], capture_output=True, text=True)
+    subprocess.run(['sudo', '-u', 'www-data', 'php', '/srv/mediawiki/1.41/maintenance/run.php', '/srv/mediawiki/1.41/extensions/CreateWiki/maintenance/setContainersAccess.php', '--wiki', wiki])
 
-matches = re.findall(r"Making sure 'mwstore:\/\/miraheze-swift\/([^']+)' [^\n]+\.failed\.", out.stdout)
-for match in matches:
-    os.system(f"swift post --read-acl 'mw:media' --write-acl 'mw:media' miraheze-{wiki}-{match}")
 
-os.system(f'sudo -u www-data php /srv/mediawiki/1.41/maintenance/run.php /srv/mediawiki/1.41/extensions/CreateWiki/maintenance/setContainersAccess.php --wiki {wiki}')
+def main() -> None:
+    parser = argparse.ArgumentParser(description='Fix container permissions for a specified wiki')
+    parser.add_argument('--wiki', required=True, help='wiki database name')
+
+    args = parser.parse_args()
+    fix_container_perms(args.wiki)
+
+
+if __name__ == '__main__':
+    main()

setup.cfg

@@ -13,3 +13,4 @@ python_requires = >=3.11
 console_scripts =
     partial-reset-wiki = miraheze.salt.mwcli.partial_reset_wiki:main
     rename-wiki = miraheze.salt.mwcli.rename_wiki:main
+    fix-container-perms = miraheze.swift.fix_container_permissions:main