def main():
args = parse_args()
log = get_logger()
if args.status_bar:
configure_logger(filename=join(conf.DIRECTORY, 'scan.log'))
log.info('-' * 37)
log.info('Starting up!')
else:
configure_logger(filename=None)
log.setLevel(args.log_level)
AccountManager.register('captcha_queue', callable=get_captchas)
AccountManager.register('extra_queue', callable=get_extras)
AccountManager.register('lv30_captcha_queue', callable=get_lv30_captchas)
AccountManager.register('lv30_account_queue', callable=get_lv30_accounts)
if conf.MAP_WORKERS:
AccountManager.register('worker_dict',
callable=get_workers,
proxytype=DictProxy)
AccountManager.register('lv30_worker_dict',
callable=get_lv30_workers,
proxytype=DictProxy)
address = get_address()
manager = AccountManager(address=address, authkey=conf.AUTHKEY)
try:
manager.start(mgr_init)
except (OSError, EOFError) as e:
if platform == 'win32' or not isinstance(address, str):
raise OSError(
'Another instance is running with the same manager address. Stop that process or change your MANAGER_ADDRESS.'
) from e
else:
raise OSError(
'Another instance is running with the same socket. Stop that process or: rm {}'
.format(address)) from e
LOOP.set_exception_handler(exception_handler)
overseer = Overseer(manager)
overseer.start(args.status_bar)
launcher = LOOP.create_task(overseer.launch(args.bootstrap, args.pickle))
if conf.GO_HASH:
hashkey = conf.GO_HASH_KEY
else:
hashkey = conf.HASH_KEY
activate_hash_server(hashkey,
go_hash=conf.GO_HASH,
hash_endpoint=conf.HASH_ENDPOINT,
gohash_endpoint=conf.GOHASH_ENDPOINT)
if platform != 'win32':
LOOP.add_signal_handler(SIGINT, launcher.cancel)
LOOP.add_signal_handler(SIGTERM, launcher.cancel)
try:
LOOP.run_until_complete(launcher)
except (KeyboardInterrupt, SystemExit):
launcher.cancel()
finally:
cleanup(overseer, manager)
def main():
args = parse_args()
log = get_logger()
if args.status_bar:
configure_logger(filename=join(conf.DIRECTORY, 'scan.log'))
log.info('-' * 37)
log.info('Starting up!')
else:
configure_logger(filename=None)
log.setLevel(args.log_level)
AccountManager.register('captcha_queue', callable=get_captchas)
AccountManager.register('extra_queue', callable=get_extras)
if conf.MAP_WORKERS:
AccountManager.register('worker_dict', callable=get_workers,
proxytype=DictProxy)
address = get_address()
manager = AccountManager(address=address, authkey=conf.AUTHKEY)
try:
manager.start(mgr_init)
except (OSError, EOFError) as e:
if platform == 'win32' or not isinstance(address, str):
raise OSError('Another instance is running with the same manager address. Stop that process or change your MANAGER_ADDRESS.') from e
else:
raise OSError('Another instance is running with the same socket. Stop that process or: rm {}'.format(address)) from e
LOOP.set_exception_handler(exception_handler)
overseer = Overseer(manager)
overseer.start(args.status_bar)
launcher = LOOP.create_task(overseer.launch(args.bootstrap, args.pickle))
activate_hash_server(conf.HASH_KEY)
if platform != 'win32':
LOOP.add_signal_handler(SIGINT, launcher.cancel)
LOOP.add_signal_handler(SIGTERM, launcher.cancel)
try:
LOOP.run_until_complete(launcher)
except (KeyboardInterrupt, SystemExit):
launcher.cancel()
finally:
cleanup(overseer, manager)
def main():
args = parse_args()
logger = getLogger()
if args.status_bar:
configure_logger(filename=join(config.DIRECTORY, 'scan.log'))
logger.info('-' * 30)
logger.info('Starting up!')
else:
configure_logger(filename=None)
logger.setLevel(args.log_level)
AccountManager.register('captcha_queue', callable=get_captchas)
AccountManager.register('extra_queue', callable=get_extras)
if config.MAP_WORKERS:
AccountManager.register('worker_dict', callable=get_workers,
proxytype=DictProxy)
address = get_address()
manager = AccountManager(address=address, authkey=config.AUTHKEY)
try:
manager.start(mgr_init)
except (OSError, EOFError) as e:
if platform == 'win32' or not isinstance(address, str):
raise OSError('Another instance is running with the same manager address. Stop that process or change your MANAGER_ADDRESS.') from e
else:
raise OSError('Another instance is running with the same socket. Stop that process or: rm {}'.format(address)) from e
loop = asyncio.get_event_loop()
loop.set_exception_handler(exception_handler)
overseer = Overseer(status_bar=args.status_bar, manager=manager)
overseer.start()
overseer_thread = Thread(target=overseer.check, name='overseer', daemon=True)
overseer_thread.start()
launcher_thread = Thread(target=overseer.launch, name='launcher', daemon=True, args=(args.bootstrap, args.pickle))
launcher_thread.start()
try:
loop.run_forever()
except KeyboardInterrupt:
print('Exiting, please wait until all tasks finish')
overseer.kill()
dump_pickle('accounts', Worker.accounts)
if config.CACHE_CELLS:
dump_pickle('cells', Worker.cell_ids)
pending = asyncio.Task.all_tasks(loop=loop)
try:
loop.run_until_complete(asyncio.gather(*pending, return_exceptions=True))
except Exception:
logger.exception('A wild exception appeared during exit!')
Worker.db_processor.stop()
try:
Worker.spawns.update()
except (DBAPIError):
pass
Worker.spawns.session.close()
manager.shutdown()
Session.close()
try:
loop.close()
except RuntimeError:
pass