def run(external_processor=None):
start_time = time.time()
log = logging.getLogger('bot')
config = load_config()
if not SerialController().serial:
SerialController().run_serial(config['serial'])
mode = config['mode']
if mode == 'enhance':
enhancer = Enhancer(config['enhancer'])
enhancer.process()
if mode == 'buff':
buff_config = Configurator(config['buffer']).from_yaml()
processor = ProcessInitializer(Buffer(buff_config), config['serial'])
processor.handle()
if mode == 'make':
external_processor.process()
if mode == 'combination':
combinate = Combinator(config['enhancer'])
combinate.process()
if mode == 'taming':
Taming().run()
if mode == 'farming':
Farming().run()
if mode == 'test':
print('nothings')
exec_time = (time.time() - start_time)
final_message = "Finished work, time: {0} (sec), {1} (min)".format(
exec_time, exec_time / 60)
log.info(final_message)
return exec_time
def set_mode(mode, config):
configurator = Configurator(config)
config = configurator.from_yaml()
if mode not in AVAILABLE_MODES:
return 'Wrong mode name'
config['mode'] = mode
configurator.dump_yaml(config)
return 'I change mode to ' + mode
def initialize_configs(self, config):
import jobs.helpers.configs as markers
import sys
config = Configurator(config).from_yaml()
for c in config['templates']:
setattr(self, c['name'], getattr(sys.modules['jobs.helpers.configs'], c['name']))
for field, value in c.items():
if field is not 'name':
setattr(getattr(self, c['name']), field, value)
def __init__(self):
self.config = Configurator('config.yml').from_yaml()['window']
self.width, self.height = self.config['width'], self.config['height']
self.locate_window()
if self.windowHead is None:
log.error('Window has not initiated!')
return
self.x = self.windowHead[0]
self.y = self.windowHead[1] + self.windowHead[3]
self.screen = ui.screenshot(region=(self.x, self.y, self.width,
self.height))
self.rect = (self.x, self.y, self.width, self.height)
def quests(self, call, state):
self.bot.answer_callback_query(call.id, 'Get circus quests')
buff_cfg = Configurator(self.config['buffer']).from_yaml()
buff_cfg['spawn'] = True
buff_cfg['logout'] = True
buff = Buffer(buff_cfg)
for i in range(8):
CircusHandler().get_quest()
buff.process_flow()
return None, None
def search(docopt_arguments, source_loader):
"""
The function that facilitates the search for a specific compound.
:param docopt_arguments: A dictionary generated by docopt containing all CLI arguments.
:param source_loader: An initiated SourceLoader object pointed at the directory with the sources.
"""
conf = Configurator()
conf.set_logging(docopt_arguments["--log"], docopt_arguments["-v"])
conf.set_output(docopt_arguments["--output"], docopt_arguments["--format"],
docopt_arguments["<compound>"])
setup_crawler(docopt_arguments["<compound>"], conf.scrapy_settings,
source_loader, docopt_arguments["--attributes"].split(','))
if conf.scrapy_settings.getbool("LOG_ENABLED"):
log.start(conf.scrapy_settings.get("LOG_FILE"),
conf.scrapy_settings.get("LOG_LEVEL"),
conf.scrapy_settings.get("LOG_STDOUT"))
reactor.run()
def get_quests(config):
from utils.serial_controller import SerialController
from jobs.helpers.circus_handler import CircusHandler
from processes.wait import Wait
from jobs.buffer import Buffer
from utils.config import Config
Config().initialize_configs(config['navigator'])
if not SerialController().serial:
SerialController().run_serial(config['serial'])
buff_cfg = Configurator(config['buffer']).from_yaml()
buff = Buffer(buff_cfg)
for i in range(8):
CircusHandler().get_quest()
Wait(2).delay()
buff.process_flow()
def set_combination_mode(mode, config):
configurator = Configurator(config['enhancer'])
config = configurator.from_yaml()
config['combination']['mode'] = mode
configurator.dump_yaml(config)
def set_logout(config):
configurator = Configurator(config['buffer'])
config = configurator.from_yaml()
config['spawn'] = True
config['logout'] = True
configurator.dump_yaml(config)
def load_config():
return Configurator('config.yml').from_yaml()
from torchvision import transforms
from configparser import ConfigParser
from PIL import Image
import torch
import json
import numpy as np
import math
import os, sys
if __name__ == '__main__':
sys.path.append(os.getcwd())
sys.path[0] = os.path.dirname(os.path.realpath(__file__))
from utils.configurator import Configurator
configurator = Configurator()
DATASET_ROOT_PATH = configurator.get('Dataset', 'dataset_root_path')
CORRECT_DATASET_ROOT_PATH = configurator.get('Dataset',
'correct_dataset_root_path')
INCORRECT_DATASET_ROOT_PATH = configurator.get('Dataset',
'incorrect_dataset_root_path')
CORRECT_SAMPLES_NUMBER = 67048 # but 67049 officially declared
INCORRECT_SAMPLES_NUMBER = 66734
VGG_IMG_SIZE = 224
class LazyDataset(Dataset):
def __init__(self, mode='train'):
def load_config(self):
self.config = Configurator(self.config['enhancer']).from_yaml()
def load_config(self, config):
if not hasattr(self, 'config'):
print('load config')
self.config = Configurator(config).from_yaml()
from processes.process import Process
from utils.configurator import Configurator
from processes.click import Click
from processes.wait import Wait
from processes.key import Key
import pyautogui as ui
config = Configurator('config.yml').from_yaml()
serial = Process(config['serial']).run_serial()
for i in range(0, 6):
count = Click(536, 450, process='dclick')
count.make_click(serial)
Wait(0.5).delay()
on = Click(464, 451, process='dclick')
on.make_click(serial)
Wait(0.5).delay()
one = Key('1')
one.press(serial)
Wait(0.5).delay()
enter = Key('E')
enter.press(serial)
Wait(0.5).delay()
make = Click(352, 492)
make.make_click(serial)
Wait(1).delay()
def setUp(self):
self.conf = Configurator()
def set_cube(params, config):
configurator = Configurator(config['enhancer'])
config = configurator.from_yaml()
config['enhancement']['cube'] = [params[0], params[1]]
configurator.dump_yaml(config)
def set_cycles(params, config):
configurator = Configurator(config['enhancer'])
config = configurator.from_yaml()
config['enhancement']['cycles'] = params
configurator.dump_yaml(config)
def __init__(self, config):
self.config = Configurator(config)
self.config = self.config.from_yaml()
self.combination_cfg = self.config['combination']
self.config = self.config['enhance']
def set_enhance_mode(mode, config):
configurator = Configurator(config['enhancer'])
config = configurator.from_yaml()
config['mode'] = mode
configurator.dump_yaml(config)
def load_config(self):
self.configfile = Configurator(self.config['farming'])
self.config = self.configfile.from_yaml()
def set_buff(params, config):
configurator = Configurator(config['buffer'])
config = configurator.from_yaml()
config['refresh'] = True
config['spawn'] = False
configurator.dump_yaml(config)
def load_config(self):
self.configfile = Configurator(self.config['enhancer'])
self.config = self.configfile.from_yaml()
def serial_run():
config = Configurator('config.yml').from_yaml()
SerialController().run_serial(config['serial'])
