def configure_notifier(notifier_alias, settings_dict): """Configures notifier using given settings. Saves successful configuration. :param notifier_alias: notifier alias :param settings_dict: settings dictionary to configure notifier with :return: """ LOGGER.info('Configuring `%s` notifier ...', notifier_alias) notification_class = NotifierClassesRegistry.get(notifier_alias) if notification_class is not None: notifier = notification_class.spawn_with_settings(settings_dict) configured = notifier.test_configuration() if configured: notifier.save_settings() LOGGER.info('Notifier `%s` is configured', notifier_alias) else: LOGGER.error('Notifier `%s` configuration failed. Check your settings', notifier_alias) else: LOGGER.error('Notifier `%s` is unknown', notifier_alias)
except Exception as e: LOGGER.error(e) return if self.user and self.password: try: connection.login(self.user, self.password) except SMTPAuthenticationError as e: LOGGER.error(e) return return connection def send_message(self, msg): self.connection.sendmail(self.sender, [self.email], msg) def test_configuration(self): return bool(self.connection) def make_message(self, torrent_data): text = '''The following torrents were updated:\n%s\n\nBest regards,\ntorrt.''' \ % '\n'.join(map(lambda t: t['name'], torrent_data.values())) msg = MIMEText(text) msg['Subject'] = 'New torrents were added to download queue.' msg['From'] = self.sender msg['To'] = self.email LOGGER.info('Notification message was sent to user %s' % self.email) return msg.as_string() NotifierClassesRegistry.add(EmailNotifier)
def __init__(self, token, chat_id): """ :param token: str - Telegram's bot token :param chat_id: str - Telegram's chat ID """ self.token = token self.chat_id = chat_id def make_message(self, torrent_data): return '''The following torrents were updated:\n%s''' \ % '\n'.join(map(lambda t: t['name'], torrent_data.values())) def test_configuration(self): url = '%s%s/getMe' % (self.url, self.token) r = requests.get(url) return r.json().get('ok', False) def send_message(self, msg): url = '%s%s/sendMessage' % (self.url, self.token) r = requests.post(url, data={'chat_id': self.chat_id, 'text': msg}) if r.json()['ok']: LOGGER.info('Telegram message was sent to user %s' % self.chat_id) else: LOGGER.error('Telegram notification not send: %s' % r['description']) NotifierClassesRegistry.add(TelegramNotifier)
def process_commands(): def settings_dict_from_list(lst): settings_dict = {} for s in lst: splitted = s.split('=') settings_dict[splitted[0]] = splitted[1] return settings_dict arg_parser = argparse.ArgumentParser( 'torrt', description='Automates torrent updates for you.') arg_parser.add_argument('--version', action='version', version='%(prog)s ' + '.'.join(map(str, VERSION))) subp_main = arg_parser.add_subparsers(title='Supported commands', dest='command') subp_main.add_parser('list_rpc', help='Shows known RPCs aliases') subp_main.add_parser('list_trackers', help='Shows known trackers aliases') subp_main.add_parser('list_torrents', help='Shows torrents registered for updates') subp_main.add_parser('list_notifiers', help='Shows configured notifiers') parser_configure_tracker = subp_main.add_parser( 'configure_tracker', help='Sets torrent tracker settings (login credentials, etc.)', description= 'E.g.: configure_tracker rutracker.org username=idle password=pSW0rt') parser_configure_tracker.add_argument( 'tracker_alias', help='Tracker alias (usually domain) to apply settings to') parser_configure_tracker.add_argument( 'settings', help='Settings string, format: setting1=val1 setting2=val2. ' 'Supported settings (any of): username, password', nargs='*') parser_configure_rpc = subp_main.add_parser( 'configure_rpc', help='Sets RPCs settings (login credentials, etc.)', description='E.g.: configure_rpc transmission user=idle password=pSW0rt' ) parser_configure_rpc.add_argument('rpc_alias', help='RPC alias to apply settings to') parser_configure_rpc.add_argument( 'settings', help='Settings string, format: setting1=val1 setting2=val2. ' 'Supported settings (any of): url, host, port, user, password', nargs='*') parser_configure_notifier = subp_main.add_parser( 'configure_notifier', help='Sets Notifiers settings (smtp credentials, etc.)', description= 'E.g.: configure_notifier email [email protected] user=idle password=pSW0rt' ) parser_configure_notifier.add_argument( 'notifier_alias', help='Notifier alias to apply settings to') parser_configure_notifier.add_argument( 'settings', help='Settings string, format: setting1=val1 setting2=val2. ' 'Supported settings for email notifier (any of): email, host, port, use_tls, user, password.' 'Supported settings for telegram notifier: token, chat_id.', nargs='*') parser_configure_bot = subp_main.add_parser( 'configure_bot', help='Sets Bot settings (token, etc.)', description='E.g.: configure_bot telegram token=YourBotSuperToken') parser_configure_bot.add_argument('bot_alias', help='Bot alias to apply settings to') parser_configure_bot.add_argument( 'settings', help='Settings string, format: setting1=val1 setting2=val2. ' 'Supported settings for telegram bot: token.', nargs='*') parser_walk = subp_main.add_parser( 'walk', help='Walks through registered torrents and performs automatic updates' ) parser_walk.add_argument( '-f', help='Forces walk. Forced walks do not respect walk interval settings', dest='forced', action='store_true') parser_walk.add_argument( '--dump', help= 'Dump web pages scraped by torrt into current or a given directory', dest='dump') parser_run_bots = subp_main.add_parser('run_bots', help='Run registered bots') parser_run_bots.add_argument('aliases', help='Bots to run aliases', nargs='*') parser_set_interval = subp_main.add_parser( 'set_walk_interval', help= 'Sets an interval *in hours* between consecutive torrent updates checks' ) parser_set_interval.add_argument( 'walk_interval', help='Interval *in hours* between consecutive torrent updates checks') parser_enable_rpc = subp_main.add_parser('enable_rpc', help='Enables RPC by its alias') parser_enable_rpc.add_argument('alias', help='Alias of RPC to enable') parser_disable_rpc = subp_main.add_parser('disable_rpc', help='Disables RPC by its alias') parser_disable_rpc.add_argument('alias', help='Alias of RPC to disable') parser_add_torrent = subp_main.add_parser( 'add_torrent', help='Adds torrent from an URL both to torrt and torrent clients') parser_add_torrent.add_argument('url', help='URL to download torrent from') parser_add_torrent.add_argument( '-d', help= 'Destination path to download torrent contents into (in filesystem where torrent client daemon works)', dest='download_to', default=None) parser_add_torrent.add_argument( '--dump', help= 'Dump web pages scraped by torrt into current or a given directory', dest='dump') parser_remove_torrent = subp_main.add_parser( 'remove_torrent', help='Removes torrent by its hash both from torrt and torrent clients') parser_remove_torrent.add_argument('hash', help='Torrent identifying hash') parser_remove_torrent.add_argument( '-d', help='If set data downloaded for torrent will also be removed', dest='delete_data', action='store_true') parser_register_torrent = subp_main.add_parser( 'register_torrent', help= 'Registers torrent within torrt by its hash (for torrents already existing at torrent clients)' ) parser_register_torrent.add_argument('hash', help='Torrent identifying hash') parser_register_torrent.add_argument('-u', dest='url', default=None, help='URL to download torrent from') parser_unregister_torrent = subp_main.add_parser( 'unregister_torrent', help='Unregisters torrent from torrt by its hash') parser_unregister_torrent.add_argument('hash', help='Torrent identifying hash') parser_remove_notifier = subp_main.add_parser( 'remove_notifier', help='Remove configured notifier by its alias') parser_remove_notifier.add_argument('alias', help='Alias of notifier to remove') for parser in subp_main.choices.values(): parser.add_argument('--verbose', help='Switch to show debug messages', dest='verbose', action='store_true') args = arg_parser.parse_args() args = vars(args) configure_logging(logging.DEBUG if args.get('verbose') else logging.INFO) bootstrap() dump_into = args.get('dump') if dump_into: GlobalParam.set('dump_into', path.abspath(dump_into)) if args['command'] == 'enable_rpc': toggle_rpc(args['alias'], True) elif args['command'] == 'disable_rpc': toggle_rpc(args['alias'], False) elif args['command'] == 'list_trackers': for tracker_alias, _ in TrackerClassesRegistry.get().items(): LOGGER.info(tracker_alias) elif args['command'] == 'list_rpc': rpc_statuses = {} for rpc_alias, rpc in RPCClassesRegistry.get().items(): rpc_statuses[rpc_alias] = 'unconfigured' for rpc_alias, rpc in RPCObjectsRegistry.get().items(): rpc_statuses[rpc_alias] = 'enabled' if rpc.enabled else 'disabled' for rpc_alias, rpc_status in rpc_statuses.items(): LOGGER.info(f'{rpc_alias}\t status={rpc_status}') elif args['command'] == 'list_torrents': for torrent_hash, torrent_data in get_registered_torrents().items(): LOGGER.info(f"{torrent_hash}\t{torrent_data['name']}") elif args['command'] == 'list_notifiers': notifiers = {} for notifier_alias in NotifierClassesRegistry.get().keys(): notifiers[notifier_alias] = 'unconfigured' for notifier_alias in NotifierObjectsRegistry.get().keys(): notifiers[notifier_alias] = 'enabled' for notifier_alias, notifier_status in notifiers.items(): LOGGER.info(f"{notifier_alias}\t status={notifier_status}") elif args['command'] == 'walk': walk(forced=args['forced'], silent=True) elif args['command'] == 'set_walk_interval': set_walk_interval(args['walk_interval']) elif args['command'] == 'add_torrent': add_torrent_from_url(args['url'], args['download_to']) elif args['command'] == 'remove_torrent': remove_torrent(args['hash'], args['delete_data']) elif args['command'] == 'register_torrent': register_torrent(args['hash'], url=args['url']) elif args['command'] == 'unregister_torrent': unregister_torrent(args['hash']) elif args['command'] == 'configure_rpc': configure_rpc(args['rpc_alias'], settings_dict_from_list(args['settings'])) elif args['command'] == 'configure_tracker': configure_tracker(args['tracker_alias'], settings_dict_from_list(args['settings'])) elif args['command'] == 'configure_notifier': configure_notifier(args['notifier_alias'], settings_dict_from_list(args['settings'])) elif args['command'] == 'remove_notifier': remove_notifier(args['alias']) elif args['command'] == 'configure_bot': configure_bot(args['bot_alias'], settings_dict_from_list(args['settings'])) elif args['command'] == 'run_bots': run_bots(args['aliases'])
self.chat_id = chat_id def make_message(self, torrent_data): return '''The following torrents were updated:\n%s''' \ % '\n'.join(map(lambda t: t['name'], torrent_data.values())) def test_configuration(self): url = '%s%s/getMe' % (self.url, self.token) r = requests.get(url) return r.json().get('ok', False) def send_message(self, msg): url = '%s%s/sendMessage' % (self.url, self.token) try: response = requests.post(url, data={'chat_id': self.chat_id, 'text': msg}) except RequestException as e: LOGGER.error('Failed to send Telegram message: %s', e) else: if response.ok: json_data = response.json() if json_data['ok']: LOGGER.debug('Telegram message was sent to user %s', self.chat_id) else: LOGGER.error('Telegram notification not send: %s', json_data['description']) else: LOGGER.error('Telegram notification not send. Response code: %s (%s)', response.status_code, response.reason) NotifierClassesRegistry.add(TelegramNotifier)
def process_commands(): def settings_dict_from_list(lst): settings_dict = {} for s in lst: splitted = s.split('=') settings_dict[splitted[0]] = splitted[1] return settings_dict arg_parser = argparse.ArgumentParser('torrt', description='Automates torrent updates for you.') arg_parser.add_argument('--version', action='version', version='%(prog)s ' + '.'.join(map(str, VERSION))) arg_parser.add_argument('--verbose', help='Switch to show debug messages', dest='verbose', action='store_true') subp_main = arg_parser.add_subparsers(title='Supported commands', dest='command') subp_main.add_parser('list_rpc', help='Shows known RPCs aliases') subp_main.add_parser('list_trackers', help='Shows known trackers aliases') subp_main.add_parser('list_torrents', help='Shows torrents registered for updates') subp_main.add_parser('list_notifiers', help='Shows configured notifiers') parser_configure_tracker = subp_main.add_parser( 'configure_tracker', help='Sets torrent tracker settings (login credentials, etc.)', description='E.g.: configure_tracker rutracker.org username=idle password=pSW0rt') parser_configure_tracker.add_argument( 'tracker_alias', help='Tracker alias (usually domain) to apply settings to') parser_configure_tracker.add_argument( 'settings', help='Settings string, format: setting1=val1 setting2=val2. ' 'Supported settings (any of): username, password', nargs='*') parser_configure_rpc = subp_main.add_parser( 'configure_rpc', help='Sets RPCs settings (login credentials, etc.)', description='E.g.: configure_rpc transmission user=idle password=pSW0rt') parser_configure_rpc.add_argument( 'rpc_alias', help='RPC alias to apply settings to') parser_configure_rpc.add_argument( 'settings', help='Settings string, format: setting1=val1 setting2=val2. ' 'Supported settings (any of): url, host, port, user, password', nargs='*') parser_configure_notifier = subp_main.add_parser( 'configure_notifier', help='Sets Notifiers settings (smtp credentials, etc.)', description='E.g.: configure_notifier email [email protected] user=idle password=pSW0rt') parser_configure_notifier.add_argument( 'notifier_alias', help='Notifier alias to apply settings to') parser_configure_notifier.add_argument( 'settings', help='Settings string, format: setting1=val1 setting2=val2. ' 'Supported settings for email notifier (any of): email, host, port, use_tls, user, password.' 'Supported settings for telegram notifier: token, chat_id.', nargs='*') parser_walk = subp_main.add_parser( 'walk', help='Walks through registered torrents and performs automatic updates') parser_walk.add_argument( '-f', help='Forces walk. Forced walks do not respect walk interval settings', dest='forced', action='store_true') parser_set_interval = subp_main.add_parser( 'set_walk_interval', help='Sets an interval *in hours* between consecutive torrent updates checks') parser_set_interval.add_argument( 'walk_interval', help='Interval *in hours* between consecutive torrent updates checks') parser_enable_rpc = subp_main.add_parser('enable_rpc', help='Enables RPC by its alias') parser_enable_rpc.add_argument('alias', help='Alias of RPC to enable') parser_disable_rpc = subp_main.add_parser('disable_rpc', help='Disables RPC by its alias') parser_disable_rpc.add_argument('alias', help='Alias of RPC to disable') parser_add_torrent = subp_main.add_parser( 'add_torrent', help='Adds torrent from an URL both to torrt and torrent clients') parser_add_torrent.add_argument( 'url', help='URL to download torrent from') parser_add_torrent.add_argument( '-d', help='Destination path to download torrent contents into (in filesystem where torrent client daemon works)', dest='download_to', default=None) parser_remove_torrent = subp_main.add_parser( 'remove_torrent', help='Removes torrent by its hash both from torrt and torrent clients') parser_remove_torrent.add_argument( 'hash', help='Torrent identifying hash') parser_remove_torrent.add_argument( '-d', help='If set data downloaded for torrent will also be removed', dest='delete_data', action='store_true') parser_register_torrent = subp_main.add_parser( 'register_torrent', help='Registers torrent within torrt by its hash (for torrents already existing at torrent clients)') parser_register_torrent.add_argument( 'hash', help='Torrent identifying hash') parser_unregister_torrent = subp_main.add_parser( 'unregister_torrent', help='Unregisters torrent from torrt by its hash') parser_unregister_torrent.add_argument( 'hash', help='Torrent identifying hash') parser_remove_notifier = subp_main.add_parser( 'remove_notifier', help='Remove configured notifier by its alias') parser_remove_notifier.add_argument('alias', help='Alias of notifier to remove') args = arg_parser.parse_args() args = vars(args) loggin_level = logging.INFO if args['verbose']: loggin_level = logging.DEBUG configure_logging(loggin_level) bootstrap() if args['command'] == 'enable_rpc': toggle_rpc(args['alias'], True) elif args['command'] == 'disable_rpc': toggle_rpc(args['alias'], False) elif args['command'] == 'list_trackers': for tracker_alias, _ in TrackerClassesRegistry.get().items(): LOGGER.info(tracker_alias) elif args['command'] == 'list_rpc': rpc_statuses = {} for rpc_alias, rpc in RPCClassesRegistry.get().items(): rpc_statuses[rpc_alias] = 'unconfigured' for rpc_alias, rpc in RPCObjectsRegistry.get().items(): rpc_statuses[rpc_alias] = 'enabled' if rpc.enabled else 'disabled' for rpc_alias, rpc_status in rpc_statuses.items(): LOGGER.info('%s\t status=%s', rpc_alias, rpc_status) elif args['command'] == 'list_torrents': for torrent_hash, torrent_data in get_registered_torrents().items(): LOGGER.info('%s\t%s', torrent_hash, torrent_data['name']) elif args['command'] == 'list_notifiers': notifiers = {} for notifier_alias in NotifierClassesRegistry.get().keys(): notifiers[notifier_alias] = 'unconfigured' for notifier_alias in NotifierObjectsRegistry.get().keys(): notifiers[notifier_alias] = 'enabled' for notifier_alias, notifier_status in notifiers.items(): LOGGER.info('%s\t status=%s', notifier_alias, notifier_status) elif args['command'] == 'walk': walk(forced=args['forced'], silent=True) elif args['command'] == 'set_walk_interval': set_walk_interval(args['walk_interval']) elif args['command'] == 'add_torrent': add_torrent_from_url(args['url'], args['download_to']) elif args['command'] == 'remove_torrent': remove_torrent(args['hash'], args['delete_data']) elif args['command'] == 'register_torrent': register_torrent(args['hash']) elif args['command'] == 'unregister_torrent': unregister_torrent(args['hash']) elif args['command'] == 'configure_rpc': configure_rpc(args['rpc_alias'], settings_dict_from_list(args['settings'])) elif args['command'] == 'configure_tracker': configure_tracker(args['tracker_alias'], settings_dict_from_list(args['settings'])) elif args['command'] == 'configure_notifier': configure_notifier(args['notifier_alias'], settings_dict_from_list(args['settings'])) elif args['command'] == 'remove_notifier': remove_notifier(args['alias'])