1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
#! /usr/bin/env python
# -*- encoding:utf-8 -*-
# SPDX-License-Identifier: GPL-2.0-or-later
__author__ = "yetist"
__copyright__ = "Copyright (c) 2025 yetist <yetist@gmail.com>"
__license__ = "GPL-2.0-or-later"
import toml
import datetime
class Table(object):
def __init__(self, file):
self.file = file
self.data = {}
def load(self):
fp = open(self.file)
self.data = toml.load(fp)
self.lst = list(self.data.keys())
self.lst.sort()
self.start = self.lst[0]
self.end = self.lst[-1]
self.count = len(self.lst)
def get_year_info(self, year):
year_data = self.data[year]
info = 0
if year_data['leap_days'] == 30:
info += 1
for i in year_data['days']:
info = info << 1
if i == 30:
info |= 1
else:
info |= 0
info = info << 4
info += year_data['leap']
# print(bin(info), info, hex(info))
return hex(info)
def get_fest(self, year):
year_data = self.data[year]
return year_data['jie']
def gen_year_info(self):
header = """
long years_info[NUM_OF_YEARS] = {
/* encoding:
*
* b FEDCBA987654 3210
* bit# 1 111111000000 0000
* 6 543210987654 3210
* . ............ ....
* month# 000000000111
* M 123456789012 L
b_j = 1 for long month, b_j = 0 for short month
L is the leap month of the year if 1<=L<=12; NO leap month if L = 0.
The leap month (if exists) is long one iff M = 1.
*/
"""
print(header)
for i in self.lst:
val = self.get_year_info(i)
if int(i) % 5 == 0:
print("{} /* {} */".format(val, i), end = '\n')
elif int(i) % 5 == 1:
print(" {},".format(val), end = ' ')
elif int(i) % 5 == 4:
print("{} ".format(val), end = ' ')
else:
print("{},".format(val), end = ' ')
print(" /* {} */".format(i), end = '\n};\n\n')
def get_holidays(self, year):
"""
法定节假日编码
使用一个 uint16 类型表示当年的法定假日调整情况
bit: FEDC BA98 7654 3210
fmt: YYYY YYYD DDDD DDDD
.... .... .... ....
year: YYYY YYY0 0000 0000 year[15:9], 1 <= year <= 127, max value is (1<<7) -1 = 127
days: 0000 000D DDDD DDDD days[8:0], 1 <= days <= 366, max value is (1<<9) -1 = 511
year: 年份, 使用[15:9]表示，共7位，值加上2000表示公历年份。 (1<<7) - 1 = 127，最大可表示年份为 2127 年。
days: 日期[8:0], 共9位，值以二进制位来表示当年哪天的上班/休息状态需要调整(即休息变上班，上班变休息)，(1<<9) - 1 = 511位，可表示一年 365 天的状态。
"""
year_data = self.data[year]
if 'statutory_holidays' not in year_data.keys():
return []
days = year_data.get('statutory_holidays')
result = []
for i in days:
if len(i) != 4:
print('{} in {} is not valid date'.format(i, year))
return []
else:
year = int(year)
month, day = int(i[:2]), int(i[2:])
date = datetime.date(year, month, day)
num = date.strftime('%j')
result.append(int(num))
return result
def gen_holidays(self):
print("static gshort statutory_holidays[] = {")
for i in self.lst:
days = self.get_holidays(i)
if days:
year = int(i) - 2000
y = year << 9
for j in days:
day = y | j
print(" {}".format(hex(day)), end = ',')
print(" /* {} */".format(i), end = '\n')
print("}", end = '\n')
if __name__ == '__main__':
t = Table("table.toml")
t.load()
# t.gen_year_info()
t.gen_holidays()
|
