Add shorter way to write simple data types

randomdude999 · May 1, 2017 · 950eb0b · 950eb0b
1 parent 3a07f87
commit 950eb0b
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/mcinfo/nbt.py b/mcinfo/nbt.py
@@ -44,6 +44,12 @@ def pretty_format_nbt(nbt):
 
 class NBTTemplate(object):
     def __init__(self, data):
+        if isinstance(data, str):
+            new_data = {
+                'type': "TAG_" + data.split(" ")[0],
+                'desc': " ".join(data.split(" ")[1:])
+            }
+            data = new_data
         if 'type' not in data:
             raise ValueError("Missing data type")
         if data['type'] in nbt_all_types:

diff --git a/tests/test_nbt.py b/tests/test_nbt.py
@@ -36,6 +36,11 @@
 
 class TestNBT(unittest.TestCase):
 
+    def test_init_with_string(self):
+        a = nbt.NBTTemplate("Int test")
+        b = nbt.NBTTemplate({'type': "TAG_Int", 'desc': "test"})
+        self.assertEqual(a, b)
+
     def test_formatting(self):
         nbt_a = nbt.NBTTemplate(testData)
         self.assertEqual(nbt.pretty_format_nbt(nbt_a), expectedFormatting)