yixueServer/logic.py

from model import Wannianli, WannianliJieqi
from tools import *
from yixue import *


def get_last_day_info(year: int):
    dt = Wannianli.select().where(Wannianli.nian == year, Wannianli.yue == 12, Wannianli.ri == 31)
    data = dt[0]
    if data.ri_gan == "":
        return None
    else:
        return data


def _do_update_ri_by_year(year: int, start_rigan: str, start_rizhi: str):
    last_rigan = start_rigan
    last_rizhi = start_rizhi
    for month in range(1, 13):
        max_day = get_max_day(year, month)
        print("处理", year, month, "月数据")
        for day in range(1, max_day + 1):
            if month == 1 and day == 1:
                rigan = last_rigan
                rizhi = last_rizhi
            else:
                rigan = next_tiangan(last_rigan)
                rizhi = next_dizhi(last_rizhi)
                last_rigan = rigan
                last_rizhi = rizhi
            Wannianli.update({
                "ri_gan": rigan,
                "ri_zhi": rizhi,
            }).where(Wannianli.nian == year, Wannianli.yue == month, Wannianli.ri == day).execute()
    print("update over last:", last_rigan, last_rizhi)


def update_ri_by_year(year: int):
    start = Wannianli.select().where(Wannianli.nian == year, Wannianli.yue == 1, Wannianli.ri == 1)
    start_data = start[0]
    if start_data.ri_gan == "":
        print(year, "1月1日未配置初始值，从上一年取得数据")
        last_year = get_last_day_info(year - 1)
        if last_year is not None:
            print("上一年最后一天是", last_year.ri_gan, last_year.ri_zhi)
            next_rigan = next_tiangan(last_year.ri_gan)
            next_rizhi = next_dizhi(last_year.ri_zhi)
            print("新年第一天的开始是", next_rigan, next_rizhi)
            _do_update_ri_by_year(year, next_rigan, next_rizhi)
        else:
            print("上一年也没数据")
    else:
        _do_update_ri_by_year(year, start_data.ri_gan, start_data.ri_zhi)


def get_wannianli_data(year: int, month: int, day: int):
    dt = Wannianli.select().where(Wannianli.nian == year, Wannianli.yue == month, Wannianli.ri == day)
    if len(dt) > 0:
        return dt[0]
    else:
        print(year, month, day, "has empty data")
        return None


def get_jieqi_of_month(jieqi: WannianliJieqi, month: int):
    ary = [jieqi.xiaohan, jieqi.lichun, jieqi.jingzhe, jieqi.qingming, jieqi.lixia,
           jieqi.mangzhong, jieqi.xiaoshu, jieqi.liqiu, jieqi.bailu, jieqi.hanlu,
           jieqi.lidong, jieqi.daxue]
    return ary[month - 1]


def _do_update_yue_by_year(year: int, month: int, start_yuegan: str, start_yuezhi: str):
    jieqi = WannianliJieqi.select().where(WannianliJieqi.nianfen == year)
    jieqi_day = get_jieqi_of_month(jieqi[0], month)
    print("处理", year, month, "月数据, 节气日期是：", jieqi_day)
    key_day = str(jieqi_day).split(".")[1]
    max_day = get_max_day(year, month)
    for day in range(1, max_day + 1):
        if day < int(key_day):
            yuegan = start_yuegan
            yuezhi = start_yuezhi
        else:
            yuegan = next_tiangan(start_yuegan)
            yuezhi = next_dizhi(start_yuezhi)
        # print(year, month, day, yuegan, yuezhi)
        Wannianli.update({
            "yue_gan": yuegan,
            "yue_zhi": yuezhi,
        }).where(Wannianli.nian == year, Wannianli.yue == month, Wannianli.ri == day).execute()
    print("update month over ", year, month)


def prepare_yue_data(year: int, month: int):
    if month == 1:
        print(year, month, "月份未配置初始值，从上年12月取得数据")
        last_month = get_wannianli_data(year - 1, 12, 31)
    else:
        print(year, month, "月份未配置初始值，从当年上个月取得数据")
        max_day = get_max_day(year, month - 1)
        last_month = get_wannianli_data(year, month - 1, max_day)
    if last_month.yue_gan == "":
        print("未取得合适的数据", year, month, "，请检查上年12月或本年上个月的数据是否已录入")
        return False
    else:
        _do_update_yue_by_year(year, month, last_month.yue_gan, last_month.yue_zhi)
        return True


def update_yue_by_year(year: int):
    for month in range(1, 13):
        start = Wannianli.select().where(Wannianli.nian == year, Wannianli.yue == month, Wannianli.ri == 1)
        start_data = start[0]
        if start_data.yue_gan == "":
            if not prepare_yue_data(year, month):
                break
        else:
            if year == 1901 and month == 1:
                _do_update_yue_by_year(year, month, start_data.yue_gan, start_data.yue_zhi)
            else:
                if not prepare_yue_data(year, month):
                    break


def _do_update_nian_by_year(year: int, start_niangan: str, start_nianzhi: str):
    jieqi = WannianliJieqi.select().where(WannianliJieqi.nianfen == year)
    jieqi_key = str(jieqi[0].chuxi).split(".")
    jieqi_month = int(jieqi_key[0])
    jieqi_day = int(jieqi_key[1])
    for month in range(1, 13):
        max_day = get_max_day(year, month)
        for day in range(1, max_day + 1):
            niangan = ""
            nianzhi = ""
            if month < jieqi_month:
                niangan = start_niangan
                nianzhi = start_nianzhi
            elif month == jieqi_month:
                if day > jieqi_day:
                    niangan = next_tiangan(start_niangan)
                    nianzhi = next_dizhi(start_nianzhi)
                else:
                    niangan = start_niangan
                    nianzhi = start_nianzhi
            elif month > jieqi_month:
                niangan = next_tiangan(start_niangan)
                nianzhi = next_dizhi(start_nianzhi)
            Wannianli.update({
                "nian_gan": niangan,
                "nian_zhi": nianzhi,
            }).where(Wannianli.nian == year, Wannianli.yue == month, Wannianli.ri == day).execute()
        print(year, month, "年数据更新完成")


def update_nian_by_year(year: int):
    start = Wannianli.select().where(Wannianli.nian == year, Wannianli.yue == 1, Wannianli.ri == 1)
    start_data = start[0]
    if start_data.nian_gan == "":
        print("没有1月1日数据，从上一年取数")
        last_year = get_wannianli_data(year - 1, 12, 31)
        if last_year is not None:
            _do_update_nian_by_year(year, last_year.nian_gan, last_year.nian_zhi)
    else:
        _do_update_nian_by_year(year, start_data.nian_gan, start_data.nian_zhi)


def get_hour_of_day(day_tiangan: str, hour: int):
    start_tiangan = get_tiangan_start_of_hour(day_tiangan)
    tiangan = get_tiangan_by_hour(start_tiangan, hour)
    dizhi = get_dizhi_by_hour(hour)
    return [tiangan, dizhi]