125 lines
2.7 KiB
Python
125 lines
2.7 KiB
Python
from random import choice as rd
|
|
|
|
from classes.trashFunctions import generatePicList
|
|
|
|
|
|
class Trash:
|
|
tdic = {"paper": 1,
|
|
"bio_waste": 0,
|
|
"mixed": 4,
|
|
"glass": 3,
|
|
"metals_and_plastics": 2}
|
|
|
|
def __init__(self):
|
|
self.tree_type = 0
|
|
self.ttype = "Trash"
|
|
self.real_type = "Trash"
|
|
self.scan_id = 0
|
|
self.waga = 1
|
|
self.image = None
|
|
|
|
def getRealtype(self):
|
|
return self.real_type
|
|
|
|
def getScanid(self):
|
|
return self.scan_id
|
|
|
|
def getTreetype(self):
|
|
return self.tree_type
|
|
|
|
def setTreetype(self, treetype):
|
|
self.tree_type = treetype
|
|
return self
|
|
|
|
def getTtype(self):
|
|
return self.ttype
|
|
|
|
def setTtype(self, ttype):
|
|
self.ttype = ttype
|
|
self.setTreetype(self.tdic[ttype])
|
|
return self
|
|
|
|
def getWaga(self):
|
|
return self.waga
|
|
|
|
def setWaga(self, waga):
|
|
self.waga = waga
|
|
return self
|
|
|
|
def getImage(self):
|
|
return self.image
|
|
|
|
def setImage(self, image):
|
|
self.image = image
|
|
return self
|
|
|
|
|
|
class Papier(Trash):
|
|
lista_obrazow = generatePicList("paper")
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.tree_type = 1
|
|
self.ttype = "paper"
|
|
self.real_type = "Papier"
|
|
self.scan_id = 0
|
|
self.waga = 2
|
|
self.image = None
|
|
self.setImage(rd(self.lista_obrazow))
|
|
|
|
|
|
class MetalPlastik(Trash):
|
|
lista_obrazow = generatePicList("plastic")
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.tree_type = 2
|
|
self.ttype = "metals_and_plastics"
|
|
self.real_type = "MetalPlastik"
|
|
self.scan_id = 4
|
|
self.waga = 3
|
|
self.image = None
|
|
self.setImage(rd(self.lista_obrazow))
|
|
|
|
|
|
class Mixed(Trash):
|
|
lista_obrazow = generatePicList("mixed")
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.tree_type = 4
|
|
self.ttype = "mixed"
|
|
self.real_type = "Mixed"
|
|
self.scan_id = 2
|
|
self.waga = 1
|
|
self.image = None
|
|
self.setImage(rd(self.lista_obrazow))
|
|
|
|
|
|
class Bio(Trash):
|
|
lista_obrazow = generatePicList("bio")
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.tree_type = 0
|
|
self.ttype = "bio"
|
|
self.real_type = "Bio"
|
|
self.scan_id = 1
|
|
self.waga = 2
|
|
self.image = None
|
|
self.setImage(rd(self.lista_obrazow))
|
|
|
|
|
|
class Szklo(Trash):
|
|
lista_obrazow = generatePicList("glass")
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.tree_type = 3
|
|
self.ttype = "glass"
|
|
self.real_type = "Szklo"
|
|
self.scan_id = 3
|
|
self.waga = 5
|
|
self.image = None
|
|
self.setImage(rd(self.lista_obrazow))
|