Changeset View
Changeset View
Standalone View
Standalone View
trunk/source/tools/lobbybots/xpartamupp/lobby_ranking.py
| #!/usr/bin/env python3 | #!/usr/bin/env python3 | ||||
| # -*- coding: utf-8 -*- | |||||
| """Copyright (C) 2013 Wildfire Games. | |||||
| * This file is part of 0 A.D. | |||||
| * | |||||
| * 0 A.D. is free software: you can redistribute it and/or modify | |||||
| * it under the terms of the GNU General Public License as published by | |||||
| * the Free Software Foundation, either version 2 of the License, or | |||||
| * (at your option) any later version. | |||||
| * | |||||
| * 0 A.D. is distributed in the hope that it will be useful, | |||||
| * but WITHOUT ANY WARRANTY; without even the implied warranty of | |||||
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |||||
| * GNU General Public License for more details. | |||||
| * | |||||
| * You should have received a copy of the GNU General Public License | |||||
| * along with 0 A.D. If not, see <http://www.gnu.org/licenses/>. | |||||
| """ | |||||
| import sqlalchemy | # Copyright (C) 2021 Wildfire Games. | ||||
| from sqlalchemy import Column, ForeignKey, Integer, String, Boolean | # This file is part of 0 A.D. | ||||
| from sqlalchemy.orm import relationship, sessionmaker | # | ||||
| # 0 A.D. is free software: you can redistribute it and/or modify | |||||
| # it under the terms of the GNU General Public License as published by | |||||
| # the Free Software Foundation, either version 2 of the License, or | |||||
| # (at your option) any later version. | |||||
| # | |||||
| # 0 A.D. is distributed in the hope that it will be useful, | |||||
| # but WITHOUT ANY WARRANTY; without even the implied warranty of | |||||
| # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |||||
| # GNU General Public License for more details. | |||||
| # | |||||
| # You should have received a copy of the GNU General Public License | |||||
| # along with 0 A.D. If not, see <http://www.gnu.org/licenses/>. | |||||
| """Database schema used by the XMPP bots to store game information.""" | |||||
| import argparse | |||||
| import sys | |||||
| from sqlalchemy import Boolean, Column, ForeignKey, Integer, String, create_engine | |||||
| from sqlalchemy.orm import relationship | |||||
| from sqlalchemy.ext.declarative import declarative_base | from sqlalchemy.ext.declarative import declarative_base | ||||
| engine = sqlalchemy.create_engine('sqlite:///lobby_rankings.sqlite3') | |||||
| Session = sessionmaker(bind=engine) | |||||
| session = Session() | |||||
| Base = declarative_base() | Base = declarative_base() | ||||
| class Player(Base): | class Player(Base): | ||||
| """Model representing players.""" | |||||
| __tablename__ = 'players' | __tablename__ = 'players' | ||||
| id = Column(Integer, primary_key=True) | id = Column(Integer, primary_key=True) | ||||
| jid = Column(String(255)) | jid = Column(String(255)) | ||||
| rating = Column(Integer) | rating = Column(Integer) | ||||
| highest_rating = Column(Integer) | highest_rating = Column(Integer) | ||||
| games = relationship('Game', secondary='players_info') | games = relationship('Game', secondary='players_info') | ||||
| # These two relations really only exist to satisfy the linkage | # These two relations really only exist to satisfy the linkage | ||||
| # between PlayerInfo and Player and Game and player. | # between PlayerInfo and Player and Game and player. | ||||
| games_info = relationship('PlayerInfo', backref='player') | games_info = relationship('PlayerInfo', backref='player') | ||||
| games_won = relationship('Game', backref='winner') | games_won = relationship('Game', backref='winner') | ||||
| class PlayerInfo(Base): | class PlayerInfo(Base): | ||||
| """Model representing game results.""" | |||||
| __tablename__ = 'players_info' | __tablename__ = 'players_info' | ||||
| id = Column(Integer, primary_key=True) | id = Column(Integer, primary_key=True) | ||||
| player_id = Column(Integer, ForeignKey('players.id')) | player_id = Column(Integer, ForeignKey('players.id')) | ||||
| game_id = Column(Integer, ForeignKey('games.id')) | game_id = Column(Integer, ForeignKey('games.id')) | ||||
| civs = Column(String(20)) | civs = Column(String(20)) | ||||
| teams = Column(Integer) | teams = Column(Integer) | ||||
| economyScore = Column(Integer) | economyScore = Column(Integer) | ||||
| militaryScore = Column(Integer) | militaryScore = Column(Integer) | ||||
| totalScore = Column(Integer) | totalScore = Column(Integer) | ||||
| foodGathered = Column(Integer) | foodGathered = Column(Integer) | ||||
| foodUsed = Column(Integer) | foodUsed = Column(Integer) | ||||
| woodGathered = Column(Integer) | woodGathered = Column(Integer) | ||||
| woodUsed = Column(Integer) | woodUsed = Column(Integer) | ||||
| stoneGathered = Column(Integer) | stoneGathered = Column(Integer) | ||||
| stoneUsed = Column(Integer) | stoneUsed = Column(Integer) | ||||
| metalGathered = Column(Integer) | metalGathered = Column(Integer) | ||||
| metalUsed = Column(Integer) | metalUsed = Column(Integer) | ||||
| vegetarianFoodGathered = Column(Integer) | vegetarianFoodGathered = Column(Integer) | ||||
| treasuresCollected = Column(Integer) | treasuresCollected = Column(Integer) | ||||
| lootCollected = Column(Integer) | lootCollected = Column(Integer) | ||||
| tributesSent = Column(Integer) | tributesSent = Column(Integer) | ||||
| tributesReceived = Column(Integer) | tributesReceived = Column(Integer) | ||||
| totalUnitsTrained = Column(Integer) | totalUnitsTrained = Column(Integer) | ||||
| totalUnitsLost = Column(Integer) | totalUnitsLost = Column(Integer) | ||||
| enemytotalUnitsKilled = Column(Integer) | enemytotalUnitsKilled = Column(Integer) | ||||
| infantryUnitsTrained = Column(Integer) | infantryUnitsTrained = Column(Integer) | ||||
| infantryUnitsLost = Column(Integer) | infantryUnitsLost = Column(Integer) | ||||
| enemyInfantryUnitsKilled = Column(Integer) | enemyInfantryUnitsKilled = Column(Integer) | ||||
| workerUnitsTrained = Column(Integer) | workerUnitsTrained = Column(Integer) | ||||
| workerUnitsLost = Column(Integer) | workerUnitsLost = Column(Integer) | ||||
| enemyWorkerUnitsKilled = Column(Integer) | enemyWorkerUnitsKilled = Column(Integer) | ||||
| femaleCitizenUnitsTrained = Column(Integer) | femaleCitizenUnitsTrained = Column(Integer) | ||||
| femaleCitizenUnitsLost = Column(Integer) | femaleCitizenUnitsLost = Column(Integer) | ||||
| enemyFemaleCitizenUnitsKilled = Column(Integer) | enemyFemaleCitizenUnitsKilled = Column(Integer) | ||||
| cavalryUnitsTrained = Column(Integer) | cavalryUnitsTrained = Column(Integer) | ||||
| cavalryUnitsLost = Column(Integer) | cavalryUnitsLost = Column(Integer) | ||||
| enemyCavalryUnitsKilled = Column(Integer) | enemyCavalryUnitsKilled = Column(Integer) | ||||
| championUnitsTrained = Column(Integer) | championUnitsTrained = Column(Integer) | ||||
| championUnitsLost = Column(Integer) | championUnitsLost = Column(Integer) | ||||
| enemyChampionUnitsKilled = Column(Integer) | enemyChampionUnitsKilled = Column(Integer) | ||||
| heroUnitsTrained = Column(Integer) | heroUnitsTrained = Column(Integer) | ||||
| heroUnitsLost = Column(Integer) | heroUnitsLost = Column(Integer) | ||||
| enemyHeroUnitsKilled = Column(Integer) | enemyHeroUnitsKilled = Column(Integer) | ||||
| shipUnitsTrained = Column(Integer) | shipUnitsTrained = Column(Integer) | ||||
| shipUnitsLost = Column(Integer) | shipUnitsLost = Column(Integer) | ||||
| enemyShipUnitsKilled = Column(Integer) | enemyShipUnitsKilled = Column(Integer) | ||||
| traderUnitsTrained = Column(Integer) | traderUnitsTrained = Column(Integer) | ||||
| traderUnitsLost = Column(Integer) | traderUnitsLost = Column(Integer) | ||||
| enemyTraderUnitsKilled = Column(Integer) | enemyTraderUnitsKilled = Column(Integer) | ||||
| totalBuildingsConstructed = Column(Integer) | totalBuildingsConstructed = Column(Integer) | ||||
| totalBuildingsLost = Column(Integer) | totalBuildingsLost = Column(Integer) | ||||
| enemytotalBuildingsDestroyed = Column(Integer) | enemytotalBuildingsDestroyed = Column(Integer) | ||||
| civCentreBuildingsConstructed = Column(Integer) | civCentreBuildingsConstructed = Column(Integer) | ||||
| civCentreBuildingsLost = Column(Integer) | civCentreBuildingsLost = Column(Integer) | ||||
| enemyCivCentreBuildingsDestroyed = Column(Integer) | enemyCivCentreBuildingsDestroyed = Column(Integer) | ||||
| houseBuildingsConstructed = Column(Integer) | houseBuildingsConstructed = Column(Integer) | ||||
| houseBuildingsLost = Column(Integer) | houseBuildingsLost = Column(Integer) | ||||
| enemyHouseBuildingsDestroyed = Column(Integer) | enemyHouseBuildingsDestroyed = Column(Integer) | ||||
| economicBuildingsConstructed = Column(Integer) | economicBuildingsConstructed = Column(Integer) | ||||
| economicBuildingsLost = Column(Integer) | economicBuildingsLost = Column(Integer) | ||||
| enemyEconomicBuildingsDestroyed = Column(Integer) | enemyEconomicBuildingsDestroyed = Column(Integer) | ||||
| outpostBuildingsConstructed = Column(Integer) | outpostBuildingsConstructed = Column(Integer) | ||||
| outpostBuildingsLost = Column(Integer) | outpostBuildingsLost = Column(Integer) | ||||
| enemyOutpostBuildingsDestroyed = Column(Integer) | enemyOutpostBuildingsDestroyed = Column(Integer) | ||||
| militaryBuildingsConstructed = Column(Integer) | militaryBuildingsConstructed = Column(Integer) | ||||
| militaryBuildingsLost = Column(Integer) | militaryBuildingsLost = Column(Integer) | ||||
| enemyMilitaryBuildingsDestroyed = Column(Integer) | enemyMilitaryBuildingsDestroyed = Column(Integer) | ||||
| fortressBuildingsConstructed = Column(Integer) | fortressBuildingsConstructed = Column(Integer) | ||||
| fortressBuildingsLost = Column(Integer) | fortressBuildingsLost = Column(Integer) | ||||
| enemyFortressBuildingsDestroyed = Column(Integer) | enemyFortressBuildingsDestroyed = Column(Integer) | ||||
| wonderBuildingsConstructed = Column(Integer) | wonderBuildingsConstructed = Column(Integer) | ||||
| wonderBuildingsLost = Column(Integer) | wonderBuildingsLost = Column(Integer) | ||||
| enemyWonderBuildingsDestroyed = Column(Integer) | enemyWonderBuildingsDestroyed = Column(Integer) | ||||
| woodBought = Column(Integer) | woodBought = Column(Integer) | ||||
| foodBought = Column(Integer) | foodBought = Column(Integer) | ||||
| stoneBought = Column(Integer) | stoneBought = Column(Integer) | ||||
| metalBought = Column(Integer) | metalBought = Column(Integer) | ||||
| tradeIncome = Column(Integer) | tradeIncome = Column(Integer) | ||||
| percentMapExplored = Column(Integer) | percentMapExplored = Column(Integer) | ||||
| class Game(Base): | class Game(Base): | ||||
| """Model representing games.""" | |||||
| __tablename__ = 'games' | __tablename__ = 'games' | ||||
| id = Column(Integer, primary_key=True) | id = Column(Integer, primary_key=True) | ||||
| map = Column(String(80)) | map = Column(String(80)) | ||||
| duration = Column(Integer) | duration = Column(Integer) | ||||
| teamsLocked = Column(Boolean) | teamsLocked = Column(Boolean) | ||||
| matchID = Column(String(20)) | matchID = Column(String(20)) | ||||
| winner_id = Column(Integer, ForeignKey('players.id')) | winner_id = Column(Integer, ForeignKey('players.id')) | ||||
| player_info = relationship('PlayerInfo', backref='game') | player_info = relationship('PlayerInfo', backref='game') | ||||
| players = relationship('Player', secondary='players_info') | players = relationship('Player', secondary='players_info') | ||||
| if __name__ == '__main__': | def parse_args(args): | ||||
| """Parse command line arguments. | |||||
| Arguments: | |||||
| args (dict): Raw command line arguments given to the script | |||||
| Returns: | |||||
| Parsed command line arguments | |||||
| """ | |||||
| parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter, | |||||
| description="Helper command for database creation") | |||||
| parser.add_argument('action', help='Action to apply to the database', | |||||
| choices=['create']) | |||||
| parser.add_argument('--database-url', help='URL for the leaderboard database', | |||||
| default='sqlite:///lobby_rankings.sqlite3') | |||||
| return parser.parse_args(args) | |||||
| def main(): | |||||
| """Entry point a console script.""" | |||||
| args = parse_args(sys.argv[1:]) | |||||
| engine = create_engine(args.database_url) | |||||
| if args.action == 'create': | |||||
| Base.metadata.create_all(engine) | Base.metadata.create_all(engine) | ||||
| if __name__ == '__main__': | |||||
| main() | |||||
Wildfire Games · Phabricator