o NGi+ @sBddlmZddlZddlmZmZddlmZGdddZdS)) annotationsN)AnyOptional)settingsc@sTeZdZddZddZddZdd d Zdd d Zdd dZddZ dddZ dS)DatabaseManagercCs d|_dSN)poolselfr C/var/www/www-root/data/www/77.83.87.30/database/database_manager.py__init__ s zDatabaseManager.__init__cs.|jdurtjtjdddIdH|_dSdS)N )dsnmin_sizemax_size)rasyncpg create_poolr DATABASE_URLr r r r connect s zDatabaseManager.connectcs |jr|jIdHdSdSr)rcloser r r r rszDatabaseManager.closequerystrc sl|IdH|j4IdH}|j|g|RIdHWdIdHdS1IdHs/wYdSr)rracquireexecuter rargsconnr r r rs .zDatabaseManager.executec j|IdH|j4IdH}|j|g|RIdHWdIdHS1IdHs.wYdSr)rrrfetchrr r r fetch_all 0zDatabaseManager.fetch_allc rr)rrrfetchrowrr r r fetch_one"r"zDatabaseManager.fetch_onecs6|dIdH|dIdH|dIdHdS)NaN CREATE TABLE IF NOT EXISTS matches ( match_id TEXT PRIMARY KEY, home_team TEXT, away_team TEXT, home_goals INTEGER, away_goals INTEGER, status TEXT, start_time TIMESTAMP, competition TEXT, url TEXT ); a3 CREATE TABLE IF NOT EXISTS predictions ( match_id TEXT PRIMARY KEY, p1 REAL, px REAL, p2 REAL, lambda_home REAL, lambda_away REAL, confidence REAL, created_at TIMESTAMP DEFAULT now() ); z CREATE TABLE IF NOT EXISTS team_stats ( team_id TEXT PRIMARY KEY, team_name TEXT, attack REAL, defense REAL, updated_at TIMESTAMP DEFAULT now() ); )rr r r r init_db'szDatabaseManager.init_dbteam_idcs|d|IdH}|S)Nz7SELECT attack, defense FROM team_stats WHERE team_id=$1)r$)r r&rowr r r get_team_statsPs  zDatabaseManager.get_team_statsN)rr)r&r) __name__ __module__ __qualname__r rrrr!r$r%r(r r r r rs   )r) __future__rrtypingrrconfigrrr r r r s