Split jobs.py into files

jobs/grab_sapling.py

@@ -0,0 +1,91 @@
+import re
+import time
+import importlib
+import random
+from itertools import count
+from math import hypot, floor
+
+from minecraft.networking.types import BlockFace
+
+from protocol.managers import ChunkNotLoadedException
+
+import utils
+importlib.reload(utils)
+import path
+importlib.reload(path)
+import blocks
+importlib.reload(blocks)
+import items
+importlib.reload(items)
+import mcdata
+importlib.reload(mcdata)
+import mobs
+importlib.reload(mobs)
+
+class GrabSaplingStates:
+    def idle(self):
+        return None
+
+    def init(self):
+        self.state = self.find_saplings
+        print('Trying to grab a sapling')
+
+    def find_saplings(self):
+        w = self.g.world
+        p = utils.pint(self.g.pos)
+
+        saplings = w.find_objects(items.SAPLING_IDS)
+
+        if not saplings:
+            print('No sapling objects found, aborting')
+            self.state = self.cleanup
+            return
+
+        saplings.sort(key=lambda s: utils.phyp(p, (s.x, s.y, s.z)))
+
+        for s in saplings:
+            s_pos = utils.pint((s.x, s.y, s.z))
+
+            check = utils.padd(s_pos, path.BLOCK_BELOW)
+
+            if s.entity_id in self.eid_blacklist:
+                continue
+
+            # skip if the sapling is floating
+            if self.g.chunks.get_block_at(*check) in blocks.LEAF_IDS | {0}:
+                continue
+
+            navpath = w.path_to_place(p, s_pos)
+
+            if navpath:
+                self.g.path = navpath
+                self.state = self.going_to_sapling
+                self.sapling = s_pos
+                self.eid_blacklist.append(s.entity_id)
+                print('Going to sapling', self.sapling)
+                return
+
+        print('Cant get to any more saplings, aborting')
+        self.state = self.cleanup
+
+    def going_to_sapling(self):
+        if utils.pint(self.g.pos) == self.sapling:
+            self.state = self.find_saplings
+
+    def cleanup(self):
+        self.g.look_at = None
+        self.state = self.done
+
+    def done(self):
+        # never gets ran, placeholder
+        return None
+
+    def __init__(self, global_state):
+        self.g = global_state
+        self.state = self.idle
+
+        self.sapling = None
+        self.eid_blacklist = []
+
+    def run(self):
+        self.state()