implement block class

module/block.py

@@ -0,0 +1,54 @@
+from module import COLORS
+
+class Block:
+    id = ''
+    color = ''
+    flowable = False # 可流动
+    fallible = False # 可下坠
+    collision = True # 可以碰撞
+
+    def __init__(self, x: int, y: int):
+        self.x = x
+        self.y = y
+
+
+class GrassBlock(Block):
+    id = 'grass'
+    color = COLORS.get(id)
+
+    def __init__(self, x: int, y: int):
+        super().__init__(x, y)
+
+
+class DirtBlock(Block):
+    id = 'dirt'
+    color = COLORS.get(id)
+
+    def __init__(self, x: int, y: int):
+        super().__init__(x, y)
+
+
+class SandBlock(Block):
+    id = 'sand'
+    color = COLORS.get(id)
+    flowable = True
+
+    def __init__(self, x: int, y: int):
+        super().__init__(x, y)
+
+
+class StoneBlock(Block):
+    id = 'stone'
+    color = COLORS.get(id)
+
+    def __init__(self, x: int, y: int):
+        super().__init__(x, y)
+
+
+class AirBlock(Block):
+    id = 'air'
+    color = COLORS.get(id)
+    collision = False
+
+    def __init__(self, x: int, y: int):
+        super().__init__(x, y)

module/utils.py

@@ -1,9 +1,10 @@
 import random
 from module import WIDTH, BLOCK_SIZE, HEIGHT
+from module.block import StoneBlock, GrassBlock, DirtBlock, AirBlock, Block
 
 
 # 生成地形
-def generate_terrain():
+def generate_terrain() -> list[list[Block]]:
     world = []
     ground_level = HEIGHT // BLOCK_SIZE // 2
 
@@ -12,13 +13,13 @@ def generate_terrain():
         height = ground_level + random.randint(-3, 3)
         for y in range(HEIGHT // BLOCK_SIZE):
             if y > height + 2:
-                column.append("stone")
+                column.append(StoneBlock(x, y))
             elif y == height:
-                column.append("grass")
+                column.append(GrassBlock(x, y))
             elif y > height - 3:
-                column.append("dirt")
+                column.append(DirtBlock(x, y))
             else:
-                column.append("air")
+                column.append(AirBlock(x, y))
         world.append(column)
 
     return world