From 03e29e02b1c4cfeb98a67c580cf7ebe02bca3567 Mon Sep 17 00:00:00 2001
From: blitzmann <holmes.ryan.90@gmail.com>
Date: Thu, 25 Jun 2020 22:04:04 -0400
Subject: [PATCH] update unit to include all the translations

---
 eos/db/gamedata/unit.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/eos/db/gamedata/unit.py b/eos/db/gamedata/unit.py
index e99258050..9b33dae9f 100644
--- a/eos/db/gamedata/unit.py
+++ b/eos/db/gamedata/unit.py
@@ -26,8 +26,7 @@ from eos.gamedata import Unit
 groups_table = Table("dgmunits", gamedata_meta,
                      Column("unitID", Integer, primary_key=True),
                      Column("unitName", String),
-                     Column("displayName", String),
-                     Column("displayName_zh", String),
+                     *[Column("displayName{}".format(lang), String) for lang in eos.config.translation_mapping.values()],
                      )
 
 mapper(Unit, groups_table,