Zaktualizuj 'field.py'

field.py

@@ -1,43 +1,58 @@
-import pygame
-from colors import *
-from dimensions import *
-
-class Field(pygame.sprite.Sprite):
-    def __init__(self, row, column):
-        super(Field, self).__init__()
-        self.surf = pygame.Surface((WIDTH, HEIGHT))
-        self.surf.fill(BROWN)
-        self.rect = self.surf.get_rect(
-            topleft=((MARGIN + WIDTH) * row + MARGIN, (MARGIN + HEIGHT) * column + MARGIN))
-        self.position = [row, column]
-        self.hydration = 0
-        self.isRocky = False
-        self.planted = 0
-        self.ferility = 1
-
-    def hydrate(self):
-        if self.hydration <= 3:
-            self.hydration += 1
-            if self.hydration == 0:
-                self.surf.fill(BROWN)
-            if self.hydration == 1:
-                self.surf.fill(YELLOW)
-            if self.hydration == 2:
-                self.surf.fill(GREEN)
-            if self.hydration == 3:
-                self.surf.fill(BLUE)
-    
-    def dehydrate(self):
-        if self.hydration > 0:
-            self.hydration -= 1
-            if self.hydration == 0:
-                self.surf.fill(BROWN)
-            if self.hydration == 1:
-                self.surf.fill(YELLOW)
-            if self.hydration == 2:
-                self.surf.fill(GREEN)
-            if self.hydration == 3:
-                self.surf.fill(BLUE)
-    
-    def free(self):
-        self.planted = 0
+import pygame
+from colors import *
+from dimensions import *
+
+class Field(pygame.sprite.Sprite):
+    def __init__(self, row, column, field_type):
+        super(Field, self).__init__()
+        self.surf = pygame.Surface((WIDTH, HEIGHT))
+        self.field_type = field_type
+        if self.field_type == "soil":
+            self.moveCost = 3
+            self.surf.fill(BROWN0)
+        elif self.field_type == "rocks":
+            self.moveCost = 5
+            self.surf.fill(LBROWN)
+        elif self.field_type == "road":
+            self.moveCost = 1
+            self.surf.fill(GREY)
+        elif self.field_type == "pond":
+            self.moveCost = 1000
+            self.surf.fill(BLUE)
+        self.rect = self.surf.get_rect(
+            topleft=((MARGIN + WIDTH) * row + MARGIN, (MARGIN + HEIGHT) * column + MARGIN))
+        self.position = [row, column]
+        self.hydration = 0
+        self.planted = 0
+        self.ferility = 1
+
+    def hydrate(self):
+        if self.field_type == "soil" and self.hydration <= 4:
+            self.hydration += 1
+            if self.hydration == 0:
+                self.surf.fill(BROWN0)
+            if self.hydration == 1:
+                self.surf.fill(BROWN1)
+            if self.hydration == 2:
+                self.surf.fill(BROWN2)
+            if self.hydration == 3:
+                self.surf.fill(BROWN3)
+            if self.hydration == 4:
+                self.surf.fill(BROWN4)
+    
+    def dehydrate(self):
+        if self.field_type == "soil" and self.hydration > 0:
+            self.hydration -= 1
+            if self.hydration == 0:
+                self.surf.fill(BROWN0)
+            if self.hydration == 1:
+                self.surf.fill(BROWN1)
+            if self.hydration == 2:
+                self.surf.fill(BROWN2)
+            if self.hydration == 3:
+                self.surf.fill(BROWN3)
+            if self.hydration == 4:
+                self.surf.fill(BROWN4)
+    
+    def free(self):
+        self.planted = 0