1 #!/usr/bin/env python
2 # -*- coding:utf-8 -*-
3
4
5 """
6 ADT Date: #定义日期对象的抽象数据类型
7 Date(self, int year, int month, int day) #构造表示year/month/day的对象
8 difference(self, Date d2) #求出self和d2的日期差
9 plus(self, int n) #计算出日期第self之后n天的日期
10 num_date(self, int year, int n) #计算year年第n天的日期
11 adjust(self, int n) #将日期d调整n天（n为带符号整数）
12 year(self) #返回日期的年
13 month(self) #返回日期的月
14 day(self) #返回日期的天
15 """
16
17 class Date(object):
18 __slots__ = ('_year', '_month', '_day')
19
20 def __init__(self, year, month, day):
21 if not isinstance(year, int) or not isinstance(month, int) or not isinstance(day, int):
22 raise TypeError
23
24 if 1800<=year<year+30:
25 self._year = year
26 if 1 <= month <= 12:
27 self._month = month
28 bigmonth = (1,3,4,7,8,10,12)
29 smallmonth = (4,6,9,11)
30 if month in bigmonth:
31 if 0<day<=31:
32 self._day = day
33 else:
34 raise ValueError("%d is not valid day!" % day)
35 if month in smallmonth:
36 if 0<day<=30:
37 self._day = day
38 else:
39 raise ValueError("%d is not valid day!" % day)
40 elif month == 2:
41 if Date.leap_year(year):
42 if 0<day<=29:
43 self._day = day
44 else:
45 raise ValueError("%d is not valid day!" % day)
46 else:
47 if 0<day<=28:
48 self._day = day
49 else:
50 raise ValueError("%d is not valid day,the year is not leap_year!" % day)
51 else:
52 raise ValueError("%d is not valid month!" % month)
53 else:
54 raise ValueError("%d is not valid year!" % year)
55
56 def difference(self, other):
57 #日期差
58 DateDiff = 0
59 if self._year > other._year:
60 #换个位置，方便计算
61 tmp = (self._year,self._month,self._day)
62 (self._year,self._month,self._day) = (other._year,other._month,other._day)
63 (other._year, other._month, other._day) = tmp
64
65 #两个年之间的年直接加它一年的天数，分闰年和非闰年区别366和365
66 for i in range(self._year+1,other._year):
67 if Date.leap_year(i):
68 DateDiff += 366
69 else:
70 DateDiff += 365
71 #比较小的年，用后面的月份的天数相加在加上该月剩余的天数
72 for i in range(self._month+1, 13):
73 DateDiff += Date.month_day(self._year,i)
74 DateDiff += Date.month_day(self._year,self._month) - self._day
75
76 #比较大的年，加前面月份的天数加上本月的天数
77 for i in range(1,other._month):
78 DateDiff += Date.month_day(other._year,i)
79 DateDiff += other._day
80 return DateDiff
81
82 def plus(self,n):
83 if not isinstance(n,int):
84 raise TypeError
85 if n<0:
86 raise ValueError("%d is not valid,must >= 0" % n)
87 self._day += n
88 while self._day > Date.month_day(self._year,self._month):
89 self._day -= Date.month_day(self._year,self._month)
90 self._month += 1
91 if self._month == 13:
92 self._month = 1
93 self._year += 1
94 return Date(self._year, self._month, self._day)
95
96 def num_date(self, year, n):
97 if not isinstance(year, int) or not isinstance(n, int):
98 raise TypeError
99 if Date.leap_year(year):
100 if n>366 and n<1:
101 raise ValueError
102 else:
103 if n>365 and n<1:
104 raise ValueError
105 self._year = year
106 #判该n对应的月份和天数
107 for i in range(1,13):
108 d = n
109 n -= Date.month_day(year,i)
110 if n<=0:
111 self._month = i
112 self._day = d
113 break
114 return Date(self._year, self._month, self._day)
115
116 def adjust(self,n):
117 if not isinstance(n,int):
118 raise TypeError
119 #n为正的情况
120 if n>=0:
121 self.plus(n)
122 #n为负的情况
123 else:
124 self._day += n
125 while self._day < 0:
126 if self._month -1 == 0:
127 self._month = 13
128 self._year -= 1
129 self._day += Date.month_day(self._year, self._month-1)
130 self._month -= 1
131 return Date(self._year, self._month, self._day)
132
133 def __str__(self):
134 return str(self._year) + "-" + str(self._month) + "-" + str(self._day)
135
136
137 def year(self):
138 return self._year
139
140 def month(self):
141 return self._month
142
143 def day(self):
144 return self._day
145
146 @staticmethod
147 def leap_year(year):
148 if (year % 4 == 0 and year % 100 != 0) or year % 400 == 0:
149 return True
150 else:
151 return False
152
153 #每月的天数，字典实现
154 @staticmethod
155 def month_day(year,month):
156 d = {}
157 bigmonth = (1,3,5,7,8,10,12)
158 smallmonth = (4,6,9,11)
159 for i in range(1,13):
160 if i in bigmonth:
161 d[i] = 31
162 elif i in smallmonth:
163 d[i] = 30
164 elif i == 2:
165 if Date.leap_year(year):
166 d[i] = 29
167 else:
168 d[i] = 28
169 return d[month]
170
171
172 if __name__=='__main__':
173 d = Date(2003,12,10)
174 d1 = Date(2005,2,28)
175 print(d)
176 print("===")
177 print(d.difference(d1))
178 d.plus(30)
179 print("===")
180 print(d)
181 print("===")
182 d3 = Date(2006,12,13)
183 d3.num_date(2016,10)
184 print(d3)
185 d3.adjust(-20)
186 print("===")
187 print(d3)