From b1d1af5100c6782b2395e18486032239083ca577 Mon Sep 17 00:00:00 2001
From: Schimon Jehudah <sjehuda@yandex.com>
Date: Thu, 12 Oct 2023 11:09:31 +0000
Subject: [PATCH] Use loop.call_at

---
 slixfeed/__main__.py | 31 +++++++++++++++++++++++++++----
 1 file changed, 27 insertions(+), 4 deletions(-)

diff --git a/slixfeed/__main__.py b/slixfeed/__main__.py
index c3b770b..50718d2 100644
--- a/slixfeed/__main__.py
+++ b/slixfeed/__main__.py
@@ -64,6 +64,8 @@ class Slixfeed(slixmpp.ClientXMPP):
         # MUC messages and error messages.
         self.add_event_handler("message", self.message)
         self.add_event_handler("disconnected", self.reconnect)
+        # Initialize event loop
+        self.loop = asyncio.get_event_loop()
 
     async def start(self, event):
         """
@@ -185,6 +187,8 @@ class Slixfeed(slixmpp.ClientXMPP):
             print("> CHCK UPDATE",jid)
             await initdb(jid, download_updates)
             await asyncio.sleep(60 * 90)
+            # Schedule to call this function again in 90 minutes
+            # self.loop.call_at(self.loop.time() + 60 * 90, self.loop.create_task, self.check_updates(event, jid))
 
     async def send_update(self, event, jid):
         """
@@ -194,19 +198,20 @@ class Slixfeed(slixmpp.ClientXMPP):
         :param event:
         :param jid: Jabber ID
         """
-        print("> SEND UPDATE",jid)
         new = await initdb(
             jid,
             database.get_entry_unread
         )
         if new:
+            print("> SEND UPDATE",jid)
             self.send_message(
                 mto=jid,
                 mbody=new,
                 mtype='chat'
             )
         interval = await initdb(jid, database.get_settings_value, 'interval')
-        await asyncio.sleep(60 * interval)
+        # await asyncio.sleep(60 * interval)
+        self.loop.call_at(self.loop.time() + 60 * interval, self.loop.create_task, self.send_update(event, jid))
 
     async def send_status(self, event, jid):
         """
@@ -231,7 +236,8 @@ class Slixfeed(slixmpp.ClientXMPP):
             pto=jid,
             #pfrom=None
         )
-        await asyncio.sleep(60 * 20)
+        # await asyncio.sleep(60 * 20)
+        self.loop.call_at(self.loop.time() + 60 * 20, self.loop.create_task, self.send_status(event, jid))
 
 
 def print_help():
@@ -362,6 +368,7 @@ async def download_updates(db_file):
     urls = await database.get_subscriptions(db_file)
 
     for url in urls:
+        # print(os.path.basename(db_file), url[0])
         source = url[0]
         res = await download_feed(source)
         # TypeError: 'NoneType' object is not subscriptable
@@ -476,6 +483,7 @@ async def add_feed(db_file, url):
         res = await download_feed(url)
         if res[0]:
             feed = feedparser.parse(res[0])
+            title = await get_title(url, feed)
             if feed.bozo:
                 bozo = ("WARNING: Bozo detected. Failed to load <{}>.".format(url))
                 print(bozo)
@@ -635,7 +643,7 @@ async def add_feed(db_file, url):
                 else:
                     msg = "No news feeds were found for URL <{}>.".format(url)
             else:
-                msg = await database.add_feed(db_file, feed, url, res)
+                msg = await database.add_feed(db_file, title, url, res)
         else:
             msg = "Failed to get URL <{}>.  Reason: {}".format(url, res[1])
     else:
@@ -645,6 +653,21 @@ async def add_feed(db_file, url):
     return msg
 
 
+async def get_title(url, feed):
+    """
+    Get title of feed.
+    
+    :param url: URL
+    :param feed: Parsed feed
+    :return: Title or URL hostname.
+    """
+    try:
+        title = feed["feed"]["title"]
+    except:
+        title = urlparse(url).netloc
+    return title
+
+
 def toggle_state(jid, state):
     """
     Set status of update.