diff options
| -rw-r--r-- | FortniteStatus.py | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/FortniteStatus.py b/FortniteStatus.py index 8b5054e..7702482 100644 --- a/FortniteStatus.py +++ b/FortniteStatus.py @@ -15,6 +15,11 @@ class FortniteStatus: def __init__(self, serviceStatuses): self.serviceStatuses = serviceStatuses + + def __eq__(self, other): + if not isinstance(other, FortniteStatus.Status): + return NotImplemented + return sorted(self.serviceStatuses) == sorted(other.serviceStatuses) def prettify(self): return 'Fortnite services status:\n' + '\n'.join([serviceStatus.prettify() for serviceStatus in self.serviceStatuses]) @@ -26,6 +31,17 @@ class FortniteStatus: def __init__(self, serviceName, status): self.serviceName = serviceName self.status = status + + def __lt__(self, other): + if not isinstance(other, FortniteStatus.ServiceStatus): + return NotImplemented + return self.serviceName < other.serviceName + + def __eq__(self, other): + if not isinstance(other, FortniteStatus.ServiceStatus): + return NotImplemented + return self.serviceName == other.serviceName and self.status == other.status + def prettify(self): return f'{self.serviceName}, {self.status}' |