feat(scripts): add annotate and filter arguments to convert.py

scripts/convert.py

@@ -1,9 +1,12 @@
 #!/usr/bin/env python
+import io
 import sys
 from argparse import ArgumentParser, ArgumentTypeError
 from datetime import datetime
 
+from hslog.filter import BattlegroundsLogFilter
 from hsreplay.document import HSReplayDocument
+from hsreplay.utils import annotate_replay
 
 
 def date_arg(s):
@@ -17,13 +20,23 @@ def main():
 	parser = ArgumentParser()
 	parser.add_argument("files", nargs="*")
 	parser.add_argument("--processor", dest="processor", default="GameState")
+	parser.add_argument("--battlegrounds-filter", action="store_true")
+	parser.add_argument("--annotate", action="store_true")
 	parser.add_argument("--default-date", dest="date", type=date_arg, help="Format: YYYY-MM-DD")
 	# https://stackoverflow.com/questions/9226516/
 	args = parser.parse_args(sys.argv[1:])
 	for filename in args.files:
 		with open(filename) as f:
+			if args.battlegrounds_filter
+				f = BattlegroundsLogFilter(f)
 			doc = HSReplayDocument.from_log_file(f, args.processor, args.date)
-			print(doc.to_xml())
+			xml = doc.to_xml()
+			if args.annotate:
+				out = io.BytesIO()
+				annotate_replay(io.BytesIO(xml.encode("utf-8")), out)
+				xml = out.getvalue().decode("utf-8")
+
+			print(xml)
 
 
 if __name__ == "__main__":