From a992859c23b9fe33d2623f5ec53c58c48699ed05 Mon Sep 17 00:00:00 2001 From: emkael Date: Wed, 4 Jan 2017 00:34:59 +0100 Subject: Moving config read to entry script --- ausbutler/interface.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) (limited to 'ausbutler/interface.py') diff --git a/ausbutler/interface.py b/ausbutler/interface.py index 99b0e9d..cd8df45 100644 --- a/ausbutler/interface.py +++ b/ausbutler/interface.py @@ -6,8 +6,9 @@ import re class Interface: - def __init__(self): + def __init__(self, config): self.session = Session() + self.config = config def init_db(self, force=False): from .model import Base @@ -28,7 +29,10 @@ class Interface: aus_b.match = int(column_match.group(1), base=10) aus_b.segment = int(column_match.group(2)) aus_b.score = float(value) - aus_b.cut_score = cutoff(aus_b.score) + aus_b.cut_score = cutoff( + aus_b.score, + self.config['cutoff_point'], + self.config['cutoff_rate']) aus_b.board_count = aus_b.table.butler_count[ get_room(aus_b, butler.id)] self.session.add(aus_b) @@ -52,5 +56,6 @@ class Interface: def normalize_scores(self): for butler in self.session.query(AusButler).all(): - butler.corrected_score = normalize(butler) + butler.corrected_score = normalize( + butler, self.config['opponent_factor']) self.session.commit() -- cgit v1.2.3