Add simple creature stat builder

1 files changed, 6 insertions, 3 deletions

diff --git a/main.py b/main.py
index 138da27..22be527 100644
--- a/main.py
+++ b/main.py
@@ -1,6 +1,7 @@
 import math
 from character import CharacterPlan
-from game import Roll, d
+from game import Difficulty, Roll, d
+from creature import CreaturePlan
 
 import pb2e
 
@@ -155,8 +156,10 @@ def main4():
     print(char.toJSON())
 
 def main5():
-    from creature import stats
-    print(stats.stats)
+    creature_plan = CreaturePlan(default_difficulty=Difficulty.Moderate)
+    creature_plan["reflex"] = Difficulty.High
+    creature = creature_plan.build(10)
+    print(creature)
 
 
 if __name__ == '__main__':