showdown/config.py at master · CodeSammich/showdown · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
import sys
import logging

battle_bot_module = None
websocket_uri = None
username = None
password = None
bot_mode = None
team_name = None
pokemon_mode = None
run_count = None
user_to_challenge = None
gambit_exe_path = ""
greeting_message = 'hf'
battle_ending_message = 'gg'

use_relative_weights = False
damage_calc_type = 'average'
search_depth = 2

save_replay = False

class Config():
    def __init__(self):
        self.battle_bot_module = None
        self.websocket_uri = None
        self.username = None
        self.password = None
        self.bot_mode = None
        self.team_name = None
        self.pokemon_mode = None
        self.run_count = None
        self.user_to_challenge = None
        self.gambit_exe_path = ""
        self.greeting_message = 'hf'
        self.battle_ending_message = 'gg'

        self.use_relative_weights = False
        self.damage_calc_type = 'average'
        self.search_depth = 2

        self.save_replay = False


class CustomFormatter(logging.Formatter):
    def format(self, record):
        record.module = "[{}]".format(record.module)
        record.levelname = "[{}]".format(record.levelname)
        return "{}{} {}".format(record.module.ljust(25), record.levelname.ljust(10), record.msg)


def init_logging(level):
    websockets_logger = logging.getLogger("websockets")
    websockets_logger.setLevel(logging.INFO)
    requests_logger = logging.getLogger("urllib3")
    requests_logger.setLevel(logging.INFO)

    logger = logging.getLogger()
    logger.setLevel(level)
    default_formatter = CustomFormatter()
    default_handler = logging.StreamHandler(sys.stdout)
    default_handler.setFormatter(default_formatter)
    logger.addHandler(default_handler)