劳保类别模块代码实现
时间: 2023-08-11 15:08:33 浏览: 44
以下是一个简单的劳保类别模块的代码实现,用于管理不同类型的劳保用品。
```python
class LaborProtection:
def __init__(self, name, price, stock):
self.name = name
self.price = price
self.stock = stock
class LaborProtectionCategory:
def __init__(self, name):
self.name = name
self.labor_protections = []
def add_labor_protection(self, labor_protection):
if isinstance(labor_protection, LaborProtection):
self.labor_protections.append(labor_protection)
def remove_labor_protection(self, labor_protection):
if isinstance(labor_protection, LaborProtection) and labor_protection in self.labor_protections:
self.labor_protections.remove(labor_protection)
def get_total_price(self):
total_price = 0
for labor_protection in self.labor_protections:
total_price += labor_protection.price * labor_protection.stock
return total_price
```
这个类实现了两个类:`LaborProtection` 和 `LaborProtectionCategory`。 `LaborProtection` 类表示单个劳保用品,具有名称、价格和库存属性。 `LaborProtectionCategory` 类表示劳保用品类别,具有名称和劳保用品列表属性,可以添加和删除劳保用品。 `get_total_price` 方法用于计算该类别的所有劳保用品的总价格。