match_spider/ybw/utils/tools.py

# -*- coding: utf-8 -*-
"""
Created on 2024-02-28
---------
@summary:  工具库
---------
@author: jy
"""

import re
import socket
import time


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_host_ip():
    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    try:
        s.connect(('8.8.8.8', 80))
        ip = s.getsockname()[0]
    finally:
        s.close()
    return ip


def wait(interval=None):
    _interval = (interval or 600)
    time.sleep(_interval)


def date_to_timestamp(date, time_format="%Y-%m-%d %H:%M:%S"):
    timestamp = time.mktime(time.strptime(date, time_format))
    return int(timestamp)


def get_current_timestamp():
    return int(time.time())