func.py

import logging
import warnings
import time
import os
from datetime import datetime, timedelta
import pandas as pd


#日志函数
def logger(savedir):
    os.makedirs(savedir, exist_ok=True)
    warnings.filterwarnings('ignore', category=Warning, append=False)
    current_time = time.strftime('%Y-%m-%d_%H:%M:%S', time.localtime())

    logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s',
                        filename=os.path.join(savedir, f'{current_time}.log'))
    # 创建一个控制台处理程序并设置其级别为INFO
    console_handler = logging.StreamHandler()
    console_handler.setLevel(logging.INFO)

    # 创建一个格式化程序，用于在控制台上显示日志
    console_formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s')

    # 将格式化程序添加到控制台处理程序
    console_handler.setFormatter(console_formatter)

    # 将控制台处理程序添加到根日志记录器
    logging.getLogger('').addHandler(console_handler)
    
#从namelis.was文件中提取起止时间 
def get_time(namelist_was):
    with open(namelist_was, 'r') as file:
        namelist_content = file.read()
        
    # 使用字符串分割和切片来提取日期，并去除单引号
    start_dates = [date.strip("',") for date in namelist_content.split('start_date =')[1].split(',')[0:3]]
    end_dates = [date.strip("',") for date in namelist_content.split('end_date =')[1].split(',')[0:3]]
    # 将时间字符串转换为datetime对象
    start_dates_datetime = [datetime.strptime(date, '%Y-%m-%d_%H:%M:%S') for date in start_dates]
    end_dates_datetime = [datetime.strptime(date, '%Y-%m-%d_%H:%M:%S') for date in end_dates]
    # 使用min()函数找出最早的日期
    Earliest_start_date = min(start_dates_datetime)
    Lastest_end_date = max(end_dates_datetime)
    # 打印最早的日期
    print(f"Earliest Start Date: {Earliest_start_date}")
    print(f"Lastest End Date: {Lastest_end_date}")
    return Earliest_start_date,Lastest_end_date

# 生成时间序列
def generate_dates(start_time, end_time, freq='1H'):
    """
    生成给定时间范围内的时间序列。
    :param start_time: 开始时间，格式 'YYYY-MM-DD-HH'
    :param end_time: 结束时间，格式 'YYYY-MM-DD-HH'
    :param freq: 时间频率，默认为每小时 ('1H')
    :return: 格式化后的时间序列列表
    """
    start_time = datetime.strptime(start_time, '%Y-%m-%d-%H')
    end_time = datetime.strptime(end_time, '%Y-%m-%d-%H')
    dates = pd.date_range(start_time, end_time, freq=freq).strftime('%Y-%m-%d-%H')
    return dates