diff --git a/tg/__main__.py b/tg/__main__.py index 20db6b6..af58428 100644 --- a/tg/__main__.py +++ b/tg/__main__.py @@ -1,3 +1,4 @@ +import argparse import asyncio import json @@ -12,7 +13,12 @@ from . import tasks async def main() -> None: - with open('tg.json', 'r') as file: + parser = argparse.ArgumentParser(description='Telegram chat bot') + parser.add_argument('-c', '--config', type=str, required=True, + help='Path to the JSON configuration file') + args = parser.parse_args() + + with open(args.config, 'r') as file: config = json.load(file) print('Конфигурация загружена.') diff --git a/vk/__main__.py b/vk/__main__.py index 98411a9..3073d4d 100644 --- a/vk/__main__.py +++ b/vk/__main__.py @@ -1,3 +1,4 @@ +import argparse import json from vkbottle.bot import Bot, run_multibot @@ -12,7 +13,12 @@ from .utils import MyAPI if __name__ == '__main__': - with open('vk.json', 'r') as file: + parser = argparse.ArgumentParser(description='Telegram chat bot') + parser.add_argument('-c', '--config', type=str, required=True, + help='Path to the JSON configuration file') + args = parser.parse_args() + + with open(args.config, 'r') as file: config = json.load(file) print('Конфигурация загружена.')