updating for individual projecet

models/dumpster.py

@@ -1,84 +1,86 @@
-import numpy as np
+import numpy as np
-import pygame
+import pygame
-
+
-class trash():
+class trash():
-    def __init__(self):
+    def __init__(self):
-        self.id = 0
+        self.id = 0
-        self.plastic = 0
+        self.plastic = 0
-        self.paper = 0
+        self.paper = 0
-        self.mixed = 0
+        self.mixed = 0
-        self.organic = 0
+        self.organic = 0
-        self.glass = 0
+        self.glass = 0
-        self.color = 0
+        self.color = 0
-        self.xy = 0
+        self.xy = 0
-        self.value = 1
+        self.value = 1
-
+        self.numberx = 0
-    def giveID(self,id):
+        self.numbery = 0
-        self.id = id
+
-
+    def giveID(self,id):
-
+        self.id = id
-
+
-    def throwAway(self):
+
-        trash = np.random.randint(1,6)
+
-        if trash == 1:
+    def throwAway(self):
-            self.plastic += 1
+        trash = np.random.randint(1,6)
-            return True
+        if trash == 1:
-        elif trash == 2:
+            self.plastic += 1
-            self.paper += 1
+            return True
-            return True
+        elif trash == 2:
-        elif trash == 3:
+            self.paper += 1
-            self.glass += 1
+            return True
-            return True
+        elif trash == 3:
-        elif trash == 4:
+            self.glass += 1
-            self.organic += 1
+            return True
-            return True
+        elif trash == 4:
-        elif trash == 5:
+            self.organic += 1
-            self.mixed += 1
+            return True
-            return True
+        elif trash == 5:
-        else:
+            self.mixed += 1
-            return False
+            return True
-
+        else:
-
+            return False
-    def empty(self):
+
-        if self.plastic > 0:
+
-            self.plastic -= 1
+    def empty(self):
-            print("taking plastic")
+        if self.plastic > 0:
-            return True
+            self.plastic -= 1
-        elif self.paper > 0:
+            print("taking plastic")
-            self.paper -= 1
+            return True
-            print("taking paper")
+        elif self.paper > 0:
-            return True
+            self.paper -= 1
-        elif self.glass > 0:
+            print("taking paper")
-            self.glass -= 1
+            return True
-            print("taking glass")
+        elif self.glass > 0:
-            return True
+            self.glass -= 1
-        elif self.organic > 0:
+            print("taking glass")
-            self.organic -= 1
+            return True
-            print("taking organic")
+        elif self.organic > 0:
-            return True
+            self.organic -= 1
-        elif self.mixed > 0:
+            print("taking organic")
-            self.mixed -= 1
+            return True
-            print("taking mixed")
+        elif self.mixed > 0:
-            return True
+            self.mixed -= 1
-        else:
+            print("taking mixed")
-            print("can't take trash from this dumpster")
+            return True
-            return False
+        else:
-
+            print("can't take trash from this dumpster")
-    def colour(self):
+            return False
-        if self.plastic > 0:
+
-            self.color = 1
+    def colour(self):
-            return True
+        if self.plastic > 0:
-        elif self.paper > 0:
+            self.color = 1
-            self.color = 2
+            return True
-            return True
+        elif self.paper > 0:
-        elif self.glass > 0:
+            self.color = 2
-            self.color = 3
+            return True
-            return True
+        elif self.glass > 0:
-        elif self.organic > 0:
+            self.color = 3
-            self.color = 4
+            return True
-            return True
+        elif self.organic > 0:
-        elif self.mixed > 0:
+            self.color = 4
-            self.color = 5
+            return True
-            return True
+        elif self.mixed > 0:
-        else:
+            self.color = 5
-            return False
+            return True
+        else:
+            return False