rates_checker.py

import os
import sys
import time
import requests
import logging
import traceback
from decimal import Decimal
from currencyapicom import Client as CurrencyClient

os.environ.setdefault("DJANGO_SETTINGS_MODULE", "duc_rate_admin.settings")
import django

django.setup()

logger = logging.getLogger()
logger.setLevel(logging.DEBUG)

from duc_rate_admin.rates.models import UsdRate, DucRate

CURRENCY_API_KEY = os.getenv("CURRENCY_API_KEY")
COINGECKO_API = os.getenv("COINGECKO_API")
COINGECKO_API_KEY = os.getenv("COINGECKO_API_KEY")


FSYM = "USD"
TSYMS = [
    "INR", "GBP", "EUR", "CAD", "COP", "NGN",
    "BRL", "ARS", "AUD", "JPY", "NZD"
]

class ApiError(Exception):
    pass


def get_currency_rates(tsym, fsyms):
    currency_client = CurrencyClient(CURRENCY_API_KEY)
    response = currency_client.latest(tsym, fsyms)
    data = response["data"]

    rates = {}
    for rate_name, values in data.items():
        rates[rate_name] = values['value']

    return rates


def get_rate_coingecko(coin_id, currencies_id):
    endpoint = COINGECKO_API.format(coin_id, currencies_id)
    headers = {
        "accept": "application/json",
        "x-cg-demo-api-key": COINGECKO_API_KEY
    }

    response = requests.get(url=endpoint, headers=headers)
    if response.ok:
        rate_value = response.json().get(coin_id, {}).get(currencies_id)
        if rate_value:
            return rate_value

    logging.error(f"Error while trying get {currencies_id} rate for {coin_id} (status code: {response.status_code})")
    return None


def update_bnb_rate():
    bnb_usd_rate = get_rate_coingecko("binancecoin", "usd")
    if bnb_usd_rate:
        bnb_rate, _ = DucRate.objects.get_or_create(currency="BNB")
        bnb_rate.rate = Decimal(bnb_usd_rate)
        bnb_rate.save()

        logging.info(f"Updated rate for BNB: {bnb_usd_rate} USD")


def get_rates_main():
    rates = get_currency_rates(FSYM, TSYMS)
    for rate_name, rate_value in rates.items():
        usd_rate, _ = UsdRate.objects.get_or_create(currency=rate_name)
        usd_rate.rate = Decimal(rate_value)
        usd_rate.save()

        logging.info(f"Updated rate for {rate_name}: {rate_value} USD")


if __name__ == "__main__":
    if not all([CURRENCY_API_KEY, COINGECKO_API, COINGECKO_API_KEY]):
        raise Exception("API keys or API url is not provided")

    BNB_CHECKER_TIMEOUT = int(os.getenv("BNB_CHECKER_TIMEOUT", 60))
    RATES_CHECKER_TIMEOUT = int(os.getenv("RATES_CHECKER_TIMEOUT", 86400))

    DELAY = 1
    start = 0

    while True:
        if not bool(start % RATES_CHECKER_TIMEOUT):
            try:
                get_rates_main()
            except Exception as e:
                logging.error(e)

        if not bool(start % BNB_CHECKER_TIMEOUT):
            try:
                update_bnb_rate()
            except Exception as e:
                logging.error(e)

        time.sleep(DELAY)
        start += DELAY