upd scheduler

1 files changed, 23 insertions, 9 deletions

diff --git a/main.py b/main.py
index 5ff4015..14cb816 100644
--- a/main.py
+++ b/main.py
@@ -1,18 +1,32 @@
+from datetime import datetime
+
 from downloader import downloadCsv
 from csvReader import readCSV
 from bot import Bot
+from schedule import every, repeat, run_pending
+import time
 
 
+@repeat(every().day.at("09:00"))
 def main():
-    downloadCsv()
-    myScore = readCSV()
-    try:
-        myScore = 100 * float(myScore)
-    except ValueError:
-        myScore = 0
-    bot = Bot()
-    bot.send_message(f'Score leantime ente bulan ini {myScore}%')
+    today = datetime.today()
+    weekday = today.weekday()
+    if weekday not in [5, 6]:  # jangan ganggu aku di akhir pekan
+        downloadCsv()
+        myScore = readCSV()
+        try:
+            myScore = 100 * float(myScore)
+        except ValueError:
+            myScore = 0
+        bot = Bot()
+        msg = f'Isi leantime... score ente bulan ini {myScore}%'
+        if (17 <= today.day <= 24) and myScore < 70:  # seminggu sebelumnya udah ngingetin
+            msg = f'*WARNING!!* SEKARANG UDAH TANGGAL {today.day}, SCORE LEANTIME MASIH {myScore}%'
+
+        bot.send_message(msg)
 
 
 if __name__ == '__main__':
-    main()
+    while True:
+        run_pending()
+        time.sleep(3600)