From 053325eca6a729cfb97b9847186af101eb566553 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?David=20Bergstr=C3=B6m?= <davbe125@student.liu.se>
Date: Wed, 25 Jul 2018 16:42:07 +0200
Subject: [PATCH] Make looking up TechTree data about upgrades possible in
 Python

---
 python-api-src/library.cpp | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/python-api-src/library.cpp b/python-api-src/library.cpp
index 7f54cb2..020208d 100644
--- a/python-api-src/library.cpp
+++ b/python-api-src/library.cpp
@@ -38,7 +38,6 @@ PYBIND11_MODULE(library, m)
     TODO: These also needs to be defined
 
     typedef SC2Type<ABILITY_ID>   AbilityID;
-    typedef SC2Type<UPGRADE_ID>   UpgradeID;
     typedef SC2Type<BUFF_ID>      BuffID;
     */
     py::class_<sc2::UnitTypeID>(m, "UnitTypeID")
@@ -46,6 +45,11 @@ PYBIND11_MODULE(library, m)
 
     py::implicitly_convertible<sc2::UNIT_TYPEID, sc2::UnitTypeID>();
 
+    py::class_<sc2::UpgradeID>(m, "UpgradeID")
+        .def(py::init<sc2::UPGRADE_ID>());
+
+    py::implicitly_convertible<sc2::UPGRADE_ID, sc2::UpgradeID>();
+
     py::class_<sc2::Agent>(m, "Agent")
         .def(py::init());
 
-- 
GitLab