Fork me on GitHub

python脚本demo

import atexit
import logging
import os
import signal
import sys
from logging.handlers import RotatingFileHandler



logger = logging.getLogger()
MAX = 10 * 1024 * 1024
BACK_UP_COUNT = 10
def setup_logging():
    log_file = os.path.join(r'C:\\',"demo.log")
    logger.setLevel(logging.INFO)
    file_handler = RotatingFileHandler(log_file,"a",MAX,BACK_UP_COUNT)
    file_handler.setFormatter(
        logging.Formatter(
            "[%(asctime)%s] [%(process)d] [%(levelname)s] - %(module)s.%(funcName)s"
            "(%(filename)s:%(lineno)d) - %(message)s"
        )
    )
    logger.addHandler(file_handler)
PID_FILE = "fmpid.pid"
def write_pid():
    with open(PID_FILE,"w") as f:
        f.write('%s\n'%(os.getpid()))

def read_pid():
    try:
        with open(PID_FILE) as f:
            pid = int(f.read().strip())
    except Exception:
        pid = None
    return pid

def remove_pid():
    if os.path.exists(PID_FILE):
        os.remove(PID_FILE)

def sig_handler(sig,frame):
    logger.warning("shutting down, got signal %d"%sig)
    remove_pid()
    sys.exit(0)

def setup_application():
    pid = read_pid()
    if pid:
        sys.stdout.write('processing is running...\n')
        sys.exit(0)
    setup_logging()
    write_pid()
    atexit.register(remove_pid)
    for sig in (signal.SIGTERM,signal.SIGINT):
        signal.signal(sig,sig_handler)

def main():
    setup_application()
    # 逻辑

posted @   是阿凯啊  阅读(421)  评论(0编辑  收藏  举报
编辑推荐:
· Linux系列:如何用 C#调用 C方法造成内存泄露
· AI与.NET技术实操系列(二):开始使用ML.NET
· 记一次.NET内存居高不下排查解决与启示
· 探究高空视频全景AR技术的实现原理
· 理解Rust引用及其生命周期标识(上)
阅读排行:
· 物流快递公司核心技术能力-地址解析分单基础技术分享
· .NET 10首个预览版发布:重大改进与新特性概览!
· 单线程的Redis速度为什么快?
· 展开说说关于C#中ORM框架的用法!
· Pantheons:用 TypeScript 打造主流大模型对话的一站式集成库
点击右上角即可分享
微信分享提示