reply_gen.py

from hijri_converter import convert
from aibot_utils import concatenate_bi, cleaning
import datetime
from aibot_date import gregorian_to_jalali
from vocab import miladimonthes, qamariMonthes, num2fa_gen


tr_jalali_month = {1: "فروردین", 2: "اردیبهشت", 3: "خرداد", 4: "تیر", 5: "مرداد",
                   6: "شهریور", 7: "مهر", 8: "آبان", 9: "آذر", 10: "دی", 11: "بهمن", 12: "اسفند"}
tr_gregorian_month = {v: k for k, v in miladimonthes.items()}
tr_hijri_month = {v: k for k, v in qamariMonthes.items()}


def tr_single_date(date_, force_date=False):
    if isinstance(date_, datetime.datetime):
        date_ = date_.date()
    today = datetime.datetime.today()
    if date_ == today.date() and not force_date:
        translation = "امروز"
    elif date_ == (today + datetime.timedelta(1)).date() and not force_date:
        translation = "فردا"
    elif date_ == (today - datetime.timedelta(1)).date() and not force_date:
        translation = "دیروز"
    else:
        j = gregorian_to_jalali(date_.year, date_.month, date_.day)
        today_j = gregorian_to_jalali(today.year, today.month, today.day)
        month = tr_jalali_month[j[1]]
        if j[0] == today_j[0]:
            translation = "{} {}".format(
                num2fa_gen[j[2]], month)
        elif j[0] == today_j[0] + 1:
            translation = "{} {} سال بعد".format(
                num2fa_gen[j[2]], month)
        elif j[0] == today_j[0] - 1:
            translation = "{} {} پارسال".format(
                num2fa_gen[j[2]], month)
        else:
            translation = "{} {} سال {}".format(
                num2fa_gen[j[2]], month, j[0] if j[0] >= 1400 else j[0] - 1300)
    return translation


def tr_gregorian_date(date_):
    if isinstance(date_, datetime.datetime):
        date_ = date_.date()
    month = tr_gregorian_month[date_.month]
    translation = "{} {} سال {}".format(
        num2fa_gen[date_.day], month, date_.year)
    return translation


def tr_hijri_date(date_):
    if isinstance(date_, datetime.datetime):
        date_ = date_.date()

    date_h = convert.Gregorian(date_.year, date_.month, date_.day).to_hijri()
    month = tr_hijri_month[date_h.month]
    translation = "{} {} سال {}".format(
        num2fa_gen[date_h.day], month, date_h.year)
    return translation


def tr_date(date_list, tokens, labels):
    d_words = []
    if len(date_list) <= 2:
        for d in date_list:
            y = tr_single_date(d.date())
            d_words.append(cleaning(y))
    else:
        d_words = concatenate_bi(tokens, labels, "B_DAT", "I_DAT")
    return d_words


def tr_single_time(tim, literal=False):
    now_time = datetime.datetime.now().time()
    if literal:
        if tim.hour == now_time.hour and abs(tim.minute - now_time.minute) <= 1:
            return "الآن"

    if tim.hour != 0:
        if tim.minute != 0:
            return "ساعت {} و {} دقیقه".format(tim.hour, tim.minute)
        else:
            return "ساعت {}".format(tim.hour)
    else:
        if tim.minute != 0:
            return "ساعت ۱۲ و {} دقیقه‌ی بامداد".format(tim.minute)
        else:
            return "۱۲ بامداد"


def tr_time(time_iso, tokens, labels):
    t_words = []
    if len(time_iso) <= 2:
        for t in time_iso:
            y = tr_single_time(t)
            t_words.append(cleaning(y))
    else:
        t_words = concatenate_bi(tokens, labels, "B_TIM", "I_TIM")
    return t_words