xonstat/models/game.py

   1 from calendar import timegm
   2
   3 from xonstat.util import pretty_date, strip_colors, html_colors
   4
   5
   6 class Game(object):
   7     def __init__(self, game_id=None, start_dt=None, game_type_cd=None,
   8             server_id=None, map_id=None, winner=None):
   9         self.game_id = game_id
  10         self.start_dt = start_dt
  11         self.game_type_cd = game_type_cd
  12         self.server_id = server_id
  13         self.map_id = map_id
  14         self.winner = winner
  15
  16     def __repr__(self):
  17         return "<Game(%s, %s, %s, %s)>" % (self.game_id, self.start_dt, self.game_type_cd, self.server_id)
  18
  19     def to_dict(self):
  20         return {'game_id':self.game_id, 'start':self.start_dt.strftime('%Y-%m-%dT%H:%M:%SZ'),
  21                 'game_type_cd':self.game_type_cd, 'server_id':self.server_id}
  22
  23     def fuzzy_date(self):
  24         return pretty_date(self.start_dt)
  25
  26     def epoch(self):
  27         return timegm(self.start_dt.timetuple())
  28
  29
  30 class PlayerGameStat(object):
  31     def __init__(self, player_game_stat_id=None, create_dt=None):
  32         self.player_game_stat_id = player_game_stat_id
  33         self.create_dt = create_dt
  34
  35     def __repr__(self):
  36         return "<PlayerGameStat(%s, %s, %s)>" % (self.player_id, self.game_id, self.create_dt)
  37
  38     def to_dict(self):
  39         return {'player_id':self.player_id, 'game_id':self.game_id,
  40             'create_dt':self.create_dt.strftime('%Y-%m-%dT%H:%M:%SZ'),
  41             'alivetime':self.alivetime, 'rank':self.rank, 'score':self.score, 'team':self.team}
  42
  43     def nick_stripped(self):
  44         if self.nick is None:
  45             return "Anonymous Player"
  46         else:
  47             return strip_colors(self.nick)
  48
  49     def nick_html_colors(self, limit=None):
  50         if self.nick is None:
  51             return "Anonymous Player"
  52         else:
  53             return html_colors(self.nick, limit)
  54
  55     def team_html_color(self):
  56         if self.team == 5:
  57             return "red"
  58         if self.team == 14:
  59             return "blue"
  60         if self.team == 13:
  61             return "yellow"
  62         if self.team == 10:
  63             return "pink"
  64
  65
  66 class PlayerWeaponStat(object):
  67     def __init__(self, player_id=None, game_id=None, weapon_cd=None):
  68         self.player_id = player_id
  69         self.game_id = game_id
  70         self.weapon_cd = weapon_cd
  71         self.fired = 0
  72         self.max = 0
  73         self.hit = 0
  74         self.actual = 0
  75         self.frags = 0
  76
  77     def __repr__(self):
  78         return "<PlayerWeaponStat(%s, %s, %s)>" % (self.player_weapon_stats_id, self.player_id, self.game_id)
  79
  80     def to_dict(self):
  81         return {
  82             'weapon_cd':self.weapon_cd,
  83             'player_weapon_stats_id':self.player_weapon_stats_id,
  84             'player_id':self.player_id,
  85             'game_id':self.game_id,
  86             'fired':self.fired,
  87             'max':self.max,
  88             'hit':self.hit,
  89             'actual':self.actual,
  90             'frags':self.frags,
  91         }
  92
  93
  94 class TeamGameStat(object):
  95     def __init__(self, team_game_stat_id=None, create_dt=None):
  96         self.team_game_stat_id = team_game_stat_id
  97         self.create_dt = create_dt
  98
  99     def __repr__(self):
 100         return "<TeamGameStat(%s, %s, %s)>" % (self.team_game_stat_id, self.game_id, self.team)
 101
 102     def to_dict(self):
 103         return {
 104             'team_game_stat_id':self.team_game_stat_id,
 105             'game_id':self.game_id,
 106             'team':self.team,
 107             'score':self.score,
 108             'rounds':self.rounds,
 109             'caps':self.caps,
 110             'create_dt':self.create_dt.strftime('%Y-%m-%dT%H:%M:%SZ'),
 111         }
 112
 113     def team_html_color(self):
 114         if self.team == 5:
 115             return "red"
 116         if self.team == 14:
 117             return "blue"
 118         if self.team == 13:
 119             return "yellow"
 120         if self.team == 10:
 121             return "pink"
 122
 123
 124 class PlayerGameAnticheat(object):
 125     def __init__(self, player_id=None, game_id=None, key=None,
 126             value=None, create_dt=None):
 127         self.player_id                = player_id
 128         self.game_id                  = game_id
 129         self.key                      = key
 130         self.value                    = value
 131         self.create_dt                = create_dt
 132
 133     def __repr__(self):
 134         return "<PlayerGameAnticheat(%s, %d)>" % (self.key, self.value)
 135
 136
 137 class GameType(object):
 138     def __repr__(self):
 139         return "<GameType(%s, %s, %s)>" % (self.game_type_cd, self.descr, self.active_ind)
 140
 141     def to_dict(self):
 142         return {'game_type_cd':self.game_type_cd, 'name':self.descr, 'active':self.active_ind}
 143
 144
 145 class Weapon(object):
 146     def __repr__(self):
 147         return "<Weapon(%s, %s, %s)>" % (self.weapon_cd, self.descr, self.active_ind)
 148
 149     def to_dict(self):
 150         return {'weapon_cd':self.weapon_cd, 'name':self.descr, 'active':self.active_ind}