198 lines
7.8 KiB
Python
198 lines
7.8 KiB
Python
import os
|
|
import shutil
|
|
import sys
|
|
import json
|
|
|
|
from katrain.core.ai import ai_rank_estimation
|
|
from katrain.core.constants import (
|
|
PLAYER_HUMAN,
|
|
PLAYER_AI,
|
|
PLAYING_NORMAL,
|
|
PLAYING_TEACHING,
|
|
OUTPUT_INFO,
|
|
OUTPUT_ERROR,
|
|
OUTPUT_DEBUG,
|
|
AI_DEFAULT,
|
|
CONFIG_MIN_VERSION,
|
|
DATA_FOLDER,
|
|
)
|
|
from katrain.core.utils import find_package_resource
|
|
|
|
|
|
class Player:
|
|
def __init__(self, player="B", player_type=PLAYER_HUMAN, player_subtype=PLAYING_NORMAL, periods_used=0):
|
|
self.player = player
|
|
self.sgf_rank = None
|
|
self.calculated_rank = None
|
|
self.name = ""
|
|
self.update(player_type, player_subtype)
|
|
self.periods_used = periods_used
|
|
|
|
def update(self, player_type=PLAYER_HUMAN, player_subtype=PLAYING_NORMAL):
|
|
self.player_type = player_type
|
|
self.player_subtype = player_subtype
|
|
|
|
@property
|
|
def ai(self):
|
|
return self.player_type == PLAYER_AI
|
|
|
|
@property
|
|
def human(self):
|
|
return self.player_type == PLAYER_HUMAN
|
|
|
|
@property
|
|
def being_taught(self):
|
|
return self.player_type == PLAYER_HUMAN and self.player_subtype == PLAYING_TEACHING
|
|
|
|
@property
|
|
def strategy(self):
|
|
return self.player_subtype if self.ai else AI_DEFAULT
|
|
|
|
def __str__(self):
|
|
return f"{self.player_type} ({self.player_subtype})"
|
|
|
|
|
|
def parse_version(s):
|
|
parts = [int(p) for p in s.split(".")]
|
|
while len(parts) < 3:
|
|
parts.append(0)
|
|
return parts
|
|
|
|
|
|
class KaTrainBase:
|
|
USER_CONFIG_FILE = os.path.expanduser(os.path.join(DATA_FOLDER, "config.json"))
|
|
PACKAGE_CONFIG_FILE = "katrain/config.json"
|
|
|
|
"""Settings, logging, and players functionality, so other classes like bots who need a katrain instance can be used without a GUI"""
|
|
|
|
def __init__(self, force_package_config=False, debug_level=None, **kwargs):
|
|
self.debug_level = debug_level or 0
|
|
self.game = None
|
|
|
|
self.logger = lambda message, level=OUTPUT_INFO: self.log(message, level)
|
|
self.config_file = self._load_config(force_package_config=force_package_config)
|
|
self.debug_level = self.config("general/debug_level", OUTPUT_INFO) if debug_level is None else debug_level
|
|
|
|
self.players_info = {"B": Player("B"), "W": Player("W")}
|
|
self.reset_players()
|
|
|
|
def log(self, message, level=OUTPUT_INFO):
|
|
if level == OUTPUT_ERROR:
|
|
print(f"ERROR: {message}")
|
|
elif self.debug_level >= level:
|
|
print(message)
|
|
|
|
def _load_config(self, force_package_config):
|
|
if len(sys.argv) > 1 and sys.argv[1].endswith("config.json"):
|
|
config_file = os.path.abspath(sys.argv[1])
|
|
self.log(f"Using command line config file {config_file}", OUTPUT_INFO)
|
|
else:
|
|
user_config_file = find_package_resource(self.USER_CONFIG_FILE)
|
|
package_config_file = find_package_resource(self.PACKAGE_CONFIG_FILE)
|
|
if force_package_config:
|
|
config_file = package_config_file
|
|
else:
|
|
try:
|
|
if not os.path.exists(user_config_file):
|
|
self.log("User config does not exist, creating it", OUTPUT_DEBUG)
|
|
parent_dir = os.path.split(user_config_file)[0]
|
|
self.log(f"Creating parent directory if needed: {parent_dir}", OUTPUT_DEBUG)
|
|
os.makedirs(parent_dir, exist_ok=True)
|
|
|
|
self.log(f"Copying package config {package_config_file} to user config {user_config_file}", OUTPUT_DEBUG)
|
|
shutil.copyfile(package_config_file, user_config_file)
|
|
config_file = user_config_file
|
|
self.log(f"Copied package config to local file {config_file}", OUTPUT_INFO)
|
|
else: # user file exists
|
|
try:
|
|
with open(user_config_file, encoding="utf-8") as config_handle:
|
|
version_str = json.load(config_handle)["general"]["version"]
|
|
version = parse_version(version_str)
|
|
self.log(f"Parsed version: {version}", OUTPUT_DEBUG)
|
|
except Exception as e: # noqa E722 broken file etc
|
|
self.log(f"Failed to read version from user config: {e}", OUTPUT_DEBUG)
|
|
version_str = "0.0.0"
|
|
version = [0, 0, 0]
|
|
min_version = parse_version(CONFIG_MIN_VERSION)
|
|
if version < min_version:
|
|
backup = f"{user_config_file}.{version_str}.backup"
|
|
shutil.copyfile(user_config_file, backup)
|
|
shutil.copyfile(package_config_file, user_config_file)
|
|
self.log(
|
|
f"Copied package config file to {user_config_file} as user file is outdated or broken ({version}<{min_version}). Old version stored as {backup}",
|
|
OUTPUT_INFO,
|
|
)
|
|
config_file = user_config_file
|
|
self.log(f"Using user config file {config_file}", OUTPUT_INFO)
|
|
except Exception as e:
|
|
config_file = package_config_file
|
|
self.log(
|
|
f"Using package config file {config_file} (exception {e} occurred when finding or creating user config)",
|
|
OUTPUT_INFO,
|
|
)
|
|
try:
|
|
self._config_store = JsonConfigStore(config_file)
|
|
except Exception as e:
|
|
self.log(f"Failed to load config {config_file}: {e}", OUTPUT_ERROR)
|
|
sys.exit(1)
|
|
self._config = dict(self._config_store)
|
|
return config_file
|
|
|
|
def save_config(self, key=None):
|
|
if key is None:
|
|
for k, v in self._config.items():
|
|
self._config_store.put(k, **v)
|
|
else:
|
|
self._config_store.put(key, **self._config[key])
|
|
|
|
def config(self, setting, default=None):
|
|
try:
|
|
if "/" in setting:
|
|
cat, key = setting.split("/")
|
|
return self._config.get(cat, {}).get(key, default)
|
|
else:
|
|
return self._config.get(setting, default)
|
|
except KeyError:
|
|
self.log(f"Missing configuration option {setting}", OUTPUT_ERROR)
|
|
|
|
def update_player(self, bw, **kwargs):
|
|
self.players_info[bw].update(**kwargs)
|
|
self.update_calculated_ranks()
|
|
|
|
def update_calculated_ranks(self):
|
|
for bw, player_info in self.players_info.items():
|
|
if player_info.player_type == PLAYER_AI:
|
|
settings = self.config(f"ai/{player_info.strategy}")
|
|
player_info.calculated_rank = ai_rank_estimation(player_info.player_subtype, settings)
|
|
else:
|
|
player_info.calculated_rank = None
|
|
|
|
def reset_players(self):
|
|
self.update_player("B")
|
|
self.update_player("W")
|
|
for v in self.players_info.values():
|
|
v.periods_used = 0
|
|
|
|
@property
|
|
def last_player_info(self) -> Player:
|
|
return self.players_info[self.game.current_node.player]
|
|
|
|
@property
|
|
def next_player_info(self) -> Player:
|
|
return self.players_info[self.game.current_node.next_player]
|
|
|
|
|
|
class JsonConfigStore(dict):
|
|
"""Small replacement for Kivy's JsonStore used by the reusable core."""
|
|
|
|
def __init__(self, filename):
|
|
self.filename = filename
|
|
with open(filename, encoding="utf-8") as config_handle:
|
|
super().__init__(json.load(config_handle))
|
|
|
|
def put(self, key, **value):
|
|
self[key] = value
|
|
with open(self.filename, "w", encoding="utf-8") as config_handle:
|
|
json.dump(self, config_handle, indent=4)
|
|
config_handle.write("\n")
|