From 85ae90eab815856861777fa4eccc1288450c7a99 Mon Sep 17 00:00:00 2001
From: Jylpah <jylpah@gmail.com>
Date: Tue, 2 Oct 2018 14:40:11 +0800
Subject: [PATCH] Get Tankopedia JSON data from wotinspector.com

---
 getTankopedia.py | 67 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 67 insertions(+)
 create mode 100644 getTankopedia.py

diff --git a/getTankopedia.py b/getTankopedia.py
new file mode 100644
index 0000000..7e3d24d
--- /dev/null
+++ b/getTankopedia.py
@@ -0,0 +1,67 @@
+#!/usr/local/bin/python3.7    
+
+# Get Tankpedia from WoTinspector.com
+
+import sys, argparse, json, os, inspect, aiohttp, asyncio, re, logging
+from pathlib import Path
+
+logging.getLogger("asyncio").setLevel(logging.DEBUG)
+
+VERBOSE = False
+DEBUG   = False
+
+tank_type = [ 'lightTank', 'mediumTank', 'heavyTank', 'AT-SPG' ]
+
+#@asyncio.coroutine
+async def get_tankopedia(filename: str):
+    url="https://wotinspector.com/static/armorinspector/tank_db_blitz.js"
+    async with aiohttp.ClientSession() as client:
+        async with client.get(url) as r:
+            if r.status == 200:
+                WI_tank_db=await r.text()
+                WI_tank_db = WI_tank_db.split("\n")
+            else:
+                print('Error: Could not get valid HTTPS response. HTTP: ' + str(r.status) )  
+                sys.exit(1) 
+            tanks = {}
+            n = 0
+            p = re.compile('\\s*(\\d+):\\s{"en":"([^"]+)",.*?"tier":(\\d+), "type":(\\d), "premium":(\\d).*')
+            for line in WI_tank_db[1:-1]:
+                m = p.match(line)
+                tank = {}
+                tank['tank_id'] = int(m.group(1))
+                tank['name'] = m.group(2)
+                tank['tier'] = int(m.group(3))
+                tank['type'] = tank_type[int(m.group(4))]
+                tank['is_premium'] = (int(m.group(5)) == 1)
+                tanks[str(m.group(1))] = tank
+                n += 1
+
+            tankopedia = {}
+            tankopedia['status'] = "ok"
+            tankopedia['meta'] = {"count" : n}
+            tankopedia['data'] = tanks
+            print("Tankopedia has " + str(n) + " tanks in: " + filename)
+            with open(filename,'w') as outfile:
+                outfile.write(json.dumps(tankopedia, ensure_ascii=False, indent=4, sort_keys=True))
+            return None
+
+
+def main(argv):
+    global VERBOSE, DEBUG
+
+    parser = argparse.ArgumentParser(description='Retrieve Tankopedia from WoTinspector.com')
+    parser.add_argument('--file', dest="outfile", help='Write Tankopedia to file')
+    parser.add_argument('--debug', '-d', action='store_true', default=False, help='Debug mode')
+    parser.add_argument('--verbose', '-v', action='store_true', default=False, help='Verbose mode')
+    
+    args = parser.parse_args()
+    VERBOSE = args.verbose
+    DEBUG = args.debug
+    if DEBUG: VERBOSE = True
+
+    asyncio.run(get_tankopedia(args.outfile))
+    
+
+if __name__ == "__main__":
+   main(sys.argv[1:])