11.11共享流量包推荐

class SharedTrafficPackage:
    def __init__(self, total_traffic):
        self.total_traffic = total_traffic
        self.used_traffic = 0
        self.users = {}

    def add_user(self, user_id, initial_quota):
        if self.total_traffic - self.used_traffic >= initial_quota:
            self.users[user_id] = initial_quota
            self.used_traffic += initial_quota
            return True
        return False

    def use_traffic(self, user_id, amount):
        if user_id in self.users and self.users[user_id] >= amount:
            self.users[user_id] -= amount
            self.used_traffic -= amount
            return True
        return False

    def get_user_quota(self, user_id):
        return self.users.get(user_id, 0)

# 示例使用
traffic_package = SharedTrafficPackage(total_traffic=1000)
traffic_package.add_user('user1', 300)
traffic_package.use_traffic('user1', 100)
print(f"User1 remaining quota: {traffic_package.get_user_quota('user1')}")