m_if_s_4_n_ppw.py

"""针对于没有支付密码的闲鱼工资程序入口模块"""
from datetime import date

from pacc.config import UnitPrice
from pacc.mysql import RetrieveDispatchRecords, RetrieveIdleFishByUsername, \
    RetrieveIdleFishByConsignee, RetrieveIdleFish, UpdateRecordDispatch, UpdateIdleFishStaff, \
    RetrieveIdleFishStaff, RetrieveRecordDispatch
from pacc.tools import create_dir, get_now_time, send_wechat_msg


class IdleFishSalary4NoPaymentPassword:
    """针对于没有支付密码的闲鱼工资类"""

    @classmethod
    def get_no_payee_jns(cls):
        """获取没有收款人的记录

        :return: 没有收款人的记录
        """
        records = []
        for user_name, dispatch_consignee, base_payee, middle_payee in RetrieveDispatchRecords. \
                query_no_payee_records():
            if not base_payee or not middle_payee:
                un_jn = RetrieveIdleFishByUsername(user_name).job_number
                dc_jn = RetrieveIdleFishByConsignee(dispatch_consignee).job_number
                if un_jn != dc_jn:
                    print(f'un_jn={un_jn}, dc_jn={dc_jn}')
                    print('工号信息不一致，请核对')
                    input()
                records.append(un_jn)
        return records

    @classmethod
    def update_no_payee_records(cls) -> None:
        """更新没有收款人的记录"""
        for job_number in cls.get_no_payee_jns():
            retrieve_ins = RetrieveIdleFish(job_number)
            base_payee, middle_payee = retrieve_ins.base_payee, retrieve_ins.middle_payee
            if not base_payee or not middle_payee:
                print(retrieve_ins.job_number, retrieve_ins.role, retrieve_ins.user_name,
                      base_payee, middle_payee)
                print('还未设置收款人信息，请先设置')
                input()
            update_ins = UpdateRecordDispatch(retrieve_ins.user_name)
            update_ins.update_role(retrieve_ins.role)
            update_ins.update_job_number(job_number)
            update_ins.update_base_payee(base_payee)
            update_ins.update_middle_payee(middle_payee)

    @classmethod
    def get_payee_group_records(cls, level='base'):
        """获取所有人员账号分组汇总后的信息"""
        res = []
        if level == 'base':
            res = RetrieveDispatchRecords.query_base_payee_group_records()
        elif level == 'middle':
            res = RetrieveDispatchRecords.query_middle_payee_group_records()
        # pylint: disable=duplicate-code
        res_dic = {}
        for names, payee in res:
            res_dic.update({payee: str(names).split('||')})
        return res_dic

    @classmethod
    def get_base_payee_group_records(cls):
        """获取所有基层人员账号分组汇总后的信息"""
        # pylint: disable=duplicate-code
        return cls.get_payee_group_records()

    @classmethod
    def get_middle_payee_group_records(cls):
        """获取所有中层人员账号分组汇总后的信息"""
        # pylint: disable=duplicate-code
        return cls.get_payee_group_records('middle')

    @classmethod
    def get_payee_dic(cls):  # pylint: disable=too-many-locals, too-many-branches
        """获取所有人员账号汇总后的字典信息"""
        # pylint: disable=duplicate-code
        base_payee_group_records = cls.get_base_payee_group_records()
        middle_payee_group_records = cls.get_middle_payee_group_records()
        base_mid_dic = {}
        base_dic = {}
        middle_dic = {}
        for base_payee, job_num_li in base_payee_group_records.items():
            base_mid = set(job_num_li) & set(middle_payee_group_records.get(base_payee, []))
            if base_mid:
                base_mid_dic.update({base_payee: list(base_mid)})
        for base_payee, job_num_li in base_payee_group_records.items():
            base = set(job_num_li) - set(middle_payee_group_records.get(base_payee, []))
            if base:
                base_dic.update({base_payee: list(base)})
        for middle_payee, job_num_li in middle_payee_group_records.items():
            middle = set(job_num_li) - set(base_payee_group_records.get(middle_payee, []))
            if middle:
                middle_dic.update({middle_payee: list(middle)})
        res_dic = {}
        for name, job_num_li in base_mid_dic.items():
            dic_li = []
            for job_number in job_num_li:
                retrieve_ins = RetrieveRecordDispatch(job_number)
                dic = {'job_number': retrieve_ins.job_number,
                       'role': retrieve_ins.role,
                       'user_name': retrieve_ins.user_name,
                       'if_mn': retrieve_ins.if_mn,
                       'buy_coins': retrieve_ins.buy_coins,
                       'dispatch_consignee': retrieve_ins.dispatch_consignee,
                       'confirm_date': retrieve_ins.confirm_date,
                       'level': 'base_mid',
                       }
                dic_li.append(dic)
            res_dic.update({name: dic_li})
        for name, job_num_li in base_dic.items():
            dic_li = []
            for job_number in job_num_li:
                retrieve_ins = RetrieveRecordDispatch(job_number)
                dic = {'job_number': retrieve_ins.job_number,
                       'role': retrieve_ins.role,
                       'user_name': retrieve_ins.user_name,
                       'if_mn': retrieve_ins.if_mn,
                       'buy_coins': retrieve_ins.buy_coins,
                       'dispatch_consignee': retrieve_ins.dispatch_consignee,
                       'confirm_date': retrieve_ins.confirm_date,
                       'level': 'base',
                       }
                dic_li.append(dic)
            if name not in res_dic:
                res_dic.update({name: dic_li})
            else:
                res_dic.update({name: dic_li + res_dic.get(name)})
        for name, job_num_li in middle_dic.items():
            dic_li = []
            for job_number in job_num_li:
                retrieve_ins = RetrieveRecordDispatch(job_number)
                dic = {'job_number': retrieve_ins.job_number,
                       'role': retrieve_ins.role,
                       'user_name': retrieve_ins.user_name,
                       'if_mn': retrieve_ins.if_mn,
                       'buy_coins': retrieve_ins.buy_coins,
                       'dispatch_consignee': retrieve_ins.dispatch_consignee,
                       'confirm_date': retrieve_ins.confirm_date,
                       'level': 'middle',
                       }
                dic_li.append(dic)
            if name not in res_dic:
                res_dic.update({name: dic_li})
            else:
                res_dic.update({name: dic_li + res_dic.get(name)})
        if None in res_dic:
            print('未指定收款人的记录如下：')
            print(res_dic.get(None))
            print('\n请对未指定收款人的记录进行处理\n')
            return False
        return res_dic

    @classmethod
    def get_single_info(cls, index, record_dic):
        """获取单条信息

        :param index: 索引
        :param record_dic: 单条记录字典
        :return: 单条信息
        """
        info = f'序号：{str(index).zfill(3)}，工号：{record_dic.get("job_number")}' \
               f'{record_dic.get("role")}，账号：{record_dic.get("user_name")}，' \
               f'手机：{record_dic.get("if_mn")}，鱼币：{record_dic.get("buy_coins") // 10000}万，' \
               f'收货：{record_dic.get("dispatch_consignee")}\n'
        return info

    # pylint: disable=too-many-statements, too-many-branches, too-many-locals
    @classmethod
    def get_payee_message(cls):
        """获取所有人员的通知消息"""
        # pylint: disable=duplicate-code
        payee_group_records = cls.get_payee_dic()
        payee_cnt = 0
        send_cnt = 0
        today = date.today()
        for name, records in payee_group_records.items():
            # print(name, records)
            base_mid_coins = 0
            base_coins = 0
            middle_coins = 0
            for record in records:
                if record.get('level') == 'base_mid':
                    base_mid_coins += record.get('buy_coins')
                elif record.get('level') == 'middle':
                    middle_coins += record.get('buy_coins')
                elif record.get('level') == 'base':
                    base_coins += record.get('buy_coins')
            # print(base_mid_coins, base_coins, middle_coins)
            dir_path = f'D:/0/computer/闲鱼挂机/结账信息/' \
                       f'{str(date.today()).replace("-", "_")}_无支付密码'
            create_dir(dir_path)
            sum_money = UnitPrice.get_base_mid_money(base_mid_coins) + UnitPrice.get_base_money(
                base_coins) + UnitPrice.get_middle_money(middle_coins)
            sum_info = f'收款人：{name}，总钱数：{sum_money}元'
            txt = f'{dir_path}/{name}_{sum_money}元.txt'
            if base_mid_coins != 0:
                sum_info += f'，中基层鱼币共：{base_mid_coins // 10000}万' \
                            f'（{UnitPrice.get_base_mid_money(base_mid_coins)}元）'
            if middle_coins != 0:
                sum_info += f'，中层鱼币共：{middle_coins // 10000}万' \
                            f'（{UnitPrice.get_middle_money(middle_coins)}元）'
            if base_coins != 0:
                sum_info += f'，基层鱼币共：{base_coins // 10000}万' \
                            f'（{UnitPrice.get_base_money(base_coins)}元）'
            sum_info += '。\n\n'
            if base_mid_coins != 0:
                sum_info += '【中基层鱼币账号信息详情如下】\n'
                index = 0
                for record in records:
                    if record.get('level') == 'base_mid':
                        index += 1
                        sum_info += cls.get_single_info(index, record)
            if middle_coins != 0:
                if base_mid_coins != 0:
                    sum_info += '\n'
                sum_info += '【中层鱼币账号信息详情如下】\n'
                index = 0
                for record in records:
                    if record.get('level') == 'middle':
                        index += 1
                        sum_info += cls.get_single_info(index, record)
            if base_coins != 0:
                if base_mid_coins != 0 or middle_coins != 0:
                    sum_info += '\n'
                sum_info += '【基层鱼币账号信息详情如下】\n'
                index = 0
                for record in records:
                    if record.get('level') == 'base':
                        index += 1
                        sum_info += cls.get_single_info(index, record)
            sum_info += '\n'
            sum_info += f'收款人：{name}，总钱数：{sum_money}元，日期：{today}'
            payee_cnt += 1
            retrieve_idlefish_staff_ins = RetrieveIdleFishStaff(name)
            remark = retrieve_idlefish_staff_ins.remark
            if not remark:
                print(f'该员工{name}不存在，或备注{remark}未设置，请处理！')
                input()
            print(f'序号：{payee_cnt}, 姓名：{name}, 备注：{remark}, 金额：{sum_money}元')
            if retrieve_idlefish_staff_ins.last_npps_date != today:
                send_wechat_msg(remark, sum_info)
                update_ins = UpdateIdleFishStaff(name)
                update_ins.update_last_npps_amount(sum_money)
                update_ins.update_last_npps_date(date.today())
                update_ins.update_last_npps_time(get_now_time())
                send_cnt += 1
            else:
                print('今日已发送信息，无需重复发送')
            if send_cnt and not send_cnt % 3:
                print('本轮次人员已完成发送，请按回车键以继续')
                input()
            # print(sum_info)
            with open(txt, 'w+', encoding='utf-8') as file:
                file.write(sum_info)


IdleFishSalary4NoPaymentPassword.update_no_payee_records()
IdleFishSalary4NoPaymentPassword.get_payee_message()