utils.py

import os
import json
import sys
import time
from datetime import datetime
import pytz
from tzlocal import get_localzone
import requests
import textwrap
import telegramify_markdown


def load_json_file(file_path):
    try:
        with open(file_path, 'r') as f:
            return json.load(f)
    except FileNotFoundError:
        print("\n{} not found or is invalid.\n".format(file_path))
        return {}


def save_json_file(file_path, data):
    with open(file_path, 'w') as f:
        if isinstance(data, list):
            json.dump(data, f, separators=(',', ':'), ensure_ascii=False)
        else:
            json.dump(data, f, indent=4)


def load_txt_file_to_list(file_path):
    try:
        with open(file_path, 'r') as file:
            lines = file.readlines()

        lines = [line.strip() for line in lines]
        return lines
    except FileNotFoundError:
        print("\nThe file not found or is invalid.\n")
        sys.exit(1)


def can_be_float(string):
    try:
        float(string)
        return True
    except ValueError:
        return False


def convert_utc_to_user_timezone(utc_time_str):
    utc_time = datetime.strptime(utc_time_str,
                                 "%Y-%m-%dT%H:%M:%SZ").replace(tzinfo=pytz.UTC)
    local_timezone = get_localzone()
    user_time = utc_time.astimezone(local_timezone)

    return user_time


def is_list_of_strings(variable):
    return isinstance(variable, list) and all(
        isinstance(item, str) for item in variable)


def is_list_of_dicts(variable):
    return isinstance(variable, list) and all(
        isinstance(item, dict) for item in variable)


def send_telegram_message(bot_token, chat_id, message, verbose=True):
    url = f"https://api.telegram.org/bot{bot_token}/sendMessage"
    payload = {"chat_id": chat_id, "text": message, "parse_mode": "MarkdownV2"}
    response = requests.post(url, json=payload)

    if verbose:
        if response.status_code == 200:
            print("Message sent to Telegram successfully!")
        else:
            print(f"Failed to send message. Error: {response.text}")


def chunk_message(messages, max_length=4096):
    chunks = []
    current_chunk = ""

    for message in messages:
        markdown_text = textwrap.dedent(message)
        can_be_sent = telegramify_markdown.markdownify(markdown_text)

        if len(current_chunk) + len(can_be_sent) <= max_length:
            current_chunk += can_be_sent + "\n"
        else:
            chunks.append(current_chunk.strip())
            current_chunk = can_be_sent + "\n"

    if current_chunk:
        chunks.append(current_chunk.strip())

    return chunks


def send_to_telegram(msg_list,
                     bot,
                     chat_id,
                     max_retries,
                     try_sleep=3,
                     except_sleep=60):

    chunks = chunk_message(msg_list)
    for chunk in chunks:
        retry_count = 0
        while retry_count < max_retries:
            try:
                bot.send_message(chat_id, chunk, parse_mode='MarkdownV2')
                time.sleep(try_sleep)
                break
            except:
                retry_count += 1
                time.sleep(except_sleep)
        else:
            print("Max retries reached. No new message will be sent.\n")
            break