halib 0.1.29__tar.gz → 0.1.31__tar.gz

{halib-0.1.29/halib.egg-info → halib-0.1.31}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: halib
-Version: 0.1.29
+Version: 0.1.31
 Summary: Small library for common tasks
 Author: Hoang Van Ha
 Author-email: hoangvanhauit@gmail.com

{halib-0.1.29 → halib-0.1.31}/halib/tele_noti.py

@@ -7,12 +7,14 @@ import telegram
 import pandas as pd
 
 from rich.pretty import pprint
+from rich.console import Console
 import plotly.graph_objects as go
 
 from .system import filesys as fs
 from .filetype import textfile, csvfile
 
 from argparse import ArgumentParser
+tele_console = Console()
 
 def parse_args():
     parser = ArgumentParser(
@@ -90,7 +92,7 @@ def compose_message_and_img_path(
     return context_msg, img_path
 
 
-async def send_to_telegram(cfg_dict):
+async def send_to_telegram(cfg_dict, interval_in_sec):
     # pprint(cfg_dict)
     token = cfg_dict["telegram"]["token"]
     chat_id = cfg_dict["telegram"]["chat_id"]
@@ -114,10 +116,20 @@ async def send_to_telegram(cfg_dict):
                 out_img_scale,
                 output_img_dir,
             )
-            pprint('Sending message to telegram...')
+            time_now = next_time = pd.Timestamp.now().strftime("%Y-%m-%d %H:%M:%S")
+            sep_line = "-" * 50
+            context_msg = f"{sep_line}\n>> Time: {time_now}\n{context_msg}"
+            # calculate the next time to send message
+            next_time = pd.Timestamp.now() + pd.Timedelta(seconds=interval_in_sec)
+            next_time = next_time.strftime("%Y-%m-%d %H:%M:%S")
+            next_time_info = f"Next msg: {next_time}"
+            tele_console.rule()
+            tele_console.print("[green] Send message to telegram [/green]")
+            tele_console.print(f"[red] Next message will be sent at <{next_time}> [/red]")
             await bot.send_message(text=context_msg, chat_id=chat_id)
             if img_path:
                 await bot.send_photo(chat_id=chat_id, photo=open(img_path, "rb"))
+            await bot.send_message(text=next_time_info, chat_id=chat_id)
         except Exception as e:
             pprint(f'Error: {e}')
             pprint('Message not sent to telegram')
@@ -130,7 +142,7 @@ async def run_forever(cfg_path):
     interval_in_sec = int(interval_in_min * 60)
     pprint(f'Message will be sent every {interval_in_min} minutes or {interval_in_sec} seconds')
     while True:
-        await send_to_telegram(cfg_dict)
+        await send_to_telegram(cfg_dict, interval_in_sec)
         await asyncio.sleep(interval_in_sec)
 
 async def main():

{halib-0.1.29 → halib-0.1.31/halib.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: halib
-Version: 0.1.29
+Version: 0.1.31
 Summary: Small library for common tasks
 Author: Hoang Van Ha
 Author-email: hoangvanhauit@gmail.com

{halib-0.1.29 → halib-0.1.31}/setup.py

@@ -8,7 +8,7 @@ with open("requirements.txt") as f:
 
 setuptools.setup(
     name="halib",
-    version="0.1.29",
+    version="0.1.31",
     author="Hoang Van Ha",
     author_email="hoangvanhauit@gmail.com",
     description="Small library for common tasks",