Python实践——datetime日期操作脚本

#!/usr/bin/python
# -*- coding: utf-8 -*-
#countDays.py
# version 0.10      edited by lingyue.wkl 20110819 11:00:00
# version 0.11      modified by lingyue.wkl 20110820 11:37:00   add functions for days list
#this script count days,between two date or one date and the days between them

#考虑下，很多方法可以抽象出来，进一步优化，先期先实现功能吧
#下一个版本  改进所有函数，优化之，抽象之


import time,getopt,sys,datetime

def date_to_str(in_date):
    return str(in_date)[:10]
#计算两个日期之间相隔天数
def get_count_between_two_date(begin_date,end_date):
  b_date = begin_date.split("-")
  b_date = [int(num) for num in b_date]
  b_date_time = datetime.datetime(b_date[0],b_date[1],b_date[2])

  e_date = end_date.split("-")
  e_date = [int(num) for num in e_date]
  e_date_time = datetime.datetime(e_date[0],e_date[1],e_date[2])

  return (e_date_time - b_date_time).days

#计算某个日期前n天是哪一天   默认日期是今天
def get_n_days_before_or_after_oneday(n_days,in_date=str(datetime.date.today())[:10]):
  begin_date = in_date.split("-")
  begin_date = [int(num) for num in begin_date]
  return str(datetime.datetime(begin_date[0],begin_date[1],begin_date[2]) + datetime.timedelta(days=n_days))[:10]


def get_year():
    return str(datetime.date.today())[:4]

def get_month():
    return str(datetime.date.today())[5:7]

def get_day():
    return str(datetime.date.today())[8:]

def get_now():
    return datetime.datetime.now()

def get_today():
    return datetime.date.today()

def get_yesterday():
  return get_n_days_before_or_after_oneday(-1,str(datetime.date.today())[:10])

def get_tomorrow():
  return get_n_days_before_or_after_oneday(1,str(datetime.date.today())[:10])

#两个日期之间  n天的日期列表

def get_n_daystimes_list_of_two_date(begin_date,end_date):
  b_date = begin_date.split("-")
  b_date = [int(num) for num in b_date]
  b_date_time = datetime.datetime(b_date[0],b_date[1],b_date[2])

  e_date = end_date.split("-")
  e_date = [int(num) for num in e_date]
  e_date_time = datetime.datetime(e_date[0],e_date[1],e_date[2])

  days = (e_date_time - b_date_time).days
  n_days_list = []
  for i in range(0,days+1):
    n_days_list.append(str(b_date_time + datetime.timedelta(days=i)))
  return n_days_list

def get_n_days_list_of_two_date(begin_date,end_date):
  return [str(day)[:10] for day in get_n_daystimes_list_of_two_date(begin_date,end_date)]

def get_n_dayswiththreetimes_list_of_two_date(begin_date,end_date):
  days =  get_n_days_list_of_two_date(begin_date,end_date)
  days_three_time_list = []
  for day in days:
    for i in range(0,3):
        if i == 0:
           days_three_time_list.append(day+" 00:00:00")
        elif i == 1:
           days_three_time_list.append(day+" 12:00:00")
        else:
           days_three_time_list.append(day+" 23:59:59")
  return days_three_time_list

#某个日期之前n天  所有日期列表

def get_n_daystimes_list_before_or_after_one_day(n_days,end_date=str(datetime.date.today())[:10]):
  begin_date = get_n_days_before_or_after_oneday(n_days,end_date)
  return get_n_daystimes_list_of_two_date(begin_date,end_date)

def get_n_days_list_before_or_after_one_day(n_days,end_date=str(datetime.date.today())[:10]):
  begin_date = get_n_days_before_or_after_oneday(n_days,end_date)
  return get_n_days_list_of_two_date(begin_date,end_date)

def get_n_dayswiththreetimes_list_before_or_after_one_day(n_days,end_date=str(datetime.date.today())[:10]):
  begin_date = get_n_days_before_or_after_oneday(n_days,end_date)
  return get_n_dayswiththreetimes_list_of_two_date(begin_date,end_date)

