ybw/utils/tools.py

# -*- coding: utf-8 -*-
"""
Created on 2024-02-02
---------
@summary: 工具类
---------
@author: Lzz
"""
import datetime
import re
import time

import bson
import requests

import setting
from log import logger


def clean_title(title):
    if title:
        rule_list = [
            '\(\d{1,20}\)',
            '\[[\u4e00-\u9fa5]{1,9}\]',
            '【[\u4e00-\u9fa5]{1,9}】',
        ]
        for rule in rule_list:
            title = re.sub(rule, '', title)

    return title


def get_proxy():
    url = setting.PROXY_API
    headers = {"Authorization": setting.PROXY_TOKEN}
    proxy = requests.get(url, headers=headers, timeout=10).json().get("data")
    logger.info("切换代理：{}".format(proxy))
    return proxy


def int2long(param: int):
    """int 转换成 long """
    return bson.int64.Int64(param)


def get_current_date(date_format="%Y-%m-%d %H:%M:%S"):
    return datetime.datetime.now().strftime(date_format)


def date_to_timestamp(date, fmt="%Y-%m-%d %H:%M:%S"):
    """
    @summary:
    ---------
    @param date:将"2011-09-28 10:00:00"时间格式转化为时间戳
    @param fmt:时间格式
    ---------
    @result: 返回时间戳
    """
    if ":" in date:
        timestamp = time.mktime(time.strptime(date, fmt))
    else:
        timestamp = time.mktime(time.strptime(date, "%Y-%m-%d"))
    return int(timestamp)


def get_today_of_day(day_offset=0):
    return str(datetime.date.today() + datetime.timedelta(days=day_offset))