Improve trading villager selection based on profession

mosfet/info/items.py

@@ -103,6 +103,7 @@ WHEAT_ID = get_id('wheat')
 WHEAT_SEEDS_ID = get_id('wheat_seeds')
 BEETROOT_SEEDS_ID = get_id('beetroot_seeds')
 PUMPKIN_ID = get_id('pumpkin')
+BEETROOT_ID = get_id('beetroot')
 
 EMERALD_ID = get_id('emerald')
 BERRIES_ID = get_id('sweet_berries')

mosfet/info/mobs.py

@@ -1,8 +1,33 @@
+from mosfet.info import items
 import json
 
 with open('minecraft_data/registries.json') as f:
     MOBS = json.load(f)['minecraft:entity_type']['entries']
 
+VILLAGER = 93
+ARMORER = 1
+BUTCHER = 2
+CARTOGRAPHER = 3
+CLERIC = 4
+FARMER = 5
+FISHERMAN = 6
+FLETCHER = 7
+LEATHERWORKER = 8
+LIBRARIAN = 9
+MASON = 10
+NITWIT = 11
+SHEPHERD = 12
+TOOLSMITH = 13
+WEAPONSMITH = 14
+
+TRADES = {
+    ARMORER: [items.IRON_INGOT_ID],
+    BUTCHER: [items.BERRIES_ID],
+    FARMER: [items.PUMPKIN_ID, items.WHEAT_ID, items.POTATO_ID, items.CARROT_ID, items.BEETROOT_ID],
+    TOOLSMITH: [items.IRON_INGOT_ID],
+    WEAPONSMITH: [items.IRON_INGOT_ID],
+}
+
 EVIL = [
     'blaze',
     'cave_spider',