From 5e97545b906b75a9acb82fee3c045dd28ec2b8c9 Mon Sep 17 00:00:00 2001
From: Will Toohey <will@mon.im>
Date: Sat, 6 Jan 2018 15:18:21 +1000
Subject: [PATCH] Graceful tree-size fallback

---
 ifstools/ifs.py | 4 +++-
 setup.py        | 2 +-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/ifstools/ifs.py b/ifstools/ifs.py
index 0b65919..1fd73d7 100644
--- a/ifstools/ifs.py
+++ b/ifstools/ifs.py
@@ -67,7 +67,9 @@ class IFS:
         self.manifest = KBinXML(file.data[HEADER_SIZE:])
         self.tree = GenericFolder(self.data_blob, self.manifest.xml_doc)
 
-        assert ifs_tree_size == self.tree_size
+        if ifs_tree_size != self.tree_size:
+            print('Expected tree size {} but got {}. Repacking may fail!'
+                .format(self.tree_size, ifs_tree_size))
 
     def load_dir(self, path):
         self.is_file = False
diff --git a/setup.py b/setup.py
index ff32046..b9eecb2 100644
--- a/setup.py
+++ b/setup.py
@@ -12,7 +12,7 @@ if sys.version_info < (3,0):
 
 setup(
     name='ifstools',
-    version='1.0',
+    version='1.1',
     entry_points = {
         'console_scripts': ['ifstools=ifstools:main'],
     },