Implement multiple group archiving

archive.py

@@ -1,24 +1,26 @@
-import asyncio
-from turtle import down
-import vk_api
 import argparse
-from mastodon import Mastodon
+import sys
+
+import dataset
 import requests
 import toml
-import pathlib
-import sys
-import dataset
+import vk_api
+from mastodon import Mastodon
 
 parser = argparse.ArgumentParser()
 parser.add_argument("-c", "--config", help="Config path")
-parser.add_argument("-g", "--group", help="VK group domain")
+parser.add_argument("-g", "--group", help="VK group to archive")
 args = parser.parse_args()
 
 config = toml.load(args.config)
 
+if config["mastodon"].get(args.group) is None:
+    print("invalid group")
+    sys.exit(1)
+
 mastodon = Mastodon(
-    access_token = config["mastodon"]["access_token"],
-    api_base_url = config["mastodon"]["instance"]
+    access_token=config["mastodon"][args.group]["access_token"],
+    api_base_url=config["mastodon"][args.group]["instance"]
 )
 
 vk_session = vk_api.VkApi(token=config["vk"]["access_token"])

config.example.toml

@@ -1,6 +1,6 @@
 [vk]
 access_token = "***"
 
-[mastodon]
-instance = "https://mastodon.ml"
+[mastodon.group_name]
+instance = "https://mastodon.social"
 access_token = "***"