From f5b45dc7db7c8d008c0a270adc169f1bb746f082 Mon Sep 17 00:00:00 2001 From: shaaqu Date: Wed, 10 Jun 2020 11:58:27 +0200 Subject: [PATCH] levels method --- shelf.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/shelf.py b/shelf.py index 0d74593..82eede8 100644 --- a/shelf.py +++ b/shelf.py @@ -25,3 +25,15 @@ class Shelf: def get_color(self): return self.color + + def levels(self): + r = '' + str = [0, 0, 0] + if len(self.level1) > 0: + str[0] = 1 + if len(self.level2) > 0: + str[1] = 1 + if len(self.level1) > 0: + str[2] = 1 + r.join(str) + return r \ No newline at end of file