def help_msg():
  print("功能：日期相关操作")
  print("选项:")
  print("\t 默认，无选项，输出当天日期，格式2011-08-20")
  print("\t -y   [可选，输出当前年份]")
  print("\t -m   [可选，输出当前月份]")
  print("\t -d   [可选，输出当前日]")
  print("\t -n +-数字  [可选，计算当前日期前后多少天的日期，数字为负表示往前]")
  print("\t -f 2011-10-22[可选,指定坐标日期，即以指定日期开始计算,若不指定，坐标日期为当天]")
  print("\t -t 2011-10-25  [可选，目标日期，可用于计算两个日期相隔天数]")
  print("\t -l [1|2|3]  [可选，是否列表，若选定，输出日期间的所有序列，1 2 3 代表三种不同格式]")
  sys.exit(0)

def print_list(l):
  for i in l:
    print(i)
#print(get_year())
#print(get_month())
#print(get_day())
#print(get_now())
#print(get_today())
#print(get_yesterday())
#print(get_tomorrow())
#print(get_n_days_before_or_after_oneday(2,"2011-08-20"))
#print(get_n_daystimes_list_of_two_date("2011-08-01","2011-08-05"))
#print(get_n_days_list_of_two_date("2011-08-01","2011-08-05"))
#print(get_n_dayswiththreetimes_list_of_two_date("2011-08-01","2011-08-05")) 
#print(get_n_daystimes_list_before_or_after_ond_day(-5,"2011-08-20"))
#print(get_n_days_list_before_or_after_ond_day(-5,"2011-08-20"))
#print(get_n_dayswiththreetimes_list_before_or_after_ond_day(-5,"2011-08-20"))

#程序入口，读入参数，执行
def main():
    is_list = False
    try:
        opts,args = getopt.getopt(sys.argv[1:],"n:f:t:o:ymdhrl:")
        
        if len(opts) == 0:
          print(get_today())
          sys.exit(0)

        for op,value in opts:
          if op in ("-h","-H","--help"):
            help_msg()
          if op == "-y":
            print(get_year())
            sys.exit(0)
          elif op == "-m":
            print(get_month())
            sys.exit(0)
          elif op == "-d":
            print(get_day())
            sys.exit(0)
          elif op == "-n":
            n_days = int(value)
          elif op == "-f":
            from_date = value
          elif op == "-t":
            to_date = value
          elif op == "-l":
            is_list = True
            list_type = value

    except getopt.GetoptError:
      print(sys.argv[0]+" : params are not defined well!")

    #if "n_days" not in dir() and "from_date" not in dir() and "to_date" not in dir():
    #     print(result_str)
   
    if "n_days" in dir() and "from_date" not in dir() and "to_date" not in dir():
      if not is_list:
         print(get_n_days_before_or_after_today(n_days))
      else:
         if list_type == "1":
           result_list = get_n_days_list_before_or_after_one_day(n_days)
         elif list_type == "2":
           result_list = get_n_daystimes_list_before_or_after_one_day(n_days)
         elif list_type == "3":
           result_list = get_n_dayswiththreetimes_list_before_or_after_one_day(n_days)
         print_list(result_list)
   
    if "n_days" in dir() and "from_date" in dir() and "to_date" not in dir():
      if not is_list:
         print(get_n_days_before_or_after_oneday(n_days,from_date))
      else:
         if list_type == "1":
           result_list = get_n_days_list_before_or_after_one_day(n_days,from_date)
         elif list_type == "2":
           result_list = get_n_daystimes_list_before_or_after_one_day(n_days,from_date)
         elif list_type == "3":
           result_list = get_n_dayswiththreetimes_list_before_or_after_one_day(n_days,from_date)
         print_list(result_list)


    if "n_days" not in dir() and "from_date" in dir() and "to_date" in dir():
      if not is_list:
         print(get_count_between_two_date(from_date,to_date))
      else:
         if list_type == "1":
           result_list = get_n_days_list_of_two_date(from_date,to_date)
         elif list_type == "2":
           result_list = get_n_daystimes_list_of_two_date(from_date,to_date)
         elif list_type == "3":
           result_list = get_n_dayswiththreetimes_list_of_two_date(from_date,to_date)
         print_list(result_list)
      
      

main()

最近需要做一些日期的操作，写了一个脚本