From eb6b7aefb0de70311e74922193966a5411fc6056 Mon Sep 17 00:00:00 2001 From: Eljakim Herrewijnen Date: Wed, 10 Nov 2021 22:07:37 +0100 Subject: [PATCH] Added function to remove 'old' services --- services.py | 17 +++++++++++++++-- sqlite.db | Bin 24576 -> 24576 bytes 2 files changed, 15 insertions(+), 2 deletions(-) diff --git a/services.py b/services.py index d0e485c..72f4653 100644 --- a/services.py +++ b/services.py @@ -51,6 +51,8 @@ class Services: self.cursor.execute(query) def AddService(self, date, groups): + if(date < datetime.datetime.now()): + return "You cant add a date in the past" query = f""" INSERT INTO services (date, groups) VALUES('{date}', '{groups}'); @@ -70,6 +72,17 @@ class Services: self.cursor.execute(query) self.db.commit() + def CleanServices(self): + """ + Remove all services that are in the past + """ + ser = self.GetAllServices() + for i in ser: + if(ser[i][0] < datetime.datetime.now()): + #delete service that lives in the past + self.RemoveService(ser[i][0]) + + class UserManager(): def __init__(self, db) -> None: self.db = db @@ -126,9 +139,9 @@ if __name__ == "__main__": ser = Services(db) us = UserManager(db) - us.AddUser("Eljakim", "Kunnenwe?") + # us.AddUser("Eljakim", "Kunnenwe?") us.Authenticate("Eljakim", "Kunnenwe?") - us.Authenticate("Eljakim", "Kunnenwe2?") + ser.CleanServices() # for i in range(100): # d = datetime.datetime.now() # d = d.replace(hour=random.randint(1, 23), day = random.randint(1, 28), month = random.randint(1, 12)) diff --git a/sqlite.db b/sqlite.db index 9caeaeb83f564ad812b8614b9b7c448dc5f596ec..37ba680ef0fea8bfa082299318c490b1712f39ac 100644 GIT binary patch delta 60 zcmZoTz}Rqrae_1>`$QRMM)r*fOZeFs1Q;0jFK-r9IK$7VJF!udi;DpWK!Pte3kE#o NXOx)eD6+`F0086N4>AA% delta 120 zcmZoTz}Rqrae_1>+e8^>Mz)O!OZeFs`7bl@U*0UJaFX9#n~8